def test_class(self):
"""Pass a class into the function.
Assert that the class passed in is returned.
"""
self.assertEqual(SampleEntity, orm._get_class(SampleEntity))
def test_name_module(self):
"""Pass a class name and module name into the function.
Assert that the named class is returned from the named module.
"""
self.assertEqual(
SampleEntity,
orm._get_class('SampleEntity', 'tests.robottelo.test_orm'),
)
def test_name(self):
"""Pass a string name into the function.
Assert that the named class is returned from :mod:`robottelo.entities`.
"""
self.assertEqual(
entities.ActivationKey,
orm._get_class('ActivationKey'),
)
