def test_pass_in_entity_id(self):
"""Let the ``entity_obj_or_id`` argument be an integer."""
entity_id = gen_integer(min_value=1)
entity_obj = entity_mixins._make_entity_from_id(
SampleEntity, entity_id, self.cfg)
self.assertIsInstance(entity_obj, SampleEntity)
self.assertEqual(entity_obj.id, entity_id)
def test_pass_in_entity_id(self):
"""Let the ``entity_obj_or_id`` argument be an integer."""
entity_id = gen_integer(min_value=1)
entity_obj = entity_mixins._make_entity_from_id(
SampleEntity,
entity_id,
self.cfg
)
self.assertIsInstance(entity_obj, SampleEntity)
self.assertEqual(entity_obj.id, entity_id) # pylint:disable=no-member
def test_pass_in_entity_obj(self):
"""Let the ``entity_obj_or_id`` argument be an entity object."""
self.assertIsInstance(
entity_mixins._make_entity_from_id(
SampleEntity,
SampleEntity(self.cfg),
self.cfg
),
SampleEntity
)
def test_pass_in_entity_id(self):
"""Pass in an entity class and an entity ID."""
entity_id = gen_integer(min_value=1)
entity_obj = entity_mixins._make_entity_from_id(
SampleEntity,
entity_id,
self.server_config
)
self.assertIsInstance(entity_obj, SampleEntity)
self.assertEqual(entity_obj.id, entity_id) # pylint:disable=no-member
def test_pass_in_entity_obj(self):
"""Pass in an entity class and an entity."""
self.assertIsInstance(
entity_mixins._make_entity_from_id(
SampleEntity,
SampleEntity(self.server_config),
self.server_config
),
SampleEntity
)
def test_pass_in_entity_obj(self):
"""Let the ``entity_obj_or_id`` argument be an entity object."""
self.assertIsInstance(
entity_mixins._make_entity_from_id(
SampleEntity,
SampleEntity(self.cfg),
self.cfg
),
SampleEntity
)
