def test_models_is_not_in_manager___no_model_is_removed(self):
manager = OasisExposuresManager([
fake_model('supplier', 'model', 'version'),
fake_model('supplier2', 'model2', 'version2'),
])
expected = manager.models
manager.delete_models([fake_model('supplier3', 'model3', 'version3')])
self.assertEqual(expected, manager.models)
def test_models_exist_in_manager___models_are_removed(self):
models = [
fake_model('supplier', 'model', 'version'),
fake_model('supplier2', 'model2', 'version2'),
fake_model('supplier3', 'model3', 'version3'),
]
manager = OasisExposuresManager(models)
manager.delete_models(models[1:])
self.assertEqual({models[0].key: models[0]}, manager.models)