refactored single_test_context()

9unit.c

@@ -27,19 +27,12 @@
 
 // Internal Types
 
-typedef struct RunTestWith RunTestWith;
 typedef struct ContextData ContextData;
 typedef struct SingleTestContext SingleTestContext;
 typedef struct TestContextWith TestContextWith;
+typedef struct SingleTestContextWith SingleTestContextWith;
 typedef struct CheckValue CheckValue;
 
-struct RunTestWith
-{
-	void *ptr; // state's prevoius ptr value
-	TestResult (*test)(TestState *, void *); // the test
-	void *val; // the value being passed in
-};
-
 struct ContextData
 {
 	const char *old_c; // previous context
@@ -61,6 +54,13 @@ struct TestContextWith
 	void (*test)(TestState *, void *); // the test function
 };
 
+struct SingleTestContextWith
+{
+	void *ptr;
+	TestResult (*test)(TestState *, void *);
+	void *val;
+};
+
 struct CheckValue
 {
 	void *chk_val; // the value being checked
@@ -79,8 +79,9 @@ static TestResult run_test_with_test(TestState *);
 static void build_new_context(TestState *, ContextData *);
 static void display_context(TestState *);
 static void restore_context(TestState *, ContextData *);
-static void run_single_test_context(TestState *);
+static void single_test_context_test(TestState *);
 static void test_context_with_test(TestState *);
+static TestResult single_test_context_with_test(TestState *);
 static void check_value_test(TestState *, void *);
 
 // Public Functions
@@ -121,14 +122,7 @@ run_test_with(
 	void *val // the value being passed in
 )
 {
-	if (!s) return;
-	if (!test) run_test(s, 0);
-	RunTestWith d;
-	d.ptr = s->ptr;
-	d.test = test;
-	d.val = val;
-	s->ptr = &d;
-	run_test(s, run_test_with_test);
+	single_test_context_with(s, 0, test, val);
 }
 
 void
@@ -206,11 +200,11 @@ single_test_context(
 )
 {
 	if (!s) return;
-	SingleTestContext stc;
-	stc.ptr = s->ptr;
-	stc.test = test;
-	s->ptr = &stc;
-	test_context(s, label, run_single_test_context);
+	SingleTestContext d;
+	d.ptr = s->ptr;
+	d.test = test;
+	s->ptr = &d;
+	test_context(s, label, single_test_context_test);
 }
 
 void
@@ -230,6 +224,24 @@ test_context_with(
 	test_context(s, context, test_context_with_test);
 }
 
+void
+single_test_context_with(
+	TestState *s,
+	const char *context,
+	TestResult (*test)(TestState *, void *),
+	void *val
+)
+{
+	if (!s) return;
+	if (!test) single_test_context(s, context, 0);
+	SingleTestContextWith d;
+	d.ptr = s->ptr;
+	d.test = test;
+	d.val = val;
+	s->ptr = &d;
+	single_test_context(s, context, single_test_context_with_test);
+}
+
 void
 check_value(
 	TestState *s,
@@ -305,14 +317,6 @@ report(const char *str)
 	print("%s", str);
 }
 
-static TestResult
-run_test_with_test(TestState *s)
-{
-	RunTestWith *d = s->ptr;
-	s->ptr = d->ptr;
-	return (*d->test)(s, d->val);
-}
-
 static void
 build_new_context(TestState *s, ContextData *cd)
 {
@@ -352,11 +356,11 @@ restore_context(TestState *s, ContextData *cd)
 }
 
 static void
-run_single_test_context(TestState *s)
+single_test_context_test(TestState *s)
 {
-	SingleTestContext *stc = s->ptr;
-	s->ptr = stc->ptr;
-	run_test(s, stc->test);
+	SingleTestContext *d = s->ptr;
+	s->ptr = d->ptr;
+	run_test(s, d->test);
 }
 
 static void
@@ -367,6 +371,14 @@ test_context_with_test(TestState *s)
 	(*d->test)(s, d->val);
 }
 
+static TestResult
+single_test_context_with_test(TestState *s)
+{
+	SingleTestContextWith *d = s->ptr;
+	s->ptr = d->ptr;
+	return (*d->test)(s, d->val);
+}
+
 static void
 check_value_test(TestState *s, void *ptr)
 {

9unit.h

@@ -109,6 +109,20 @@ extern void test_context_with(
 	void * // the value being passed in
 );
 
+// Runs a single test with a context and input
+extern void single_test_context_with(
+	TestState *, // the state
+	const char *, // the context
+
+	// the test
+	TestResult (*)(
+		TestState *, // the state
+		void * // the value being passed
+	),
+
+	void * // the value being passed
+);
+
 // Runs a test to check a value
 extern void check_value(
 	TestState *, // the current test state