#include "test_utils.h"

UnittestTestResult_t* test_simple_array_init(void) {
    UNITTEST_BEGIN_TEST("TestArrayInit", "test_simple_array_init");

    RAVA_TEST_RUN(_unittest_result,
        "public class Test {\n"
        "    public static int main() {\n"
        "        int[] arr = {1, 2, 3};\n"
        "        return arr[0] + arr[1] + arr[2];\n"
        "    }\n"
        "}\n",
        "Test", "main", 6, "simple array init sum should return 6");

    UNITTEST_END_TEST();
}

UnittestTestResult_t* test_array_init_with_expressions(void) {
    UNITTEST_BEGIN_TEST("TestArrayInit", "test_array_init_with_expressions");

    RAVA_TEST_RUN(_unittest_result,
        "public class Test {\n"
        "    public static int main() {\n"
        "        int[] arr = {1 + 1, 2 * 2, 3 + 3};\n"
        "        return arr[0] + arr[1] + arr[2];\n"
        "    }\n"
        "}\n",
        "Test", "main", 12, "array init with expressions sum should return 12");

    UNITTEST_END_TEST();
}

UnittestTestResult_t* test_single_element_array(void) {
    UNITTEST_BEGIN_TEST("TestArrayInit", "test_single_element_array");

    RAVA_TEST_RUN(_unittest_result,
        "public class Test {\n"
        "    public static int main() {\n"
        "        int[] arr = {42};\n"
        "        return arr[0];\n"
        "    }\n"
        "}\n",
        "Test", "main", 42, "single element array should return 42");

    UNITTEST_END_TEST();
}

UnittestTestResult_t* test_array_init_length(void) {
    UNITTEST_BEGIN_TEST("TestArrayInit", "test_array_init_length");

    RAVA_TEST_RUN(_unittest_result,
        "public class Test {\n"
        "    public static int main() {\n"
        "        int[] arr = {5, 10, 15, 20};\n"
        "        int sum = 0;\n"
        "        for (int i = 0; i < 4; i = i + 1) {\n"
        "            sum = sum + arr[i];\n"
        "        }\n"
        "        return sum;\n"
        "    }\n"
        "}\n",
        "Test", "main", 50, "array init length loop sum should return 50");

    UNITTEST_END_TEST();
}

UnittestTestResult_t* test_array_init_negative_values(void) {
    UNITTEST_BEGIN_TEST("TestArrayInit", "test_array_init_negative_values");

    RAVA_TEST_RUN(_unittest_result,
        "public class Test {\n"
        "    public static int main() {\n"
        "        int[] arr = {-1, -2, -3};\n"
        "        return arr[0] + arr[1] + arr[2];\n"
        "    }\n"
        "}\n",
        "Test", "main", -6, "array init with negative values sum should return -6");

    UNITTEST_END_TEST();
}

UnittestTestResult_t* test_array_init_with_zero(void) {
    UNITTEST_BEGIN_TEST("TestArrayInit", "test_array_init_with_zero");

    RAVA_TEST_RUN(_unittest_result,
        "public class Test {\n"
        "    public static int main() {\n"
        "        int[] arr = {0, 0, 0};\n"
        "        return arr[0] + arr[1] + arr[2];\n"
        "    }\n"
        "}\n",
        "Test", "main", 0, "array init with zeros sum should return 0");

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

    UnittestTestCase_t *tc = unittest_test_case_create("TestArrayInit");
    unittest_test_case_add_result(tc, test_simple_array_init());
    unittest_test_case_add_result(tc, test_array_init_with_expressions());
    unittest_test_case_add_result(tc, test_single_element_array());
    unittest_test_case_add_result(tc, test_array_init_length());
    unittest_test_case_add_result(tc, test_array_init_negative_values());
    unittest_test_case_add_result(tc, test_array_init_with_zero());
    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;
}