import x.json2 as json

enum Bar {
    a
    b  @[json: 'BBB']
    c = 10
    e  @[badattr: 'bad'; foobar]
}

type BarAlias = Bar

fn test_number_decode() {
    assert json.decode[Bar]('0')! == Bar.a
    assert json.decode[Bar]('1')! == Bar.b
    assert json.decode[Bar]('10')! == Bar.c

    assert json.decode[BarAlias]('0')! == Bar.a
    assert json.decode[BarAlias]('1')! == Bar.b
    assert json.decode[BarAlias]('10')! == Bar.c
}

fn test_number_decode_fails() {
    if _ := json.decode[Bar]('2') {
        assert false
    } else {
        if err is json.JsonDecodeError {
            assert err.line == 1
            assert err.character == 1
            assert err.message == 'Data: Number value: `2` does not match any field in enum: &Bar'
        }
    }

    if _ := json.decode[BarAlias]('2') {
        assert false
    } else {
        if err is json.JsonDecodeError {
            assert err.line == 1
            assert err.character == 1
            assert err.message == 'Data: Number value: `2` does not match any field in enum: &BarAlias'
        }
    }
}

fn test_string_decode() {
    assert json.decode[Bar]('"a"')! == Bar.a
    assert json.decode[Bar]('"BBB"')! == Bar.b
    assert json.decode[Bar]('"c"')! == Bar.c

    assert json.decode[BarAlias]('"a"')! == Bar.a
    assert json.decode[BarAlias]('"BBB"')! == Bar.b
    assert json.decode[BarAlias]('"c"')! == Bar.c
}

fn test_string_decode_uses_only_json_attr() {
    assert json.decode[Bar]('"e"')! == Bar.e
    assert json.decode[BarAlias]('"e"')! == Bar.e
    for badval in ['"badattr: bad"', '"bad"', '"foobar"'] {
        if _ := json.decode[Bar](badval) {
            assert false, '${badval} must not be decoded to Bar'
        } else {
            assert true
        }
        if _ := json.decode[BarAlias](badval) {
            assert false, '${badval} must not be decoded to BarAlias'
        } else {
            assert true
        }
    }
}

fn test_string_decode_fails() {
    if _ := json.decode[Bar]('"d"') {
        assert false
    } else {
        if err is json.JsonDecodeError {
            assert err.line == 1
            assert err.character == 1
            assert err.message == 'Data: String value: `d` does not match any field in enum: &Bar'
        }
    }

    if _ := json.decode[BarAlias]('"d"') {
        assert false
    } else {
        if err is json.JsonDecodeError {
            assert err.line == 1
            assert err.character == 1
            assert err.message == 'Data: String value: `d` does not match any field in enum: &BarAlias'
        }
    }
}

fn test_invalid_decode_fails() {
    if _ := json.decode[Bar]('true') {
        assert false
    } else {
        if err is json.JsonDecodeError {
            assert err.line == 1
            assert err.character == 1
            assert err.message == 'Data: Expected number or string value for enum, got: boolean'
        }
    }

    if _ := json.decode[BarAlias]('true') {
        assert false
    } else {
        if err is json.JsonDecodeError {
            assert err.line == 1
            assert err.character == 1
            assert err.message == 'Data: Expected number or string value for enum, got: boolean'
        }
    }
}

fn test_map_string_enum_decode() {
    m := json.decode[map[string]Bar]('{"foo": "a", "bar": "b", "baz": "c"}')!
    assert m['foo'] == .a
    assert m['bar'] == .b
    assert m['baz'] == .c

    m2 := json.decode[map[string]Bar]('{"x": 0, "y": 1, "z": 10}')!
    assert m2['x'] == .a
    assert m2['y'] == .b
    assert m2['z'] == .c
}