From 7da79cf1e24fbc58e0ff2e498a7f3d189a51cbd6 Mon Sep 17 00:00:00 2001
From: Tavian Barnes <tavianator@tavianator.com>
Date: Tue, 18 Dec 2012 16:54:23 -0500
Subject: Improve the unit test macro API a bit.

---
 libdimension/tests/unit-test.c | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

(limited to 'libdimension/tests/unit-test.c')

diff --git a/libdimension/tests/unit-test.c b/libdimension/tests/unit-test.c
index a15c960..d6d256e 100644
--- a/libdimension/tests/unit-test.c
+++ b/libdimension/tests/unit-test.c
@@ -26,6 +26,24 @@
 
 dmnsn_dictionary *dmnsn_test_cases = NULL;
 
+TCase *
+dmnsn_get_test_case(const char* name)
+{
+  if (!dmnsn_test_cases) {
+    dmnsn_test_cases = dmnsn_new_dictionary(sizeof(TCase *));
+  }
+
+  TCase **tcp = dmnsn_dictionary_at(dmnsn_test_cases, name);
+  if (tcp) {
+    return *tcp;
+  }
+
+  TCase *tc = tcase_create(name);
+  tcase_add_checked_fixture(tc, dmnsn_test_setup, dmnsn_test_teardown);
+  dmnsn_dictionary_insert(dmnsn_test_cases, name, &tc);
+  return tc;
+}
+
 void
 dmnsn_test_setup(void)
 {
@@ -53,7 +71,7 @@ dmnsn_test_suite()
 {
   dmnsn_suite = suite_create("Dimension");
 
-  if (dmnsn_test_cases != NULL) {
+  if (dmnsn_test_cases) {
     dmnsn_dictionary_apply(dmnsn_test_cases, dmnsn_add_test_cases);
   }
 
-- 
cgit v1.2.3