Merge pull request #166 from joshuawarner32/master

Improve c++ unit test harness

.gitignore

@@ -9,3 +9,4 @@ bin
 /distrib
 *.pdb
 *.swp
+/*.sublime-*

unittest/codegen/assembler-test.cpp

@@ -62,25 +62,12 @@ public:
 };
 
 
-class BasicAssemblerTest : public Test {
-public:
-  BasicAssemblerTest():
-    Test("BasicAssembler")
-  {}
-
-  virtual void run() {
+TEST(BasicAssembler) {
   BasicEnv env;
   Asm a(env);
 }
-} basicAssemblerTest;
 
-class ArchitecturePlanTest : public Test {
-public:
-  ArchitecturePlanTest():
-    Test("ArchitecturePlan")
-  {}
-
-  virtual void run() {
+TEST(ArchitecturePlan) {
   BasicEnv env;
 
   for(int op = (int)lir::Call; op < (int)lir::AlignedJump; op++) {
@@ -93,4 +80,3 @@ public:
   }
 
 }
-} architecturePlanTest;

unittest/codegen/registers-test.cpp

@@ -19,13 +19,7 @@ using namespace avian::codegen;
 using namespace vm;
 
 
-class RegisterIteratorTest : public Test {
-public:
-  RegisterIteratorTest():
-    Test("RegisterIterator")
-  {}
-
-  virtual void run() {
+TEST(RegisterIterator) {
   RegisterMask regs(0x55);
   assertEqual<unsigned>(0, regs.start);
   assertEqual<unsigned>(7, regs.limit);
@@ -41,4 +35,3 @@ public:
   assertEqual<unsigned>(6, it.next());
   assertFalse(it.hasNext());
 }
-} registerIteratorTest;

unittest/test-harness.h

@@ -22,21 +22,23 @@ private:
 
   friend int main(int argc, char** argv);
 
-
-  void print(uint64_t value) {
+  void print(uint64_t value)
+  {
     fprintf(stderr, "%p", reinterpret_cast<void*>(value));
   }
 
-  void print(uint32_t value) {
+  void print(uint32_t value)
+  {
     fprintf(stderr, "%p", reinterpret_cast<void*>(value));
   }
 
-
-  void print(uint8_t value) {
+  void print(uint8_t value)
+  {
     print(static_cast<uint32_t>(value));
   }
 
-  void print(bool value) {
+  void print(bool value)
+  {
     fprintf(stderr, "%s", value ? "true" : "false");
   }
 
@@ -45,7 +47,8 @@ private:
 
  protected:
   template <class T>
-  void assertEqual(T expected, T actual) {
+  void assertEqual(T expected, T actual)
+  {
     if (expected != actual) {
       fprintf(stderr, "assertion failure, expected: ");
       print(expected);
@@ -57,16 +60,22 @@ protected:
     runs++;
   }
 
-  void assertEqual(const char* expected, const char* actual) {
-    if((expected == 0 && actual != 0) || (expected != 0 && actual == 0) || strcmp(expected, actual) != 0) {
-      fprintf(stderr, "assertion failure, expected: \"%s\", actual: \"%s\"\n", expected, actual);
+  void assertEqual(const char* expected, const char* actual)
+  {
+    if ((expected == 0 && actual != 0) || (expected != 0 && actual == 0)
+        || strcmp(expected, actual) != 0) {
+      fprintf(stderr,
+              "assertion failure, expected: \"%s\", actual: \"%s\"\n",
+              expected,
+              actual);
       failures++;
     }
     runs++;
   }
 
   template <class T>
-  void assertNotEqual(T expected, T actual) {
+  void assertNotEqual(T expected, T actual)
+  {
     if (expected == actual) {
       fprintf(stderr, "assertion failure, expected: not ");
       print(expected);
@@ -78,11 +87,13 @@ protected:
     runs++;
   }
 
-  void assertTrue(bool value) {
+  void assertTrue(bool value)
+  {
     assertEqual(true, value);
   }
 
-  void assertFalse(bool value) {
+  void assertFalse(bool value)
+  {
     assertEqual(false, value);
   }
 
@@ -95,4 +106,14 @@ public:
   static bool runAll();
 };
 
+#define TEST(name)                      \
+  class name##TestClass : public Test { \
+   public:                              \
+    name##TestClass() : Test(#name)     \
+    {                                   \
+    }                                   \
+    virtual void run();                 \
+  } name##TestInstance;                 \
+  void name##TestClass::run()
+
 #endif  // TEST_HARNESS_H

unittest/util/arg-parser-test.cpp

@@ -19,13 +19,7 @@
 
 using namespace avian::util;
 
-class ArgParserTest : public Test {
-public:
-  ArgParserTest():
-    Test("ArgParser")
-  {}
-
-  virtual void run() {
+TEST(ArgParser) {
   {
     ArgParser parser;
     Arg arg1(parser, false, "arg1", "<value>");
@@ -66,4 +60,3 @@ public:
     assertFalse(parser.parse(sizeof(args) / sizeof(char*) - 1, args));
   }
 }
-} argParserTest;