From b8d9d24e2c954745f5e04dcf58df4ba3c9f34e83 Mon Sep 17 00:00:00 2001 From: FreneticScribbler Date: Fri, 7 Feb 2020 00:17:28 +0000 Subject: [PATCH] Add delete sample data test for completeness Chasing that sweet 100% coverage... --- assets/tests/test_assets.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/assets/tests/test_assets.py b/assets/tests/test_assets.py index f0464ce1..caa5a763 100644 --- a/assets/tests/test_assets.py +++ b/assets/tests/test_assets.py @@ -129,7 +129,7 @@ class TestAssetForm(AutoLoginTest): self.assertEqual(models.Asset.objects.get(asset_id="9000").description, new_description) -# @tag('slow') TODO: Django 2 +# @tag('slow') TODO: req. Django 3.0 class TestAccessLevels(TestCase): @override_settings(DEBUG=True) def setUp(self): @@ -264,10 +264,18 @@ class TestSampleDataGenerator(TestCase): self.assertTrue(models.Asset.objects.all().count() > 50) self.assertTrue(models.Supplier.objects.all().count() > 50) + @override_settings(DEBUG=True) + def test_delete_sample_data(self): + call_command('deleteSampleData') + + self.assertTrue(models.Asset.objects.all().count() == 0) + self.assertTrue(models.Asset.objects.all().count() == 0) + def test_production_exception(self): from django.core.management.base import CommandError self.assertRaisesRegex(CommandError, ".*production", call_command, 'generateSampleAssetsData') + self.assertRaisesRegex(CommandError, ".*production", call_command, 'deleteSampleData') class TestVersioningViews(TestCase):