import pytest

from rpdk.core.jsonutils.resolver import MULTIPLE, ContainerType, ResolvedType

from rpdk.java.resolver import PRIMITIVE_TYPES, translate_type

RESOLVED_TYPES = [

(ResolvedType(ContainerType.PRIMITIVE, item_type), formats["default"])

for item_type, formats in PRIMITIVE_TYPES.items()

]

RESOLVED_INTEGER_FORMATS = [

(

ResolvedType(ContainerType.PRIMITIVE, "integer", "int64"),

PRIMITIVE_TYPES["integer"]["int64"],

),

(

ResolvedType(ContainerType.PRIMITIVE, "integer", "int32"),

PRIMITIVE_TYPES["integer"]["int32"],

),

]

def test_translate_type_model_passthrough():

item_type = object()

traslated = translate_type(ResolvedType(ContainerType.MODEL, item_type))

assert traslated is item_type

@pytest.mark.parametrize("resolved_type,native_type", RESOLVED_TYPES)

def test_translate_type_primitive(resolved_type, native_type):

assert translate_type(resolved_type) == native_type

@pytest.mark.parametrize("resolved_type,native_type", RESOLVED_TYPES)

def test_translate_type_dict(resolved_type, native_type):

traslated = translate_type(ResolvedType(ContainerType.DICT, resolved_type))

assert traslated == f"Map<String, {native_type}>"

def test_translate_type_multiple():

assert translate_type(ResolvedType(ContainerType.MULTIPLE, MULTIPLE)) == "Object"

@pytest.mark.parametrize("resolved_type,native_type", RESOLVED_TYPES)

def test_translate_type_list(resolved_type, native_type):

traslated = translate_type(ResolvedType(ContainerType.LIST, resolved_type))

assert traslated == f"List<{native_type}>"

@pytest.mark.parametrize("resolved_type,native_type", RESOLVED_TYPES)

def test_translate_type_set(resolved_type, native_type):

traslated = translate_type(ResolvedType(ContainerType.SET, resolved_type))

assert traslated == f"Set<{native_type}>"

@pytest.mark.parametrize("resolved_type,_java_type", RESOLVED_TYPES)

def test_translate_type_unknown(resolved_type, _java_type):

with pytest.raises(ValueError):

translate_type(ResolvedType("foo", resolved_type))

@pytest.mark.parametrize("resolved_type,java_type", RESOLVED_INTEGER_FORMATS)

def test_translate_type_integer_formats(resolved_type, java_type):

assert translate_type(resolved_type) == java_type

def test_translate_type_unavailable_format():

resolved_type = ResolvedType(ContainerType.PRIMITIVE, "integer", "int128")

assert translate_type(resolved_type) == PRIMITIVE_TYPES["integer"]["default"]