def test_get_foreign_key_for(self): def add_in_registry(): @register(Model) class Test: id = Integer(primary_key=True) @register(Model) class Test2: id = Integer(primary_key=True) test_id = Integer(foreign_key=Model.Test.use('id')) registry = self.init_registry(add_in_registry) mr = ModelRepr('Model.Test2') mas = mr.foreign_keys_for(registry, 'Model.Test') assert len(mas) == 1 assert [x.attribute_name for x in mas] == ['test_id']
def test_without_model(self, registry_blok): with pytest.raises(ModelReprException): ModelRepr(None)
def test_from_declaration(self): mr = ModelRepr('Model.System.Model') mra = ModelAdapter(mr) assert mr is mra
def test_get_registry_name(self, registry_blok): mr = ModelRepr('Model.System.Model') assert mr.model_name == 'Model.System.Model'
def test_get_primary_keys(self, registry_blok): registry = registry_blok mr = ModelRepr('Model.System.Model') mas = mr.primary_keys(registry) assert len(mas) == 1 assert [x.attribute_name for x in mas] == ['name']
def test_unexisting_model(self, registry_blok): registry = registry_blok mr = ModelRepr('Model.Unexisting.Model') with pytest.raises(ModelReprException): mr.check_model(registry)
def test_get_tablename(self, registry_blok): registry = registry_blok mr = ModelRepr('Model.System.Model') assert mr.tablename(registry) == 'system_model'
def test_capable_by_model_repr(self): assert ModelMapper.capable(ModelRepr('Model.System.Model'))
def test_model_repr(self): assert ModelMapper(ModelRepr('Model.System.Model'), None)
def test_from_declaration(self): mr = ModelRepr('Model.System.Model') mra = ModelAdapter(mr) self.assertIs(mr, mra)
def test_str(self, registry_blok): mr = ModelRepr('Model.System.Model') assert str(mr) == 'Model.System.Model'
def test_without_model(self): with self.assertRaises(ModelReprException): ModelRepr(None)
def test_get_primary_keys(self): registry = self.init_registry(None) mr = ModelRepr('Model.System.Model') mas = mr.primary_keys(registry) self.assertEqual(len(mas), 1) self.assertEqual([x.attribute_name for x in mas], ['name'])
def test_get_registry_name(self): mr = ModelRepr('Model.System.Model') self.assertEqual(mr.model_name, 'Model.System.Model')
def test_get_tablename(self): registry = self.init_registry(None) mr = ModelRepr('Model.System.Model') self.assertEqual(mr.tablename(registry), 'system_model')
def test_unexisting_model(self): registry = self.init_registry(None) mr = ModelRepr('Model.Unexisting.Model') with self.assertRaises(ModelReprException): mr.check_model(registry)