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
}