def test_get_fk_name_with_name_different_of_column_name(self):
def add_in_registry():
@register(Model)
class Test:
name = String(primary_key=True, db_column_name='other')
registry = self.init_registry(add_in_registry)
ma = ModelAttribute('Model.Test', 'name')
assert ma.get_fk_name(registry) == 'test.other'
def test_get_fk_name_with_name_different_of_column_name(self):
def add_in_registry():
@register(Model)
class Test:
name = String(primary_key=True, db_column_name='other')
registry = self.init_registry(add_in_registry)
ma = ModelAttribute('Model.Test', 'name')
self.assertEqual(ma.get_fk_name(registry), 'test.other')
def test_get_fk_name(self, registry_blok):
registry = registry_blok
ma = ModelAttribute('Model.System.Model', 'name')
assert ma.get_fk_name(registry) == 'system_model.name'
def test_get_fk_name_unexisting_attribute(self, registry_blok):
registry = registry_blok
ma = ModelAttribute('Model.System.Model', 'id')
with pytest.raises(ModelAttributeException):
ma.get_fk_name(registry)
def test_get_fk_name_with_unexisting_model(self, registry_blok):
registry = registry_blok
ma = ModelAttribute('Model.Unexisting.Model', 'name')
with pytest.raises(ModelAttributeException):
ma.get_fk_name(registry)
def test_get_fk_name_with_wrong_model(self, registry_blok):
registry = registry_blok
ma = ModelAttribute('Model.System', 'name')
with pytest.raises(ModelAttributeException):
ma.get_fk_name(registry)
def test_get_fk_name(self):
registry = self.init_registry(None)
ma = ModelAttribute('Model.System.Model', 'name')
self.assertEqual(ma.get_fk_name(registry), 'system_model.name')
def test_get_fk_name_unexisting_attribute(self):
registry = self.init_registry(None)
ma = ModelAttribute('Model.System.Model', 'id')
with self.assertRaises(ModelAttributeException):
ma.get_fk_name(registry)
def test_get_fk_name_with_unexisting_model(self):
registry = self.init_registry(None)
ma = ModelAttribute('Model.Unexisting.Model', 'name')
with self.assertRaises(ModelAttributeException):
ma.get_fk_name(registry)
