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)