def test_pass_in_both(self):
"""Let ``entity_objs_and_ids`` be an iterable of integers and IDs."""
entities = entity_mixins._make_entities_from_ids(
SampleEntity, [SampleEntity(self.cfg), 5], self.cfg)
self.assertEqual(len(entities), 2)
for entity in entities:
self.assertIsInstance(entity, SampleEntity)
def test_pass_in_emtpy_iterable(self):
"""Let the ``entity_objs_and_ids`` argument be an empty iterable."""
for iterable in ([], tuple()):
self.assertEqual(
entity_mixins._make_entities_from_ids(SampleEntity, iterable, self.cfg),
[],
)
def test_pass_in_entity_obj(self):
"""Let the ``entity_objs_and_ids`` arg be an iterable of entities."""
for num_entities in range(4):
input_entities = [SampleEntity(self.cfg) for _ in range(num_entities)]
output_entities = entity_mixins._make_entities_from_ids(SampleEntity, input_entities, self.cfg)
self.assertEqual(num_entities, len(output_entities))
for output_entity in output_entities:
self.assertIsInstance(output_entity, SampleEntity)
def test_pass_in_entity_ids(self):
"""Let the ``entity_objs_and_ids`` arg be an iterable of integers."""
for num_entities in range(4):
entity_ids = [gen_integer(min_value=1) for _ in range(num_entities)]
entities = entity_mixins._make_entities_from_ids(SampleEntity, entity_ids, self.cfg)
self.assertEqual(len(entities), len(entity_ids))
for i, entity_id in enumerate(entity_ids):
self.assertIsInstance(entities[i], SampleEntity)
self.assertEqual(entities[i].id, entity_id)
def test_pass_in_entity_ids(self):
"""Let the ``entity_objs_and_ids`` arg be an iterable of integers."""
for num_entities in range(4):
entity_ids = [gen_integer(min_value=1) for _ in range(num_entities)]
entities = entity_mixins._make_entities_from_ids(SampleEntity, entity_ids, self.cfg)
self.assertEqual(len(entities), len(entity_ids))
for i in range(len(entity_ids)):
self.assertIsInstance(entities[i], SampleEntity)
self.assertEqual(entities[i].id, entity_ids[i])
def test_pass_in_both(self):
"""Let ``entity_objs_and_ids`` be an iterable of integers and IDs."""
entities = entity_mixins._make_entities_from_ids(
SampleEntity,
[SampleEntity(self.cfg), 5],
self.cfg
)
self.assertEqual(len(entities), 2)
for entity in entities:
self.assertIsInstance(entity, SampleEntity)
def test_pass_in_entity_obj(self):
"""Let the ``entity_objs_and_ids`` arg be an iterable of entities."""
for num_entities in range(4):
input_entities = [SampleEntity(self.cfg) for _ in range(num_entities)]
output_entities = entity_mixins._make_entities_from_ids(
SampleEntity, input_entities, self.cfg
)
self.assertEqual(num_entities, len(output_entities))
for output_entity in output_entities:
self.assertIsInstance(output_entity, SampleEntity)
def test_pass_in_both(self):
"""Pass in an entity class and an iterable with entities and IDs."""
entities = entity_mixins._make_entities_from_ids(
SampleEntity,
[SampleEntity(self.server_config), 5],
self.server_config
)
self.assertEqual(len(entities), 2)
for entity in entities:
self.assertIsInstance(entity, SampleEntity)
def test_pass_in_emtpy_iterable(self):
"""Let the ``entity_objs_and_ids`` argument be an empty iterable."""
for iterable in ([], tuple()):
self.assertEqual(
entity_mixins._make_entities_from_ids(
SampleEntity,
iterable,
self.cfg
),
[],
)
def test_pass_in_emtpy_iterable(self):
"""Pass in an entity class and an empty iterable."""
for iterable in ([], tuple()):
self.assertEqual(
[],
entity_mixins._make_entities_from_ids(
SampleEntity,
iterable,
self.server_config
)
)
def test_pass_in_entity_obj(self):
"""Pass in an entity class and an iterable containing entities."""
for num_entities in range(4):
input_entities = [
SampleEntity(self.server_config) for _ in range(num_entities)
]
output_entities = entity_mixins._make_entities_from_ids(
SampleEntity,
input_entities,
self.server_config
)
self.assertEqual(num_entities, len(output_entities))
for output_entity in output_entities:
self.assertIsInstance(output_entity, SampleEntity)
def test_pass_in_entity_ids(self):
"""Pass in an entity class and an iterable containing entity IDs."""
for num_entities in range(4):
entity_ids = [
gen_integer(min_value=1) for _ in range(num_entities)
]
entities = entity_mixins._make_entities_from_ids(
SampleEntity,
entity_ids,
self.server_config
)
self.assertEqual(len(entities), len(entity_ids))
for i in range(len(entity_ids)):
self.assertIsInstance(entities[i], SampleEntity)
self.assertEqual(entities[i].id, entity_ids[i])
