From 316ba45e3c9fa72ff781011a770748fc53f20ec6 Mon Sep 17 00:00:00 2001
From: Viktor Rennert <vrennert@squarepx.com>
Date: Sat, 2 Jan 2016 20:54:55 +0100
Subject: [PATCH] Added unittest to cover utility::add_class<Enum>
 registration.

---
 unittests/compiled_tests.cpp | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/unittests/compiled_tests.cpp b/unittests/compiled_tests.cpp
index da7eb7c..90095ac 100644
--- a/unittests/compiled_tests.cpp
+++ b/unittests/compiled_tests.cpp
@@ -517,6 +517,41 @@ TEST_CASE("Utility_Test utility class wrapper")
 
 }
 
+enum Utility_Test_Numbers
+{
+	ONE,
+  TWO,
+  THREE
+};
+
+TEST_CASE("Utility_Test utility class wrapper")
+{
+
+  chaiscript::ModulePtr m = chaiscript::ModulePtr(new chaiscript::Module());
+
+  using namespace chaiscript;
+
+  chaiscript::utility::add_class<Utility_Test_Numbers>(*m,
+      "Utility_Test_Numbers",
+      {
+      },
+      { { const_var(ONE), "ONE" },
+        { const_var(TWO), "TWO" },
+        { const_var(THREE), "THREE" }
+ 
+        }
+      );
+
+
+  chaiscript::ChaiScript chai;
+  chai.add(m);
+
+  CHECK(chai.eval<Utility_Test_Numbers>("ONE ") == 0);
+  CHECK(chai.eval<Utility_Test_Numbers>("TWO ") == 1);
+  CHECK(chai.eval<Utility_Test_Numbers>("THREE ") == 2);
+
+}
+
 
 ////// Object copy count test