def test_custom_type_generator(self):
def int_generator():
return 9
Generator.register(int_generator, "IntegerField")
my_obj = Generator.create(CharModel)
self.assertEquals(my_obj.int_field, 9)
my_obj2 = Generator.create(CharModel)
self.assertEquals(my_obj2.int_field, 9)
def test_custom_field_generator(self):
def int_generator():
return 12
Generator.register(int_generator, CharModel, "int_field",)
my_obj = Generator.create(CharModel)
self.assertEquals(my_obj.int_field, 12)
my_obj = Generator.create(CharModel, int_field=18)
self.assertEquals(my_obj.int_field, 18)
def test_many2many_generator(self):
my_obj = Generator.create(Many2ManyModel)
self.assertTrue(my_obj.many2many.count()>0)
def test_value_generator(self):
Generator.register(10, CharModel, "int_field",)
my_obj = Generator.create(CharModel)
self.assertEquals(my_obj.int_field, 10)
def test_partial_value(self):
my_obj = Generator.create(CharModel, int_field=18)
self.assertEquals(my_obj.int_field, 18)
def check_this_model(self, model):
self.assertRaises(IntegrityError, model.objects.create)
my_obj = Generator.create(model)
self.assertTrue(isinstance(my_obj,model), "The object should be of the %s class" % model)
