def handle_noargs(self, **options): # Discover datasets in apps fixtures = find_datasets(settings.INSTALLED_APPS) if not len(fixtures): raise CommandError('No datasets fixtures found.') fill_database(fixtures, int(options.get('verbosity', 1)))
def handle(self, *apps, **options): fixtures = [] for app in apps: datasets = get_datasets(app) if not len(datasets): raise LookupError('No dataset found for %s.' % app) fixtures.extend(datasets) fill_database(fixtures, int(options.get('verbosity', 1)))
def test_raises_when_call_fill_database_with_empty_list(self): """Raises when calling fill_database not passing fixtures""" self.assertRaises(ValueError, lambda: fill_database([])) self.assertRaises(TypeError, lambda: fill_database('None')) self.assertRaises(TypeError, lambda: fill_database(DummyData))
def test_fill_database_with_data_from_fixture(self): """Check that fill_database loaded DummyData to the database""" fill_database([DummyData], verbosity=0) obj = Dummy.objects.get(name='Buster') self.assertEquals(obj.name, DummyData.buster.name)