def test_accept_model_as_string(self):
person = mommy.make_one('model_mommy.person')
self.assertTrue(isinstance(person, Person))
person = mommy.prepare_one('model_mommy.Person')
self.assertTrue(isinstance(person, Person))
people = mommy.make_many('model_mommy.person')
[self.assertTrue(isinstance(person, Person)) for person in people]
def test_accept_model_as_string(self):
person = mommy.make_one('model_mommy.person')
self.assertIsInstance(person, Person)
person = mommy.prepare_one('model_mommy.Person')
self.assertIsInstance(person, Person)
people = mommy.make_many('model_mommy.person')
[self.assertIsInstance(person, Person) for person in people]
def test_prepare_one_should_not_create_one_object(self):
dog = mommy.prepare_one(Dog)
self.assertTrue(isinstance(dog, Dog))
self.assertTrue(isinstance(dog.owner, Person))
# makes sure database is clean
self.assertEqual(Person.objects.all().count(), 0)
self.assertEqual(Dog.objects.all().count(), 0)
def test_register_rule_no_score(self):
rulecard = mommy.prepare_one(RuleCard)
scoresheet = _prepare_scoresheet(self.game, "p1", blue = 1)
scoresheet.register_score_from_rule(rulecard, 'test', is_random = True)
self.assertEqual(rulecard, scoresheet.scores_from_rule[0].rulecard)
self.assertEqual('test', scoresheet.scores_from_rule[0].detail)
self.assertIsNone(scoresheet.scores_from_rule[0].score)
self.assertTrue(getattr(scoresheet.scores_from_rule[0], 'is_random', False))
def test_prepare_one_should_not_persist_one_object(self):
person = mommy.prepare_one(Person)
self.assertTrue(isinstance(person, Person))
# makes sure database is clean
self.assertEqual(Person.objects.all().count(), 0)
self.assertEqual(person.id, None)
def test_prepare_one_should_not_create_one_object(self):
dog = mommy.prepare_one(Dog)
self.assertTrue(isinstance(dog, Dog))
self.assertTrue(isinstance(dog.owner, Person))
# makes sure database is clean
self.assertEqual(Person.objects.all().count(), 0)
self.assertEqual(Dog.objects.all().count(), 0)
def test_prepare_one_should_not_persist_one_object(self):
person = mommy.prepare_one(Person)
self.assertTrue(isinstance(person, Person))
# makes sure database is clean
self.assertEqual(Person.objects.all().count(), 0)
self.assertEqual(person.id, None)
def test_prepare_one_should_not_persist_one_object(self):
from model_mommy import mommy
from model_mommy.models import Person
person = mommy.prepare_one(Person)
self.assertTrue(isinstance(person, Person))
# makes sure database is clean
self.assertEqual(Person.objects.all().count(), 0)
def test_prepare_one_should_not_persist_one_object(self):
"""
prepare_one method is deprecated, so this test must be removed when the
method is removed
"""
person = mommy.prepare_one(Person)
self.assertIsInstance(person, Person)
self.assertEqual(Person.objects.all().count(), 0)
self.assertEqual(person.id, None)
def test_prepare_one_should_not_persist_one_object(self):
"""
prepare_one method is deprecated, so this test must be removed when the
method is removed
"""
person = mommy.prepare_one(Person)
self.assertIsInstance(person, Person)
self.assertEqual(Person.objects.all().count(), 0)
self.assertEqual(person.id, None)
def test_prepare_model_with_M2MField_does_not_hit_database(self):
from model_mommy.models import DummyM2MModel
from model_mommy.models import DummyRelationModel
from model_mommy import mommy
dummy_m2m_model = mommy.prepare_one(DummyM2MModel)
# relation is not created if parent model is not persisted
self.assertEqual(DummyRelationModel.objects.count(), 0)
self.assertRaises(ValueError, lambda: dummy_m2m_model.m2m_field)
def test_prepare_one_object_with_fill_null_as_false(self):
from model_mommy import mommy
from model_mommy.models import Person
person = mommy.prepare_one(Person, fill_null=False)
meta = person._meta
fields = meta.fields
for field in fields:
name = field.name
if field.null:
field_value = getattr(person, name)
self.assertIsNone(field_value)
def test_ForeignKey_model_field_population_should_work_with_prepare(self):
dog = mommy.prepare_one(Dog, breed='X1', owner__name='Bob')
self.assertEqual('X1', dog.breed)
self.assertEqual('Bob', dog.owner.name)
def test_unicode_regression(self):
obj = mommy.prepare_one(u'generic.Person')
self.assertIsInstance(obj, Person)
def test_model_class(self):
obj = mommy.prepare_one(Person)
self.assertIsInstance(obj, Person)
def test_model_string(self):
obj = mommy.prepare_one('Person')
self.assertIsInstance(obj, Person)
def test_raise_on_ambiguous_model_string(self):
with self.assertRaises(AmbiguousModelName):
obj = mommy.prepare_one('Ambiguous')
def test_ForeignKey_model_field_population_should_work_with_prepare(self):
dog = mommy.prepare_one(Dog, breed='X1', owner__name='Bob')
self.assertEqual('X1', dog.breed)
self.assertEqual('Bob', dog.owner.name)