#include "test_utils.h"

UnittestTestResult_t* test_forloop_example(void) {
    UNITTEST_BEGIN_TEST("TestForLoop", "test_forloop_example");

    RAVA_TEST_FILE_EXECUTES(_unittest_result,
        "examples/16_ForLoop.java",
        "ForLoop", "main",
        "for loop example should execute successfully");

    UNITTEST_END_TEST();
}

UnittestTestResult_t* test_forloop_sum(void) {
    UNITTEST_BEGIN_TEST("TestForLoop", "test_forloop_sum");

    RAVA_TEST_RUN(_unittest_result,
        "public class Test {\n"
        "    public static int main() {\n"
        "        int sum = 0;\n"
        "        for (int i = 1; i <= 10; i++) {\n"
        "            sum = sum + i;\n"
        "        }\n"
        "        return sum;\n"
        "    }\n"
        "}\n",
        "Test", "main", 55, "sum 1 to 10 should equal 55");

    UNITTEST_END_TEST();
}

UnittestTestResult_t* test_forloop_countdown(void) {
    UNITTEST_BEGIN_TEST("TestForLoop", "test_forloop_countdown");

    RAVA_TEST_RUN(_unittest_result,
        "public class Test {\n"
        "    public static int main() {\n"
        "        int count = 0;\n"
        "        for (int i = 10; i > 0; i--) {\n"
        "            count = count + 1;\n"
        "        }\n"
        "        return count;\n"
        "    }\n"
        "}\n",
        "Test", "main", 10, "countdown from 10 should iterate 10 times");

    UNITTEST_END_TEST();
}

UnittestTestResult_t* test_forloop_nested(void) {
    UNITTEST_BEGIN_TEST("TestForLoop", "test_forloop_nested");

    RAVA_TEST_RUN(_unittest_result,
        "public class Test {\n"
        "    public static int main() {\n"
        "        int count = 0;\n"
        "        for (int i = 0; i < 3; i++) {\n"
        "            for (int j = 0; j < 4; j++) {\n"
        "                count = count + 1;\n"
        "            }\n"
        "        }\n"
        "        return count;\n"
        "    }\n"
        "}\n",
        "Test", "main", 12, "3x4 nested loops should count 12");

    UNITTEST_END_TEST();
}

UnittestTestResult_t* test_forloop_step_by_two(void) {
    UNITTEST_BEGIN_TEST("TestForLoop", "test_forloop_step_by_two");

    RAVA_TEST_RUN(_unittest_result,
        "public class Test {\n"
        "    public static int main() {\n"
        "        int sum = 0;\n"
        "        for (int i = 0; i < 10; i = i + 2) {\n"
        "            sum = sum + i;\n"
        "        }\n"
        "        return sum;\n"
        "    }\n"
        "}\n",
        "Test", "main", 20, "sum even numbers 0-8 should equal 20");

    UNITTEST_END_TEST();
}

UnittestTestResult_t* test_forloop_zero_iterations(void) {
    UNITTEST_BEGIN_TEST("TestForLoop", "test_forloop_zero_iterations");

    RAVA_TEST_RUN(_unittest_result,
        "public class Test {\n"
        "    public static int main() {\n"
        "        int count = 0;\n"
        "        for (int i = 0; i < 0; i++) {\n"
        "            count = count + 1;\n"
        "        }\n"
        "        return count;\n"
        "    }\n"
        "}\n",
        "Test", "main", 0, "loop with false condition should not execute");

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

    UnittestTestCase_t *tc = unittest_test_case_create("TestForLoop");
    unittest_test_case_add_result(tc, test_forloop_example());
    unittest_test_case_add_result(tc, test_forloop_sum());
    unittest_test_case_add_result(tc, test_forloop_countdown());
    unittest_test_case_add_result(tc, test_forloop_nested());
    unittest_test_case_add_result(tc, test_forloop_step_by_two());
    unittest_test_case_add_result(tc, test_forloop_zero_iterations());
    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;
}