class TypesRegistryTest(TestCase):
async def setUp(self):
self.registry = TypesRegistry()
async def test_simple_set_get_object(self):
class SomeObject:
pass
instance = SomeObject()
self.registry.set(instance)
self.assertEqual(instance, self.registry.get(SomeObject))
async def test_get_object_not_found(self):
class OtherObject:
pass
self.assertIsNone(self.registry.get(OtherObject))
async def test_simple_coroutine_with_typehint(self):
class MyClass:
pass
async def my_coro(p: MyClass):
return p
instance = MyClass()
self.assertTrue(instance)
async def test_register_a_new_type_with_its_name(self):
"""
Um parametro registrado apenas com o tipo é diferente de um
parametro registrado com tipo e nome
"""
class MyClass:
async def my_coro(self, obj_id: int):
return obj_id
self.registry.set(42, param_name="obj_id")
self.registry.set(10)
self.assertEqual(42, self.registry.get(int, param_name="obj_id"))
async def test_register_returns_None_when_name_not_found(self):
"""
Lança uma exception quando tentamos buscar um parametro pelo
tipo e pelo nome mas ele não é encontrado.
"""
class MyClass:
async def my_coro(self, obj_id: int):
return obj_id
self.registry.set(42, param_name="obj_id")
self.assertEqual(None, self.registry.get(str, param_name="obj_id"))
class TypesRegistryTest(TestCase):
async def setUp(self):
self.registry = TypesRegistry()
async def test_simple_set_get_object(self):
class SomeObject:
pass
instance = SomeObject()
self.registry.set(instance)
self.assertEqual(instance, self.registry.get(SomeObject))
async def test_get_object_not_found(self):
class OtherObject:
pass
self.assertIsNone(self.registry.get(OtherObject))
async def test_get_object_of_list_type(self):
"""
Searching for List[Class]
"""
class MyClass:
pass
_list = [MyClass(), MyClass()]
self.registry.set(_list)
self.assertEqual(_list, self.registry.get(List[MyClass]))
async def test_get_object_list_type_empty_list(self):
class MyClass:
pass
_list: List[MyClass] = []
self.registry.set(_list, List[MyClass])
self.assertEqual(_list, self.registry.get(List[MyClass]))
async def test_simple_coroutine_with_typehint(self):
class MyClass:
pass
async def my_coro(p: MyClass):
return p
instance = MyClass()
self.assertTrue(instance)
async def test_resolve_multiple_coros_with_same_registry(self):
async def coro(p: int):
return p
registry_one = TypesRegistry()
registry_one.set(42)
resolver = ArgResolver(registry_one)
result = await resolver.wrap(coro)
self.assertEqual(42, await resolver.wrap(coro))
resolver2 = ArgResolver(registry_one)
self.assertEqual(42, await resolver2.wrap(coro))
self.assertEqual(42, registry_one.get(int))
class TypesRegistryTest(TestCase):
async def setUp(self):
self.registry = TypesRegistry()
async def test_simple_set_get_object(self):
class SomeObject:
pass
instance = SomeObject()
self.registry.set(instance)
self.assertEqual(instance, self.registry.get(SomeObject))
async def test_get_object_not_found(self):
class OtherObject:
pass
self.assertIsNone(self.registry.get(OtherObject))
async def test_simple_coroutine_with_typehint(self):
class MyClass:
pass
async def my_coro(p: MyClass):
return p
instance = MyClass()
self.assertTrue(instance)
async def test_register_a_new_type_with_its_name(self):
"""
Um parametro registrado apenas com o tipo é diferente de um
parametro registrado com tipo e nome
"""
class MyClass:
async def my_coro(self, obj_id: int):
return obj_id
self.registry.set(42, param_name="obj_id")
self.registry.set(10)
self.assertEqual(42, self.registry.get(int, param_name="obj_id"))
async def test_register_returns_None_when_name_not_found(self):
"""
Lança uma exception quando tentamos buscar um parametro pelo
tipo e pelo nome mas ele não é encontrado.
"""
class MyClass:
async def my_coro(self, obj_id: int):
return obj_id
self.registry.set(42, param_name="obj_id")
self.assertEqual(None, self.registry.get(str, param_name="obj_id"))
async def test_aceita_tipo_generico_um_arg_tipo_simples(self):
class MyGeneric(Generic[T]):
def __init__(self, val: T) -> None:
self._val: T = val
v: MyGeneric[int] = MyGeneric(10)
self.registry.set(v, MyGeneric[int])
self.assertEqual(v, self.registry.get(MyGeneric[int]))
async def test_aceita_generico_dois_args(self):
P = TypeVar("P")
class MyGenericDoisArgs(Generic[T, P]):
def __init__(self, val: T) -> None:
self._val: T = val
v: MyGenericDoisArgs[int, str] = MyGenericDoisArgs(10)
self.registry.set(v, MyGenericDoisArgs[int, str])
self.assertEquals(v, self.registry.get(MyGenericDoisArgs[int, str]))
async def test_generico_dois_args_ordem_importa(self):
P = TypeVar("P")
class MyGenericDoisArgs(Generic[T, P]):
def __init__(self, val: T) -> None:
self._val: T = val
v: MyGenericDoisArgs[int, str] = MyGenericDoisArgs(10)
self.registry.set(v, MyGenericDoisArgs[int, str])
self.assertIsNone(self.registry.get(MyGenericDoisArgs[str, int]))
async def test_generico_um_arg_tipo_complexo(self):
class MyGeneric(Generic[T]):
def __init__(self, val: T) -> None:
self._val: T = val
class OtherObject:
pass
v: MyGeneric[OtherObject] = MyGeneric(OtherObject())
self.registry.set(v, MyGeneric[OtherObject])
self.assertEquals(v, self.registry.get(MyGeneric[OtherObject]))
async def test_generico_um_arg_get_by_name(self):
class MyGeneric(Generic[T]):
def __init__(self, val: T) -> None:
self._val: T = val
obj = MyGeneric(42)
self.registry.set(obj,
type_definition=MyGeneric[int],
param_name="p_name")
self.assertEqual(
obj, self.registry.get(MyGeneric[int], param_name="p_name"))
async def test_dois_generic_um_arg_get_by_name(self):
class MyGeneric(Generic[T]):
def __init__(self, val: T) -> None:
self._val: T = val
obj = MyGeneric(42)
obj_2 = MyGeneric(99)
self.registry.set(obj,
type_definition=MyGeneric[int],
param_name="p_name")
self.registry.set(obj_2,
type_definition=MyGeneric[int],
param_name="p_name_2")
self.assertEqual(
obj._val,
self.registry.get(MyGeneric[int], param_name="p_name")._val,
)
self.assertEqual(
obj_2._val,
self.registry.get(MyGeneric[int], param_name="p_name_2")._val,
)
