1 files changed, 77 insertions, 0 deletions
diff --git a/test/source_type_test.cc b/test/source_type_test.cc
new file mode 100644
index 0000000..2da1ac4
--- /dev/null
+++ b/test/source_type_test.cc
@@ -0,0 +1,77 @@
+#include <uunit.h>
+
+#include <libctor.h>
+#include <task_cc.h>
+#include <settings.h>
+
+std::ostream& operator<<(std::ostream& stream, const Language& lang)
+{
+	switch(lang)
+	{
+	case Language::Auto:
+		stream << "Language::Auto";
+		break;
+	case Language::C:
+		stream << "Language::C";
+		break;
+	case Language::Cpp:
+		stream << "Language::Cpp";
+		break;
+	case Language::Asm:
+		stream << "Language::Asm";
+		break;
+	}
+
+	return stream;
+}
+class TestableTaskCC
+	: public TaskCC
+{
+public:
+	TestableTaskCC(const Source& source)
+		: TaskCC({}, {}, {}, source)
+	{}
+
+	Language language() const
+	{
+		return source_language;
+	}
+};
+
+class SourceTypeTest
+	: public uUnit
+{
+public:
+	SourceTypeTest()
+	{
+		uTEST(SourceTypeTest::test);
+	}
+
+	void test()
+	{
+		{ // c++
+			TestableTaskCC task("hello.cc");
+			uASSERT_EQUAL(Language::Cpp, task.language());
+		}
+
+		{ // c
+			TestableTaskCC task("hello.c");
+			uASSERT_EQUAL(Language::C, task.language());
+		}
+
+		{ // asm
+			TestableTaskCC task("hello.s");
+			uASSERT_EQUAL(Language::Asm, task.language());
+		}
+
+		{ // custom/explicit language
+			TestableTaskCC task( {"hello.foo", Language::Asm} );
+			uASSERT_EQUAL(Language::Asm, task.language());
+		}
+
+		// Note: Failure state will result in exit(1) so cannot be tested
+	}
+};
+
+// Registers the fixture into the 'registry'
+static SourceTypeTest test;