Add tests and redirect on GET

TOMToolkit · Feb 6, 2025 · c505fef · c505fef
1 parent 48844bb
commit c505fef
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 0 deletions.
diff --git a/tom_targets/tests/tests.py b/tom_targets/tests/tests.py
@@ -1930,3 +1930,19 @@ def test_merge_observation_records(self):
         target_merge(self.st1, self.st2)
         for observationrecord in st2_observationrecords:
             self.assertIn(observationrecord, ObservationRecord.objects.filter(target=self.st1))
+
+
+class TestTargetSeed(TestCase):
+    def test_seed_targets_authenticated(self):
+        user = User.objects.create(username='testuser')
+        self.client.force_login(user)
+        self.assertFalse(Target.objects.exists())
+        response = self.client.post(reverse('targets:seed'))
+        self.assertEqual(response.status_code, 302)
+        self.assertTrue(Target.objects.exists())
+
+    def test_seed_targets_unauthenticated(self):
+        self.assertFalse(Target.objects.exists())
+        response = self.client.post(reverse('targets:seed'))
+        self.assertEqual(response.status_code, 302)
+        self.assertFalse(Target.objects.exists())
diff --git a/tom_targets/views.py b/tom_targets/views.py
@@ -929,3 +929,6 @@ class TargetSeedView(LoginRequiredMixin, View):
     def post(self, request, *args, **kwargs):
         seed_messier_targets()
         return redirect(reverse('targets:list'))
+
+    def get(self, request, *args, **kwargs):
+        return redirect(reverse('targets:list'))