From c505fef4f820b9662e9bdd1383814d32a12202fb Mon Sep 17 00:00:00 2001 From: Austin Riba Date: Thu, 6 Feb 2025 09:45:28 -0800 Subject: [PATCH] Add tests and redirect on GET --- tom_targets/tests/tests.py | 16 ++++++++++++++++ tom_targets/views.py | 3 +++ 2 files changed, 19 insertions(+) diff --git a/tom_targets/tests/tests.py b/tom_targets/tests/tests.py index 10ff6ee5..58fd4761 100644 --- 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 index 6c3939b1..212e44e8 100644 --- 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'))