#include <cstdio>
#include <cstdlib>
#include <cstring>
#include "mujson.h"

static int main_ret = 0;
static int test_count = 0;
static int test_pass = 0;

#define EXPECT_EQ_BASE(equality, expect, actual, format) \
    do {\
        test_count++;\
        if (equality)\
            test_pass++;\
        else {\
            fprintf(stderr, "%s:%d: expect: " format " actual: " format "\n", __FILE__, __LINE__, expect, actual);\
            main_ret = 1;\
        }\
    } while(0)

#define EXPECT_EQ_INT(expect, actual) EXPECT_EQ_BASE((expect) == (actual), expect, actual, "%d")
#define EXPECT_EQ_DOUBLE(expect, actual) EXPECT_EQ_BASE((expect) == (actual), expect, actual, "%.17g")
#define EXPECT_EQ_STRING(expect, actual, alength) \
    EXPECT_EQ_BASE(sizeof(expect) - 1 == alength && memcmp(expect, actual, alength) == 0, expect, actual, "%s")
#define EXPECT_TRUE(actual) EXPECT_EQ_BASE((actual) != 0, "true", "false", "%s")
#define EXPECT_FALSE(actual) EXPECT_EQ_BASE((actual) == 0, "false", "true", "%s")


using namespace mujson;

static void test_parse_null() {
    JsonValue v;
    valueInit(&v);
    setBoolean(&v, 0); //v.type = TYPE::TYPE_FALSE;
    EXPECT_EQ_INT(STATE::PARSE_OK, parse(&v, "null"));
    EXPECT_EQ_INT(TYPE::TYPE_NULL, getType(&v));
    valueFree(&v);
}

static void test_parse_true() {
    JsonValue v;
    valueInit(&v);
    setBoolean(&v, 0);//v.type = TYPE::TYPE_FALSE;
    EXPECT_EQ_INT(STATE::PARSE_OK, parse(&v, "true"));
    EXPECT_EQ_INT(TYPE::TYPE_TRUE, getType(&v));
    valueFree(&v);
}

static void test_parse_false() {
    JsonValue v;
    valueInit(&v);
    setBoolean(&v, 0);//v.type = TYPE::TYPE_TRUE;
    EXPECT_EQ_INT(STATE::PARSE_OK, parse(&v, "false"));
    EXPECT_EQ_INT(TYPE::TYPE_FALSE, getType(&v));
    valueFree(&v);
}


#define TEST_NUMBER(expect, json)\
    do {\
        JsonValue v;\
        valueInit(&v);\
        EXPECT_EQ_INT(STATE::PARSE_OK, parse(&v, json));\
        EXPECT_EQ_INT(TYPE::TYPE_NUMBER, getType(&v));\
        EXPECT_EQ_DOUBLE(expect, getNumber(&v));\
        valueFree(&v);\
    } while(0)

static void test_parse_number() {
    TEST_NUMBER(0.0, "0");
    TEST_NUMBER(0.0, "-0");
    TEST_NUMBER(0.0, "-0.0");
    TEST_NUMBER(1.0, "1");
    TEST_NUMBER(-1.0, "-1");
    TEST_NUMBER(1.5, "1.5");
    TEST_NUMBER(-1.5, "-1.5");
    TEST_NUMBER(3.1416, "3.1416");
    TEST_NUMBER(1E10, "1E10");
    TEST_NUMBER(1e10, "1e10");
    TEST_NUMBER(1E+10, "1E+10");
    TEST_NUMBER(1E-10, "1E-10");
    TEST_NUMBER(-1E10, "-1E10");
    TEST_NUMBER(-1e10, "-1e10");
    TEST_NUMBER(-1E+10, "-1E+10");
    TEST_NUMBER(-1E-10, "-1E-10");
    TEST_NUMBER(1.234E+10, "1.234E+10");
    TEST_NUMBER(1.234E-10, "1.234E-10");
    TEST_NUMBER(0.0, "1e-10000"); /* must underflow */

    TEST_NUMBER(1.0000000000000002, "1.0000000000000002"); /* the smallest number > 1 */
    TEST_NUMBER( 4.9406564584124654e-324, "4.9406564584124654e-324"); /* minimum denormal */
    TEST_NUMBER(-4.9406564584124654e-324, "-4.9406564584124654e-324");
    TEST_NUMBER( 2.2250738585072009e-308, "2.2250738585072009e-308");  /* Max subnormal double */
    TEST_NUMBER(-2.2250738585072009e-308, "-2.2250738585072009e-308");
    TEST_NUMBER( 2.2250738585072014e-308, "2.2250738585072014e-308");  /* Min normal positive double */
    TEST_NUMBER(-2.2250738585072014e-308, "-2.2250738585072014e-308");
    TEST_NUMBER( 1.7976931348623157e+308, "1.7976931348623157e+308");  /* Max double */
    TEST_NUMBER(-1.7976931348623157e+308, "-1.7976931348623157e+308");
}

