def test_get_model_returns_registered_if_exists(self, monkeypatch, model_registry, model_schema):
factory = ModelFactory(model_schema)
model = factory.make()
assert model_registry.is_registered(model_schema.model_name)
# prevent future calls to `make` and assume model is current
monkeypatch.setattr(factory, 'make', lambda: None)
monkeypatch.setattr(model_schema, 'is_current_model', lambda x: True)
assert factory.get_model() == model
def _factory(self):
return ModelFactory(self)
def test_destroy_model_unregisters(self, model_registry, model_schema):
factory = ModelFactory(model_schema)
factory.make_model()
assert model_registry.is_registered(model_schema.model_name)
factory.destroy_model()
assert not model_registry.is_registered(model_schema.model_name)
def test_destroy_model_disconnects_signal(self, model_schema):
factory = ModelFactory(model_schema)
model = factory.make_model()
assert self.schema_checker_is_connected(model)
factory.destroy_model()
assert not self.schema_checker_is_connected(model)
def test_make_model_registers(self, model_registry, model_schema):
ModelFactory(model_schema).make_model()
assert model_registry.is_registered(model_schema.model_name)
def test_make_model_connects_signals(self, model_schema):
model = ModelFactory(model_schema).make_model()
assert self.schema_checker_is_connected(model)
def test_schema_defines_model_meta(self, model_schema):
model = ModelFactory(model_schema).make_model()
assert model.__name__ == model_schema.model_name
assert model._meta.db_table == model_schema.db_table
assert model._meta.verbose_name == model_schema.name
def test_model_has_field_with_field_on_schema(self, model_schema, field_schema):
model = ModelFactory(model_schema).make_model()
assert isinstance(model._meta.get_field(field_schema.name), models.IntegerField)
def test_model_has_base_attributes(self, model_schema):
model = ModelFactory(model_schema).make_model()
for attr in ("_declared", "__module__"):
assert hasattr(model, attr)
def test_get_model_makes_if_not_exists(self, model_registry, model_schema):
model_registry.unregister_model(model_schema.model_name)
assert not model_registry.is_registered(model_schema.model_name)
ModelFactory(model_schema).get_model()
assert model_registry.is_registered(model_schema.model_name)
def test_model_has_field_with_field_on_schema(self,
model_schema_with_field):
model = ModelFactory(model_schema_with_field).make()
assert isinstance(model._meta.get_field('integer'),
models.IntegerField)
def test_model_has_base_attributes(self, model_schema):
model = ModelFactory(model_schema).make()
for attr in ('_schema', '_declared', '__module__'):
assert hasattr(model, attr)
def test_get_model_makes_if_not_exists(self, model_registry,
unsaved_model_schema):
assert not model_registry.is_registered(
unsaved_model_schema.model_name)
ModelFactory(unsaved_model_schema).get_model()
assert model_registry.is_registered(unsaved_model_schema.model_name)