def test_create_csv_per_sub_category_invalid_sub_category(self):
main_category = _get_horeca_main_category()
category = Category.objects.filter(parent_id__isnull=False).exclude(
parent_id=main_category.pk).first()
with self.assertRaises(NotImplementedError):
create_csv_per_sub_category(category, '/tmp', 1, 2019)
def test_create_csv_files_save(self, patched_model):
# Usage of Django storage means the difference between local and remote
# storage is abstracted away, so the previously 2 tests were merged.
patched_model.uploaded_file = mock.MagicMock()
patched_model.uploaded_file.save = mock.MagicMock()
# create a Signal with a horeca sub-category
main_category = _get_horeca_main_category()
category = Category.objects.filter(
parent_id__isnull=False, parent_id=main_category.pk).first()
SignalFactory.create(category_assignment__category=category)
self.assertEqual(Signal.objects.count(), 1)
# Check that the storage backend is called, and that shutil.copy is not.
csv_files = create_csv_files(isoweek=1, isoyear=2019)
self.assertEqual(len(csv_files), 7)
def test_create_csv_per_sub_category_invalid_category(self):
main_category = _get_horeca_main_category()
with self.assertRaises(ValidationError):
create_csv_per_sub_category(main_category, '/tmp', 1, 2019)
def test__get_horeca_main_category(self):
main_category = _get_horeca_main_category()
self.assertEqual(main_category.slug, 'overlast-bedrijven-en-horeca')
self.assertIsNone(main_category.parent_id)
