1 files changed, 317 insertions, 0 deletions

diff --git a/lib/tests/test_generic_output.c b/lib/tests/test_generic_output.c
new file mode 100644
index 00000000..e4a78bcd
--- /dev/null
+++ b/lib/tests/test_generic_output.c
@@ -0,0 +1,317 @@
+#include "../lib/output.h"
+#include "../../tap/tap.h"
+#include "./states.h"
+
+#include <string.h>
+
+void test_one_subcheck(void);
+void test_two_subchecks(void);
+
+void test_perfdata_formatting(void);
+void test_perfdata_formatting2(void);
+
+void test_deep_check_hierarchy(void);
+void test_deep_check_hierarchy2(void);
+
+void test_default_states1(void);
+void test_default_states2(void);
+
+int main(void) {
+        plan_tests(19);
+
+        diag("Simple test with one subcheck");
+        test_one_subcheck();
+
+        diag("Test with two subchecks");
+        test_two_subchecks();
+
+        diag("Test for performance data formatting");
+        test_perfdata_formatting();
+
+        diag("Another test for performance data formatting");
+        test_perfdata_formatting2();
+
+        diag("Test for deeper hierarchies");
+        test_deep_check_hierarchy();
+
+        diag("Another test for deeper hierarchies");
+        test_deep_check_hierarchy2();
+
+        diag("Testing the default state logic");
+        test_default_states1();
+
+        diag("Testing the default state logic #2");
+        test_default_states2();
+
+        return exit_status();
+}
+
+void test_one_subcheck(void) {
+        mp_subcheck sc1 = mp_subcheck_init();
+
+        sc1.output = "foobar";
+        sc1 = mp_set_subcheck_state(sc1, STATE_WARNING);
+
+        mp_check check = mp_check_init();
+        mp_add_subcheck_to_check(&check, sc1);
+
+        ok(mp_compute_check_state(check) == STATE_WARNING, "Main state should be warning");
+
+        char *output = mp_fmt_output(check);
+
+        // diag("Formatted output");
+        // diag(output);
+
+        char expected[] = "[WARNING] - ok=0, warning=1, critical=0, unknown=0\n"
+                                          "\t\\_[WARNING] - foobar\n";
+
+        // diag("Expected output");
+        // diag(expected);
+
+        ok(strcmp(output, expected) == 0, "Simple output test");
+}
+
+void test_perfdata_formatting2(void) {
+        mp_perfdata pd1 = perfdata_init();
+        mp_perfdata pd2 = perfdata_init();
+
+        pd1.label = "foo";
+        pd2.label = "bar";
+
+        pd1 = mp_set_pd_value(pd1, 23);
+        pd2 = mp_set_pd_value(pd2, 1LL);
+
+        pd_list *tmp = pd_list_init();
+
+        pd_list_append(tmp, pd1);
+        pd_list_append(tmp, pd2);
+
+        char *result = pd_list_to_string(*tmp);
+
+        ok(strcmp(result, "foo=23;;; bar=1;;;") == 0, "Perfdata string formatting");
+}
+
+void test_perfdata_formatting(void) {
+        mp_perfdata pd1 = perfdata_init();
+
+        pd1.uom = "s";
+        pd1.label = "foo";
+
+        pd1 = mp_set_pd_value(pd1, 23);
+
+        char *pd_string = pd_to_string(pd1);
+
+        ok(strcmp(pd_string, "foo=23s;;;") == 0, "Perfdata string formatting");
+}
+
+void test_two_subchecks(void) {
+        mp_subcheck sc1 = mp_subcheck_init();
+
+        sc1.output = "foobar";
+        sc1 = mp_set_subcheck_state(sc1, STATE_WARNING);
+
+        ok(mp_compute_subcheck_state(sc1) == STATE_WARNING,
+           "Test subcheck state directly after setting it");
+
+        mp_perfdata pd1 = perfdata_init();
+
+        pd1 = mp_set_pd_value(pd1, 23);
+
+        pd1.uom = "s";
+        pd1.label = "foo";
+
+        mp_add_perfdata_to_subcheck(&sc1, pd1);
+
+        mp_subcheck sc2 = mp_subcheck_init();
+        sc2.output = "baz";
+        sc2 = mp_set_subcheck_state(sc2, STATE_OK);
+
+        ok(mp_compute_subcheck_state(sc2) == STATE_OK, "Test subcheck 2 state after setting it");
+
+        mp_add_subcheck_to_subcheck(&sc1, sc2);
+
+        ok(mp_compute_subcheck_state(sc1) == STATE_WARNING,
+           "Test subcheck state after adding a subcheck");
+
+        mp_check check = mp_check_init();
+        mp_add_subcheck_to_check(&check, sc1);
+
+        ok(mp_compute_check_state(check) == STATE_WARNING, "Test main check result");
+
+        char *output = mp_fmt_output(check);
+
+        // diag("Formatted output. Length: %u", strlen(output));
+        // diag(output);
+
+        ok(output != NULL, "Output should not be NULL");
+
+        char expected[] = "[WARNING] - ok=0, warning=1, critical=0, unknown=0\n"
+                                          "\t\\_[WARNING] - foobar\n"
+                                          "\t\t\\_[OK] - baz\n"
+                                          "|foo=23s;;; \n";
+
+        // diag("Expected output. Length: %u", strlen(expected));
+        // diag(expected);
+
+        ok(strcmp(output, expected) == 0, "Output is as expected");
+}
+
+void test_deep_check_hierarchy(void) {
+        // level 4
+        mp_subcheck sc4 = mp_subcheck_init();
+        sc4.output = "level4";
+        sc4 = mp_set_subcheck_state(sc4, STATE_OK);
+
+        // level 3
+        mp_subcheck sc3 = mp_subcheck_init();
+        sc3.output = "level3";
+        sc3 = mp_set_subcheck_state(sc3, STATE_OK);
+
+        // level 2
+        mp_subcheck sc2 = mp_subcheck_init();
+        sc2.output = "baz";
+        sc2 = mp_set_subcheck_state(sc2, STATE_OK);
+
+        // level 1
+        mp_subcheck sc1 = mp_subcheck_init();
+
+        sc1.output = "foobar";
+        sc1 = mp_set_subcheck_state(sc1, STATE_WARNING);
+
+        mp_perfdata pd1 = perfdata_init();
+
+        pd1.uom = "s";
+        pd1.label = "foo";
+        pd1 = mp_set_pd_value(pd1, 23);
+
+        mp_add_perfdata_to_subcheck(&sc1, pd1);
+
+        // main check
+        mp_check check = mp_check_init();
+
+        mp_add_subcheck_to_subcheck(&sc3, sc4);
+        mp_add_subcheck_to_subcheck(&sc2, sc3);
+        mp_add_subcheck_to_subcheck(&sc1, sc2);
+        mp_add_subcheck_to_check(&check, sc1);
+
+        char *output = mp_fmt_output(check);
+
+        size_t output_length = strlen(output);
+
+        // diag("Formatted output of length %i", output_length);
+        // diag(output);
+
+        ok(output != NULL, "Output should not be NULL");
+
+        char expected[] = "[WARNING] - ok=0, warning=1, critical=0, unknown=0\n"
+                                          "\t\\_[WARNING] - foobar\n"
+                                          "\t\t\\_[OK] - baz\n"
+                                          "\t\t\t\\_[OK] - level3\n"
+                                          "\t\t\t\t\\_[OK] - level4\n"
+                                          "|foo=23s;;; \n";
+
+        size_t expected_length = strlen(expected);
+
+        // diag("Expected output of length: %i", expected_length);
+        // diag(expected);
+
+        ok(output_length == expected_length, "Outputs are of equal length");
+        ok(strcmp(output, expected) == 0, "Output is as expected");
+}
+
+void test_deep_check_hierarchy2(void) {
+        // level 1
+        mp_subcheck sc1 = mp_subcheck_init();
+
+        sc1.output = "foobar";
+        sc1 = mp_set_subcheck_state(sc1, STATE_WARNING);
+
+        mp_perfdata pd1 = perfdata_init();
+        pd1.uom = "s";
+        pd1.label = "foo";
+        pd1 = mp_set_pd_value(pd1, 23);
+
+        mp_add_perfdata_to_subcheck(&sc1, pd1);
+
+        // level 2
+        mp_subcheck sc2 = mp_subcheck_init();
+        sc2.output = "baz";
+        sc2 = mp_set_subcheck_state(sc2, STATE_OK);
+
+        mp_perfdata pd2 = perfdata_init();
+        pd2.uom = "B";
+        pd2.label = "baz";
+        pd2 = mp_set_pd_value(pd2, 1024);
+        mp_add_perfdata_to_subcheck(&sc2, pd2);
+
+        // level 3
+        mp_subcheck sc3 = mp_subcheck_init();
+        sc3.output = "level3";
+        sc3 = mp_set_subcheck_state(sc3, STATE_OK);
+
+        mp_perfdata pd3 = perfdata_init();
+        pd3.label = "floatMe";
+        pd3 = mp_set_pd_value(pd3, 1024.1024);
+        mp_add_perfdata_to_subcheck(&sc3, pd3);
+
+        // level 4
+        mp_subcheck sc4 = mp_subcheck_init();
+        sc4.output = "level4";
+        sc4 = mp_set_subcheck_state(sc4, STATE_OK);
+
+        mp_check check = mp_check_init();
+
+        mp_add_subcheck_to_subcheck(&sc3, sc4);
+        mp_add_subcheck_to_subcheck(&sc2, sc3);
+        mp_add_subcheck_to_subcheck(&sc1, sc2);
+        mp_add_subcheck_to_check(&check, sc1);
+
+        char *output = mp_fmt_output(check);
+
+        // diag("Formatted output of length: %i", strlen(output));
+        // diag(output);
+
+        ok(output != NULL, "Output should not be NULL");
+
+        char expected[] = "[WARNING] - ok=0, warning=1, critical=0, unknown=0\n"
+                                          "\t\\_[WARNING] - foobar\n"
+                                          "\t\t\\_[OK] - baz\n"
+                                          "\t\t\t\\_[OK] - level3\n"
+                                          "\t\t\t\t\\_[OK] - level4\n"
+                                          "|foo=23s;;; baz=1024B;;; floatMe=1024.102400;;; \n";
+
+        // diag("Expected output of length: %i", strlen(expected));
+        // diag(expected);
+
+        ok(strcmp(output, expected) == 0, "Output is as expected");
+}
+
+void test_default_states1(void) {
+        mp_subcheck sc = mp_subcheck_init();
+
+        mp_state_enum state1 = mp_compute_subcheck_state(sc);
+        ok(state1 == STATE_UNKNOWN, "Default default state is Unknown");
+
+        sc = mp_set_subcheck_default_state(sc, STATE_CRITICAL);
+
+        mp_state_enum state2 = mp_compute_subcheck_state(sc);
+        ok(state2 == STATE_CRITICAL, "Default state is Critical");
+
+        sc = mp_set_subcheck_state(sc, STATE_OK);
+
+        mp_state_enum state3 = mp_compute_subcheck_state(sc);
+        ok(state3 == STATE_OK, "Default state is Critical");
+}
+
+void test_default_states2(void) {
+        mp_check check = mp_check_init();
+
+        mp_subcheck sc = mp_subcheck_init();
+        sc.output = "placeholder";
+        sc = mp_set_subcheck_default_state(sc, STATE_CRITICAL);
+
+        mp_add_subcheck_to_check(&check, sc);
+
+        mp_state_enum result_state = mp_compute_check_state(check);
+        ok(result_state == STATE_CRITICAL, "Derived state is the proper default state");
+}