#define TEST_STRING(expect, json)\
    do {\
        JsonValue v;\
        valueInit(&v);\
        EXPECT_EQ_INT(STATE::PARSE_OK, parse(&v, json));\
        EXPECT_EQ_INT(TYPE::TYPE_STRING, getType(&v));\
        EXPECT_EQ_STRING(expect, getString(&v), getStringLength(&v));\
        valueFree(&v);\
    } while(0)

static void test_parse_string() {
    TEST_STRING("", "\"\"");
    TEST_STRING("Hello", "\"Hello\"");
    TEST_STRING("Hello\nWorld", "\"Hello\\nWorld\"");
    TEST_STRING("\" \\ / \b \f \n \r \t", "\"\\\" \\\\ \\/ \\b \\f \\n \\r \\t\"");
    TEST_STRING("Hello\0World", "\"Hello\\u0000World\"");
    TEST_STRING("\x24", "\"\\u0024\"");         /* Dollar sign U+0024 */
    TEST_STRING("\xC2\xA2", "\"\\u00A2\"");     /* Cents sign U+00A2 */
    TEST_STRING("\xE2\x82\xAC", "\"\\u20AC\""); /* Euro sign U+20AC */
    TEST_STRING("\xF0\x9D\x84\x9E", "\"\\uD834\\uDD1E\"");  /* G clef sign U+1D11E */
    TEST_STRING("\xF0\x9D\x84\x9E", "\"\\ud834\\udd1e\"");  /* G clef sign U+1D11E */
}

#define TEST_ERROR(error, json)\
    do {\
        JsonValue v;\
        valueInit(&v);\
        v.type = TYPE::TYPE_FALSE;\
        EXPECT_EQ_INT(error, parse(&v, json));\
        EXPECT_EQ_INT(TYPE::TYPE_NULL, getType(&v));\
        valueFree(&v);\
    } while(0)

static void test_parse_expect_value() {
    TEST_ERROR(STATE::PARSE_EXPECT_VALUE, "");
    TEST_ERROR(STATE::PARSE_EXPECT_VALUE, " ");
}

static void test_parse_invalid_value() {
    TEST_ERROR(STATE::PARSE_INVALID_VALUE, "nul");
    TEST_ERROR(STATE::PARSE_INVALID_VALUE, "?");


#if 1
    /* invalid number */
    TEST_ERROR(STATE::PARSE_INVALID_VALUE, "+0");
    TEST_ERROR(STATE::PARSE_INVALID_VALUE, "+1");
    TEST_ERROR(STATE::PARSE_INVALID_VALUE, ".123"); /* at least one digit before '.' */
    TEST_ERROR(STATE::PARSE_INVALID_VALUE, "1.");   /* at least one digit after '.' */
    TEST_ERROR(STATE::PARSE_INVALID_VALUE, "INF");
    TEST_ERROR(STATE::PARSE_INVALID_VALUE, "inf");
    TEST_ERROR(STATE::PARSE_INVALID_VALUE, "NAN");
    TEST_ERROR(STATE::PARSE_INVALID_VALUE, "nan");
#endif
}

static void test_parse_root_not_singular() {
    TEST_ERROR(STATE::PARSE_ROOT_NOT_SINGULAR, "null x");

#if 1
    /* invalid number */
    TEST_ERROR(STATE::PARSE_ROOT_NOT_SINGULAR, "0123"); /* after zero should be '.' , 'E' , 'e' or nothing */
    TEST_ERROR(STATE::PARSE_ROOT_NOT_SINGULAR, "0x0");
    TEST_ERROR(STATE::PARSE_ROOT_NOT_SINGULAR, "0x123");
#endif
}

static void test_parse_number_too_big() {
#if 1
    TEST_ERROR(STATE::PARSE_NUMBER_TOO_BIG, "1e309");
    TEST_ERROR(STATE::PARSE_NUMBER_TOO_BIG, "-1e309");
#endif
}

static void test_parse_missing_quotation_mark() {
    TEST_ERROR(STATE::PARSE_MISS_QUOTATION_MARK, "\"");
    TEST_ERROR(STATE::PARSE_MISS_QUOTATION_MARK, "\"abc");
}

static void test_parse_invalid_string_escape() {
#if 1
    TEST_ERROR(STATE::PARSE_INVALID_STRING_ESCAPE, "\"\\v\"");
    TEST_ERROR(STATE::PARSE_INVALID_STRING_ESCAPE, "\"\\'\"");
    TEST_ERROR(STATE::PARSE_INVALID_STRING_ESCAPE, "\"\\0\"");
    TEST_ERROR(STATE::PARSE_INVALID_STRING_ESCAPE, "\"\\x12\"");
#endif
}

