#include "test_utils.h"

UnittestTestResult_t* test_elseif_example(void) {
    UNITTEST_BEGIN_TEST("TestElseIf", "test_elseif_example");

    RAVA_TEST_FILE_EXECUTES(_unittest_result,
        "examples/18_ElseIf.java",
        "ElseIf", "main",
        "else-if example should execute successfully");

    UNITTEST_END_TEST();
}

UnittestTestResult_t* test_elseif_first_branch(void) {
    UNITTEST_BEGIN_TEST("TestElseIf", "test_elseif_first_branch");

    RAVA_TEST_RUN(_unittest_result,
        "public class Test {\n"
        "    public static int main() {\n"
        "        int x = 10;\n"
        "        if (x > 5) {\n"
        "            return 1;\n"
        "        } else if (x > 0) {\n"
        "            return 2;\n"
        "        } else {\n"
        "            return 3;\n"
        "        }\n"
        "    }\n"
        "}\n",
        "Test", "main", 1, "should take first branch when x > 5");

    UNITTEST_END_TEST();
}

UnittestTestResult_t* test_elseif_second_branch(void) {
    UNITTEST_BEGIN_TEST("TestElseIf", "test_elseif_second_branch");

    RAVA_TEST_RUN(_unittest_result,
        "public class Test {\n"
        "    public static int main() {\n"
        "        int x = 3;\n"
        "        if (x > 5) {\n"
        "            return 1;\n"
        "        } else if (x > 0) {\n"
        "            return 2;\n"
        "        } else {\n"
        "            return 3;\n"
        "        }\n"
        "    }\n"
        "}\n",
        "Test", "main", 2, "should take else-if branch when 0 < x <= 5");

    UNITTEST_END_TEST();
}

UnittestTestResult_t* test_elseif_else_branch(void) {
    UNITTEST_BEGIN_TEST("TestElseIf", "test_elseif_else_branch");

    RAVA_TEST_RUN(_unittest_result,
        "public class Test {\n"
        "    public static int main() {\n"
        "        int x = -5;\n"
        "        if (x > 5) {\n"
        "            return 1;\n"
        "        } else if (x > 0) {\n"
        "            return 2;\n"
        "        } else {\n"
        "            return 3;\n"
        "        }\n"
        "    }\n"
        "}\n",
        "Test", "main", 3, "should take else branch when x <= 0");

    UNITTEST_END_TEST();
}

UnittestTestResult_t* test_elseif_chain(void) {
    UNITTEST_BEGIN_TEST("TestElseIf", "test_elseif_chain");

    RAVA_TEST_RUN(_unittest_result,
        "public class Test {\n"
        "    public static int main() {\n"
        "        int x = 75;\n"
        "        if (x >= 90) {\n"
        "            return 1;\n"
        "        } else if (x >= 80) {\n"
        "            return 2;\n"
        "        } else if (x >= 70) {\n"
        "            return 3;\n"
        "        } else if (x >= 60) {\n"
        "            return 4;\n"
        "        } else {\n"
        "            return 5;\n"
        "        }\n"
        "    }\n"
        "}\n",
        "Test", "main", 3, "should take third branch for grade C");

    UNITTEST_END_TEST();
}

int main(int argc, char **argv) {
    UnittestConfig_t *config = unittest_config_create();
    config->verbosity = 2;

    if (argc > 1 && strcmp(argv[1], "--json") == 0) {
        config->output_format = UNITTEST_FORMAT_JSON;
        config->use_colors = false;
    }

    UnittestTestSuite_t *suite = unittest_test_suite_create("Else-If Tests");

    UnittestTestCase_t *tc = unittest_test_case_create("TestElseIf");
    unittest_test_case_add_result(tc, test_elseif_example());
    unittest_test_case_add_result(tc, test_elseif_first_branch());
    unittest_test_case_add_result(tc, test_elseif_second_branch());
    unittest_test_case_add_result(tc, test_elseif_else_branch());
    unittest_test_case_add_result(tc, test_elseif_chain());
    unittest_test_suite_add_test_case(suite, tc);

    unittest_generate_report(suite, config);

    int failures = suite->total_failed + suite->total_errors;
    unittest_test_suite_destroy(suite);
    unittest_config_destroy(config);

    return failures > 0 ? 1 : 0;
}