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)