static void test_parse_invalid_string_char() {
#if 1
    TEST_ERROR(STATE::PARSE_INVALID_STRING_CHAR, "\"\x01\"");
    TEST_ERROR(STATE::PARSE_INVALID_STRING_CHAR, "\"\x1F\"");
#endif
}
static void test_parse_invalid_unicode_hex() {
    TEST_ERROR(STATE::PARSE_INVALID_UNICODE_HEX, "\"\\u\"");
    TEST_ERROR(STATE::PARSE_INVALID_UNICODE_HEX, "\"\\u0\"");
    TEST_ERROR(STATE::PARSE_INVALID_UNICODE_HEX, "\"\\u01\"");
    TEST_ERROR(STATE::PARSE_INVALID_UNICODE_HEX, "\"\\u012\"");
    TEST_ERROR(STATE::PARSE_INVALID_UNICODE_HEX, "\"\\u/000\"");
    TEST_ERROR(STATE::PARSE_INVALID_UNICODE_HEX, "\"\\uG000\"");
    TEST_ERROR(STATE::PARSE_INVALID_UNICODE_HEX, "\"\\u0/00\"");
    TEST_ERROR(STATE::PARSE_INVALID_UNICODE_HEX, "\"\\u0G00\"");
    TEST_ERROR(STATE::PARSE_INVALID_UNICODE_HEX, "\"\\u00/0\"");
    TEST_ERROR(STATE::PARSE_INVALID_UNICODE_HEX, "\"\\u00G0\"");
    TEST_ERROR(STATE::PARSE_INVALID_UNICODE_HEX, "\"\\u000/\"");
    TEST_ERROR(STATE::PARSE_INVALID_UNICODE_HEX, "\"\\u000G\"");
    TEST_ERROR(STATE::PARSE_INVALID_UNICODE_HEX, "\"\\u 123\"");
}

static void test_parse_invalid_unicode_surrogate() {
    TEST_ERROR(STATE::PARSE_INVALID_UNICODE_SURROGATE, "\"\\uD800\"");
    TEST_ERROR(STATE::PARSE_INVALID_UNICODE_SURROGATE, "\"\\uDBFF\"");
    TEST_ERROR(STATE::PARSE_INVALID_UNICODE_SURROGATE, "\"\\uD800\\\\\"");
    TEST_ERROR(STATE::PARSE_INVALID_UNICODE_SURROGATE, "\"\\uD800\\uDBFF\"");
    TEST_ERROR(STATE::PARSE_INVALID_UNICODE_SURROGATE, "\"\\uD800\\uE000\"");
}

static void test_access_null() {
    JsonValue v;
    valueInit(&v);
    setString(&v, "a", 1);
    setNull(&v);
    EXPECT_EQ_INT(TYPE::TYPE_NULL, getType(&v));
    valueFree(&v);
}

static void test_parse() {
    test_parse_null();
    test_parse_true();
    test_parse_false();
    test_parse_number();
    test_parse_string();
    test_parse_expect_value();
    test_parse_invalid_value();
    test_parse_root_not_singular();
    test_parse_number_too_big();
    test_parse_missing_quotation_mark();
    test_parse_invalid_string_escape();
    test_parse_invalid_string_char();
    test_parse_invalid_unicode_hex();
    test_parse_invalid_unicode_surrogate();
}

static void test_access_boolean() {
    /* Use EXPECT_TRUE() and EXPECT_FALSE() */
    JsonValue v;
    valueInit(&v);
    setString(&v, "a", 1);
    setBoolean(&v, 1);
    EXPECT_TRUE(getBoolean(&v));
    setBoolean(&v, 0);
    EXPECT_FALSE(getBoolean(&v));
    valueFree(&v);
}

static void test_access_number() {
    JsonValue v;
    valueInit(&v);
    setString(&v, "a", 1);
    setNumber(&v, 1234.5);
    EXPECT_EQ_DOUBLE(1234.5, getNumber(&v));
    valueFree(&v);
}

static void test_access_string() {
    JsonValue v;
    valueInit(&v);
    setString(&v, "", 0);
    EXPECT_EQ_STRING("", getString(&v), getStringLength(&v));
    setString(&v, "Hello", 5);
    EXPECT_EQ_STRING("Hello", getString(&v), getStringLength(&v));
    valueFree(&v);
}

static void test_access() {
    test_access_null();
    test_access_boolean();
    test_access_number();
    test_access_string();
}



int main() {
    test_parse();
    printf("%d/%d (%3.2f%%) passed\n", test_pass, test_count, test_pass * 100.0 / test_count);
    return main_ret;
}