#include "test_utils.h"

UnittestTestResult_t* test_simple_enum_ordinal(void) {
    UNITTEST_BEGIN_TEST("TestEnums", "test_simple_enum_ordinal");

    RAVA_TEST_RUN(_unittest_result,
        "public enum Color { RED, GREEN, BLUE }\n"
        "public class Test {\n"
        "    public static int main() {\n"
        "        return Color.RED;\n"
        "    }\n"
        "}\n",
        "Test", "main", 0, "simple enum ordinal RED should return 0");

    UNITTEST_END_TEST();
}

UnittestTestResult_t* test_enum_second_value(void) {
    UNITTEST_BEGIN_TEST("TestEnums", "test_enum_second_value");

    RAVA_TEST_RUN(_unittest_result,
        "public enum Color { RED, GREEN, BLUE }\n"
        "public class Test {\n"
        "    public static int main() {\n"
        "        return Color.GREEN;\n"
        "    }\n"
        "}\n",
        "Test", "main", 1, "enum second value GREEN should return 1");

    UNITTEST_END_TEST();
}

UnittestTestResult_t* test_enum_third_value(void) {
    UNITTEST_BEGIN_TEST("TestEnums", "test_enum_third_value");

    RAVA_TEST_RUN(_unittest_result,
        "public enum Color { RED, GREEN, BLUE }\n"
        "public class Test {\n"
        "    public static int main() {\n"
        "        return Color.BLUE;\n"
        "    }\n"
        "}\n",
        "Test", "main", 2, "enum third value BLUE should return 2");

    UNITTEST_END_TEST();
}

UnittestTestResult_t* test_enum_comparison(void) {
    UNITTEST_BEGIN_TEST("TestEnums", "test_enum_comparison");

    RAVA_TEST_RUN(_unittest_result,
        "public enum Day { MON, TUE, WED, THU, FRI }\n"
        "public class Test {\n"
        "    public static int main() {\n"
        "        int day = Day.WED;\n"
        "        if (day == 2) {\n"
        "            return 42;\n"
        "        }\n"
        "        return 0;\n"
        "    }\n"
        "}\n",
        "Test", "main", 42, "enum comparison should return 42");

    UNITTEST_END_TEST();
}

UnittestTestResult_t* test_enum_in_switch(void) {
    UNITTEST_BEGIN_TEST("TestEnums", "test_enum_in_switch");

    RAVA_TEST_RUN(_unittest_result,
        "public enum Status { OK, ERROR, PENDING }\n"
        "public class Test {\n"
        "    public static int main() {\n"
        "        int s = Status.ERROR;\n"
        "        switch (s) {\n"
        "            case 0: return 10;\n"
        "            case 1: return 20;\n"
        "            case 2: return 30;\n"
        "            default: return 0;\n"
        "        }\n"
        "    }\n"
        "}\n",
        "Test", "main", 20, "enum in switch should return 20");

    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("Enum Tests");

    UnittestTestCase_t *tc = unittest_test_case_create("TestEnums");
    unittest_test_case_add_result(tc, test_simple_enum_ordinal());
    unittest_test_case_add_result(tc, test_enum_second_value());
    unittest_test_case_add_result(tc, test_enum_third_value());
    unittest_test_case_add_result(tc, test_enum_comparison());
    unittest_test_case_add_result(tc, test_enum_in_switch());
    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;
}