blogc: test "inject" functions

1 files changed, 23 insertions, 0 deletions

diff --git a/tests/blogc/check_rusage.c b/tests/blogc/check_rusage.c
index e828fb2..ff64297 100644
--- a/tests/blogc/check_rusage.c
+++ b/tests/blogc/check_rusage.c
@@ -65,6 +65,10 @@ test_rusage_format_cpu_time(void **state)
     assert_string_equal(f, "1.234ms");
     free(f);
 
+    f = blogc_rusage_format_cpu_time(3000);
+    assert_string_equal(f, "3ms");
+    free(f);
+
     f = blogc_rusage_format_cpu_time(12345678);
     assert_string_equal(f, "12.346s");
     free(f);
@@ -92,6 +96,24 @@ test_rusage_format_memory(void **state)
 }
 
 
+static void
+test_rusage_inject(void **state)
+{
+    bc_trie_t *t = bc_trie_new(free);
+
+    will_return(__wrap_getrusage, -1);
+    blogc_rusage_inject(t);
+    assert_int_equal(bc_trie_size(t), 0);
+
+    will_return(__wrap_getrusage, 0);
+    blogc_rusage_inject(t);
+    assert_int_equal(bc_trie_size(t), 2);
+    assert_string_equal(bc_trie_lookup(t, "BLOGC_RUSAGE_CPU_TIME"), "4.000s");
+    assert_string_equal(bc_trie_lookup(t, "BLOGC_RUSAGE_MEMORY"), "10.010MB");
+    bc_trie_free(t);
+}
+
+
 int
 main(void)
 {
@@ -99,6 +121,7 @@ main(void)
         unit_test(test_rusage_get),
         unit_test(test_rusage_format_cpu_time),
         unit_test(test_rusage_format_memory),
+        unit_test(test_rusage_inject),
     };
     return run_tests(tests);
 }