def test_generate_foreignkeys(self):
filler = AutoFixture(
RelatedModel,
generate_fk=True)
for obj in filler.create(100):
self.assertEqual(obj.related.__class__, BasicModel)
self.assertEqual(obj.limitedfk.name, 'foo')
def create(model, count, *args, **kwargs):
'''
Create *count* instances of *model* using the either an appropiate
autofixture that was :ref:`registry <registered>` or fall back to the
default:class:`AutoFixture` class. *model* can be a model class or its
string representation (e.g. ``"app.ModelClass"``).
All positional and keyword arguments are passed to the autofixture
constructor. It is demonstrated in the example below which will create ten
superusers::
import autofixture
admins = autofixture.create('auth.User', 10, field_values={'is_superuser': True})
.. note:: See :ref:`AutoFixture` for more information.
:func:`create` will return a list of the created objects.
'''
from django.db import models
if isinstance(model, string_types):
model = models.get_model(*model.split('.', 1))
if model in REGISTRY:
autofixture = REGISTRY[model](model, *args, **kwargs)
else:
autofixture = AutoFixture(model, *args, **kwargs)
return autofixture.create(count)
def create(model, count, *args, **kwargs):
'''
Create *count* instances of *model* using the either an appropiate
autofixture that was :ref:`registry <registered>` or fall back to the
default:class:`AutoFixture` class. *model* can be a model class or its
string representation (e.g. ``"app.ModelClass"``).
All positional and keyword arguments are passed to the autofixture
constructor. It is demonstrated in the example below which will create ten
superusers::
import autofixture
admins = autofixture.create('auth.User', 10, field_values={'is_superuser': True})
.. note:: See :ref:`AutoFixture` for more information.
:func:`create` will return a list of the created objects.
'''
from django.db import models
if isinstance(model, basestring):
model = models.get_model(*model.split('.', 1))
if model in REGISTRY:
autofixture = REGISTRY[model](model, *args, **kwargs)
else:
autofixture = AutoFixture(model, *args, **kwargs)
return autofixture.create(count)
def test_constraints(self):
filler = AutoFixture(BasicModel, overwrite_defaults=False)
for obj in filler.create(100):
self.assertTrue(len(obj.chars) > 0)
self.assertEqual(type(obj.chars), str_)
self.assertTrue(len(obj.shortchars) <= 2)
self.assertEqual(type(obj.shortchars), str_)
self.assertTrue(type(obj.blankchars), str_)
self.assertEqualOr(type(obj.nullchars), str_, None)
self.assertEqual(type(obj.slugfield), str_)
self.assertEqual(type(obj.defaultint), int)
self.assertEqual(obj.defaultint, 1)
self.assertEqual(type(obj.intfield), int)
self.assertEqual(type(obj.sintfield), int)
self.assertEqual(type(obj.pintfield), int)
self.assertEqual(type(obj.psintfield), int)
self.assertEqual(type(obj.datefield), date)
self.assertEqual(type(obj.datetimefield), datetime)
self.assertEqual(type(obj.defaultdatetime), datetime)
self.assertEqual(obj.defaultdatetime, y2k())
self.assertEqual(type(obj.decimalfield), Decimal)
self.assertTrue('@' in obj.emailfield)
self.assertTrue('.' in obj.emailfield)
self.assertTrue(' ' not in obj.emailfield)
self.assertTrue(obj.ipaddressfield.count('.'), 3)
self.assertTrue(len(obj.ipaddressfield) >= 7)
self.assertEqual(BasicModel.objects.count(), 100)
def test_generate_only_some_foreignkeys(self):
filler = AutoFixture(
RelatedModel,
generate_fk=('related',))
for obj in filler.create(100):
self.assertEqual(obj.related.__class__, BasicModel)
self.assertEqual(obj.limitedfk, None)
def test_follow_m2m(self):
related = AutoFixture(SimpleModel).create()[0]
self.assertEqual(SimpleModel.objects.count(), 1)
filler = AutoFixture(M2MModel, follow_m2m=(2, 10))
for obj in filler.create(10):
self.assertEqual(list(obj.m2m.all()), [related])
def test_generate_only_some_foreignkeys(self):
filler = AutoFixture(
RelatedModel,
generate_fk=('related',))
for obj in filler.create(100):
self.assertEqual(obj.related.__class__, BasicModel)
self.assertEqual(obj.limitedfk, None)
def test_constraints(self):
filler = AutoFixture(
BasicModel,
overwrite_defaults=False)
for obj in filler.create(100):
self.assertTrue(len(obj.chars) > 0)
self.assertEqual(type(obj.chars), unicode)
self.assertTrue(len(obj.shortchars) <= 2)
self.assertEqual(type(obj.shortchars), unicode)
self.assertTrue(type(obj.blankchars), unicode)
self.assertEqualOr(type(obj.nullchars), unicode, None)
self.assertEqual(type(obj.slugfield), unicode)
self.assertEqual(type(obj.defaultint), int)
self.assertEqual(obj.defaultint, 1)
self.assertEqual(type(obj.intfield), int)
self.assertEqual(type(obj.sintfield), int)
self.assertEqual(type(obj.pintfield), int)
self.assertEqual(type(obj.psintfield), int)
self.assertEqual(type(obj.datefield), date)
self.assertEqual(type(obj.datetimefield), datetime)
self.assertEqual(type(obj.defaultdatetime), datetime)
self.assertEqual(obj.defaultdatetime, y2k())
self.assertEqual(type(obj.decimalfield), Decimal)
self.assertTrue('@' in obj.emailfield)
self.assertTrue('.' in obj.emailfield)
self.assertTrue(' ' not in obj.emailfield)
self.assertTrue(obj.ipaddressfield.count('.'), 3)
self.assertTrue(len(obj.ipaddressfield) >= 7)
self.assertEqual(BasicModel.objects.count(), 100)
def test_generate_foreignkeys(self):
filler = AutoFixture(
RelatedModel,
generate_fk=True)
for obj in filler.create(100):
self.assertEqual(obj.related.__class__, BasicModel)
self.assertEqual(obj.limitedfk.name, 'foo')
def test_generate_m2m_with_intermediary_model(self):
filler = AutoFixture(M2MModelThrough, generate_m2m=(1, 5))
all_m2m = set()
for obj in filler.create(10):
self.assertTrue(1 <= obj.m2m.count() <= 5)
all_m2m.update(obj.m2m.all())
self.assertEqual(SimpleModel.objects.count(), len(all_m2m))
def test_unique_together(self):
filler = AutoFixture(UniqueTogetherTestModel)
count1 = len(
filler.model._meta.get_field_by_name('choice1')[0].choices)
count2 = len(
filler.model._meta.get_field_by_name('choice2')[0].choices)
for obj in filler.create(count1 * count2):
pass
def test_deep_generate_foreignkeys(self):
filler = AutoFixture(
DeepLinkModel2,
generate_fk=True)
for obj in filler.create(10):
self.assertEqual(obj.related.__class__, DeepLinkModel1)
self.assertEqual(obj.related.related.__class__, SimpleModel)
self.assertEqual(obj.related.related2.__class__, SimpleModel)
def test_unique_together(self):
filler = AutoFixture(UniqueTogetherTestModel)
count1 = len(filler.model._meta.
get_field_by_name('choice1')[0].choices)
count2 = len(filler.model._meta.
get_field_by_name('choice2')[0].choices)
for obj in filler.create(count1 * count2):
pass
def test_field_values_overwrite_defaults(self):
fixture = AutoFixture(
BasicModel,
field_values={
'defaultint': 42,
})
obj = fixture.create(1)[0]
self.assertEqual(obj.defaultint, 42)
def test_deep_generate_foreignkeys(self):
filler = AutoFixture(
DeepLinkModel2,
generate_fk=True)
for obj in filler.create(10):
self.assertEqual(obj.related.__class__, DeepLinkModel1)
self.assertEqual(obj.related.related.__class__, SimpleModel)
self.assertEqual(obj.related.related2.__class__, SimpleModel)
def test_field_values_overwrite_defaults(self):
fixture = AutoFixture(
BasicModel,
field_values={
'defaultint': 42,
})
obj = fixture.create(1)[0]
self.assertEqual(obj.defaultint, 42)
def test_deep_generate_foreignkeys2(self):
filler = AutoFixture(DeepLinkModel2,
follow_fk=False,
generate_fk=('related', 'related__related'))
for obj in filler.create(10):
self.assertEqual(obj.related.__class__, DeepLinkModel1)
self.assertEqual(obj.related.related.__class__, SimpleModel)
self.assertEqual(obj.related.related2, None)
def test_follow_fk_for_o2o_primary_key(self):
# OneToOneField on primary key should follow if it is not table inheritance
filler = AutoFixture(O2OPrimaryKeyModel, follow_fk=True)
simple = SimpleModel.objects.create()
obj = filler.create()[0]
self.assertEqual(obj.o2o, simple)
self.assertRaises(CreateInstanceError, filler.create)
def test_generate_fk_for_o2o(self):
# OneToOneField is the same as a ForeignKey with unique=True
filler = AutoFixture(O2OModel, generate_fk=True)
all_o2o = set()
for obj in filler.create(10):
all_o2o.add(obj.o2o)
self.assertEqual(set(SimpleModel.objects.all()), all_o2o)
def test_follow_fk_for_o2o(self):
# OneToOneField is the same as a ForeignKey with unique=True
filler = AutoFixture(O2OModel, follow_fk=True)
simple = SimpleModel.objects.create()
obj = filler.create()[0]
self.assertEqual(obj.o2o, simple)
self.assertRaises(CreateInstanceError, filler.create)
def test_generate_m2m_with_intermediary_model(self):
filler = AutoFixture(
M2MModelThrough,
generate_m2m=(1, 5))
all_m2m = set()
for obj in filler.create(10):
self.assertTrue(1 <= obj.m2m.count() <= 5)
all_m2m.update(obj.m2m.all())
self.assertEqual(SimpleModel.objects.count(), len(all_m2m))
def test_deep_generate_foreignkeys2(self):
filler = AutoFixture(
DeepLinkModel2,
follow_fk=False,
generate_fk=('related', 'related__related'))
for obj in filler.create(10):
self.assertEqual(obj.related.__class__, DeepLinkModel1)
self.assertEqual(obj.related.related.__class__, SimpleModel)
self.assertEqual(obj.related.related2, None)
def test_follow_fk_for_o2o_primary_key(self):
# OneToOneField on primary key should follow if it is not table inheritance
filler = AutoFixture(O2OPrimaryKeyModel, follow_fk=True)
simple = SimpleModel.objects.create()
obj = filler.create()[0]
self.assertEqual(obj.o2o, simple)
self.assertRaises(CreateInstanceError, filler.create)
def test_generate_fk_for_o2o_primary_key(self):
# OneToOneField on primary key should follow if it is not table inheritance
filler = AutoFixture(O2OPrimaryKeyModel, generate_fk=True)
all_o2o = set()
for obj in filler.create(10):
all_o2o.add(obj.o2o)
self.assertEqual(set(SimpleModel.objects.all()), all_o2o)
def test_follow_fk_for_o2o(self):
# OneToOneField is the same as a ForeignKey with unique=True
filler = AutoFixture(O2OModel, follow_fk=True)
simple = SimpleModel.objects.create()
obj = filler.create()[0]
self.assertEqual(obj.o2o, simple)
self.assertRaises(CreateInstanceError, filler.create)
def test_generate_fk_for_o2o(self):
# OneToOneField is the same as a ForeignKey with unique=True
filler = AutoFixture(O2OModel, generate_fk=True)
all_o2o = set()
for obj in filler.create(10):
all_o2o.add(obj.o2o)
self.assertEqual(set(SimpleModel.objects.all()), all_o2o)
def test_generate_fk_for_o2o_primary_key(self):
# OneToOneField on primary key should follow if it is not table inheritance
filler = AutoFixture(O2OPrimaryKeyModel, generate_fk=True)
all_o2o = set()
for obj in filler.create(10):
all_o2o.add(obj.o2o)
self.assertEqual(set(SimpleModel.objects.all()), all_o2o)
def test_follow_m2m(self):
related = AutoFixture(SimpleModel).create()[0]
self.assertEqual(SimpleModel.objects.count(), 1)
filler = AutoFixture(
M2MModel,
follow_m2m=(2, 10))
for obj in filler.create(10):
self.assertEqual(list(obj.m2m.all()), [related])
def test_follow_only_some_foreignkeys(self):
related = AutoFixture(BasicModel).create()[0]
self.assertEqual(BasicModel.objects.count(), 1)
simple = SimpleModel.objects.create(name='foo')
simple2 = SimpleModel.objects.create(name='bar')
filler = AutoFixture(RelatedModel, follow_fk=('related', ))
for obj in filler.create(100):
self.assertEqual(obj.related, related)
self.assertEqual(obj.limitedfk, None)
def test_generate_m2m(self):
filler = AutoFixture(M2MModel, none_p=0, generate_m2m=(1, 5))
all_m2m = set()
all_secondm2m = set()
for obj in filler.create(10):
self.assertTrue(1 <= obj.m2m.count() <= 5)
self.assertTrue(1 <= obj.secondm2m.count() <= 5)
all_m2m.update(obj.m2m.all())
all_secondm2m.update(obj.secondm2m.all())
self.assertEqual(SimpleModel.objects.count(), len(all_m2m))
self.assertEqual(OtherSimpleModel.objects.count(), len(all_secondm2m))
def test_follow_only_some_m2m(self):
related = AutoFixture(SimpleModel).create()[0]
self.assertEqual(SimpleModel.objects.count(), 1)
other_related = AutoFixture(OtherSimpleModel).create()[0]
self.assertEqual(OtherSimpleModel.objects.count(), 1)
filler = AutoFixture(M2MModel, none_p=0, follow_m2m={
'm2m': (2, 10),
})
for obj in filler.create(10):
self.assertEqual(list(obj.m2m.all()), [related])
self.assertEqual(list(obj.secondm2m.all()), [])
def test_follow_only_some_foreignkeys(self):
related = AutoFixture(BasicModel).create()[0]
self.assertEqual(BasicModel.objects.count(), 1)
simple = SimpleModel.objects.create(name='foo')
simple2 = SimpleModel.objects.create(name='bar')
filler = AutoFixture(
RelatedModel,
follow_fk=('related',))
for obj in filler.create(100):
self.assertEqual(obj.related, related)
self.assertEqual(obj.limitedfk, None)
def test_field_values(self):
int_value = 1
char_values = (u'a', u'b')
filler = AutoFixture(
BasicModel,
field_values={
'intfield': 1,
'chars': generators.ChoicesGenerator(values=char_values),
'shortchars': lambda: u'ab',
})
for obj in filler.create(100):
self.assertEqual(obj.intfield, int_value)
self.assertTrue(obj.chars in char_values)
self.assertEqual(obj.shortchars, u'ab')
def test_field_values(self):
int_value = 1
char_values = (u'a', u'b')
filler = AutoFixture(
BasicModel,
field_values={
'intfield': 1,
'chars': generators.ChoicesGenerator(values=char_values),
'shortchars': lambda: u'ab',
})
for obj in filler.create(100):
self.assertEqual(obj.intfield, int_value)
self.assertTrue(obj.chars in char_values)
self.assertEqual(obj.shortchars, u'ab')
def test_generate_m2m(self):
filler = AutoFixture(
M2MModel,
none_p=0,
generate_m2m=(1, 5))
all_m2m = set()
all_secondm2m = set()
for obj in filler.create(10):
self.assertTrue(1 <= obj.m2m.count() <= 5)
self.assertTrue(1 <= obj.secondm2m.count() <= 5)
all_m2m.update(obj.m2m.all())
all_secondm2m.update(obj.secondm2m.all())
self.assertEqual(SimpleModel.objects.count(), len(all_m2m))
self.assertEqual(OtherSimpleModel.objects.count(), len(all_secondm2m))
def test_follow_only_some_m2m(self):
related = AutoFixture(SimpleModel).create()[0]
self.assertEqual(SimpleModel.objects.count(), 1)
other_related = AutoFixture(OtherSimpleModel).create()[0]
self.assertEqual(OtherSimpleModel.objects.count(), 1)
filler = AutoFixture(
M2MModel,
none_p=0,
follow_m2m={
'm2m': (2, 10),
})
for obj in filler.create(10):
self.assertEqual(list(obj.m2m.all()), [related])
self.assertEqual(list(obj.secondm2m.all()), [])
def test_create(self):
filler = AutoFixture(BasicModel)
filler.create(10)
self.assertEqual(BasicModel.objects.count(), 10)
def test_inheritence_model(self):
filler = AutoFixture(InheritModel)
filler.create(10)
self.assertEqual(InheritModel.objects.count(), 10)
def test_unique_field(self):
filler = AutoFixture(UniqueTestModel)
count = len(filler.model._meta.
get_field_by_name('choice1')[0].choices)
for obj in filler.create(count):
pass
def test_inheritence_unique_together_model(self):
filler = AutoFixture(InheritUniqueTogetherModel)
filler.create(10)
self.assertEqual(InheritUniqueTogetherModel.objects.count(), 10)
def test_unique_together(self):
filler = AutoFixture(UniqueTogetherTestModel)
count1 = len(get_field(filler.model, 'choice1').choices)
count2 = len(get_field(filler.model, 'choice2').choices)
for obj in filler.create(count1 * count2):
pass
def test_create(self):
filler = AutoFixture(BasicModel)
filler.create(10)
self.assertEqual(BasicModel.objects.count(), 10)
def test_unique_together(self):
filler = AutoFixture(UniqueTogetherTestModel)
count1 = len(get_field(filler.model, 'choice1').choices)
count2 = len(get_field(filler.model, 'choice2').choices)
for obj in filler.create(count1 * count2):
pass
def test_unique_field(self):
filler = AutoFixture(UniqueTestModel)
count = len(get_field(filler.model, 'choice1').choices)
for obj in filler.create(count):
pass
def test_inheritence_model(self):
filler = AutoFixture(InheritModel)
filler.create(10)
self.assertEqual(InheritModel.objects.count(), 10)
def test_inheritence_unique_together_model(self):
filler = AutoFixture(InheritUniqueTogetherModel)
filler.create(10)
self.assertEqual(InheritUniqueTogetherModel.objects.count(), 10)
def test_unique_field(self):
filler = AutoFixture(UniqueTestModel)
count = len(filler.model._meta.get_field_by_name('choice1')[0].choices)
for obj in filler.create(count):
pass
def test_unique_field(self):
filler = AutoFixture(UniqueTestModel)
count = len(get_field(filler.model, 'choice1').choices)
for obj in filler.create(count):
pass
