def _fixture_teardown(self):
from dmqs.repository import Repository
repository = Repository()
repository.clean()
for restore_config in self.apps_config:
unpatch_models(restore_config[0], restore_config[1])
def test_repository_save():
repository = Repository()
repository.clean()
model_name = 'ModelName'
instance = model_mock(model_name)
update = repository.save(model_name, instance)
assert update == False
assert repository.__dict__['data'][model_name] == [instance]
assert type(repository.__dict__['data'][model_name]) == list
assert repository.__dict__['data'][model_name][0].id == 1
def test_repository_save():
repository = Repository()
repository.clean()
model_name = 'ModelName'
instance = model_mock(model_name)
update = repository.save(model_name, instance)
assert update == False
assert repository.__dict__['data'][model_name] == [instance]
assert type(repository.__dict__['data'][model_name]) == list
assert repository.__dict__['data'][model_name][0].id == 1
def test_repository_get_models():
repository = Repository()
repository.clean()
model_name = 'ModelName'
instance = model_mock(model_name)
update1 = repository.save(model_name, instance)
instance2 = model_mock(model_name)
update2 = repository.save(model_name, instance2)
assert update1 == False
assert update2 == False
assert repository.get_models(model_name) == [instance, instance2]
def test_repository_get_models():
repository = Repository()
repository.clean()
model_name = 'ModelName'
instance = model_mock(model_name)
update1 = repository.save(model_name, instance)
instance2 = model_mock(model_name)
update2 = repository.save(model_name, instance2)
assert update1 == False
assert update2 == False
assert repository.get_models(model_name) == [instance, instance2]
def test_repository_delete():
repository = Repository()
repository.clean()
model_name = 'ModelName'
instance = model_mock(model_name)
update1 = repository.save(model_name, instance)
instance2 = model_mock(model_name)
update2 = repository.save(model_name, instance2)
assert update1 == False
assert update2 == False
delete1 = repository.delete(model_name, [instance])
delete2 = repository.delete(model_name, [instance2])
assert len(repository.get_models(model_name)) == 0
def test_repository_delete():
repository = Repository()
repository.clean()
model_name = 'ModelName'
instance = model_mock(model_name)
update1 = repository.save(model_name, instance)
instance2 = model_mock(model_name)
update2 = repository.save(model_name, instance2)
assert update1 == False
assert update2 == False
delete1 = repository.delete(model_name, [instance])
delete2 = repository.delete(model_name, [instance2])
assert len(repository.get_models(model_name)) == 0
def pytest_runtest_setup():
repository = Repository()
repository.clean()
MemoryQuerySet.fetch_from_repo = False
