class StaticSequentialDataFixtureTestCase(TestCase, DataFixtureTestCase): def setUp(self): self.fixture = StaticSequentialDataFixture() def test_it_must_fill_fields_sequencially_by_attribute_if_field_is_unique(self): field = models.IntegerField(unique=True) field.name = 'x' self.assertEquals(1, self.fixture.generate_data(field)) self.assertEquals(2, self.fixture.generate_data(field)) def test_it_must_fill_fields_with_static_value_by_attribute_if_field_is_not_unique(self): field = models.IntegerField(unique=False) field.name = 'x' self.assertEquals(1, self.fixture.generate_data(field)) self.assertEquals(1, self.fixture.generate_data(field))
def get_data_fixture(default='sequential'): # It must be 'sequential', 'static_sequential', 'global_sequential', 'random' or 'path.to.CustomDataFixtureClass' try: INTERNAL_DATA_FIXTURES = { 'sequential': SequentialDataFixture(), 'static_sequential': StaticSequentialDataFixture(), 'global_sequential': GlobalSequentialDataFixture(), 'random': RandomDataFixture() } if hasattr(settings, 'DDF_DEFAULT_DATA_FIXTURE'): if settings.DDF_DEFAULT_DATA_FIXTURE in INTERNAL_DATA_FIXTURES.keys( ): return INTERNAL_DATA_FIXTURES[ settings.DDF_DEFAULT_DATA_FIXTURE] else: # path.to.CustomDataFixtureClass mod_name, obj_name = get_mod_func( settings.DDF_DEFAULT_DATA_FIXTURE) module = import_module(mod_name) custom_data_fixture = getattr(module, obj_name) return custom_data_fixture() else: return INTERNAL_DATA_FIXTURES[default] except: six.reraise( DDFImproperlyConfigured, DDFImproperlyConfigured( "DDF_DEFAULT_DATA_FIXTURE (%s) must be 'sequential', 'static_sequential', 'global_sequential', 'random' or 'path.to.CustomDataFixtureClass'." % settings.DDF_DEFAULT_DATA_FIXTURE), sys.exc_info()[2])
class StaticSequentialDataFixtureTestCase(TestCase, DataFixtureTestCase): def setUp(self): self.fixture = StaticSequentialDataFixture() def test_it_must_fill_fields_sequencially_by_attribute_if_field_is_unique( self): field = models.IntegerField(unique=True) field.name = 'x' self.assertEquals(1, self.fixture.generate_data(field)) self.assertEquals(2, self.fixture.generate_data(field)) def test_it_must_fill_fields_with_static_value_by_attribute_if_field_is_not_unique( self): field = models.IntegerField(unique=False) field.name = 'x' self.assertEquals(1, self.fixture.generate_data(field)) self.assertEquals(1, self.fixture.generate_data(field))
def setUp(self): self.fixture = StaticSequentialDataFixture()
settings, config_name) else default except DDFImproperlyConfigured: six.reraise( DDFImproperlyConfigured, DDFImproperlyConfigured( "%s (%s) must be True or False." % (config_name, getattr(settings, config_name))), sys.exc_info()[2]) # DDF_DEFAULT_DATA_FIXTURE default = 'sequential' # It must be 'sequential', 'static_sequential', 'global_sequential', 'random' or 'path.to.CustomDataFixtureClass' try: INTERNAL_DATA_FIXTURES = { 'sequential': SequentialDataFixture(), 'static_sequential': StaticSequentialDataFixture(), 'global_sequential': GlobalSequentialDataFixture(), 'random': RandomDataFixture() } if hasattr(settings, 'DDF_DEFAULT_DATA_FIXTURE'): if settings.DDF_DEFAULT_DATA_FIXTURE in INTERNAL_DATA_FIXTURES.keys(): DDF_DEFAULT_DATA_FIXTURE = INTERNAL_DATA_FIXTURES[ settings.DDF_DEFAULT_DATA_FIXTURE] else: # path.to.CustomDataFixtureClass mod_name, obj_name = get_mod_func( settings.DDF_DEFAULT_DATA_FIXTURE) module = import_module(mod_name) custom_data_fixture = getattr(module, obj_name) DDF_DEFAULT_DATA_FIXTURE = custom_data_fixture() else:
def setUp(self): self.fixture = StaticSequentialDataFixture()