Add a couple of explicit tests

observatorycontrolsystem · Oct 9, 2024 · 5cc79e5 · 5cc79e5
1 parent 03e182e
commit 5cc79e5
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/archive/frames/tests/test_views.py b/archive/frames/tests/test_views.py
@@ -41,11 +41,26 @@ def setUp(self):
     def test_get_frame(self):
         response = self.client.get(reverse('frame-detail', args=(self.frame.id, )))
         self.assertEqual(response.json()['basename'], self.frame.basename)
+        self.assertContains(response, 'related_frames')
+
+    def test_get_frame_exclude_related_frames(self):
+        response = self.client.get(reverse('frame-detail', args=(self.frame.id,)),
+                                   {'include_related_frames': False})
+
+        self.assertNotContains(response, 'related_frames')
 
     def test_get_frame_list(self):
         response = self.client.get(reverse('frame-list'))
         self.assertEqual(response.json()['count'], 5)
         self.assertContains(response, self.frame.basename)
+        self.assertContains(response, 'related_frames')
+
+    def test_get_frame_list_exclude_related_frames(self):
+        response = self.client.get(reverse('frame-list'),
+                                   {'include_related_frames': False})
+        self.assertEqual(response.json()['count'], 5)
+        self.assertContains(response, self.frame.basename)
+        self.assertNotContains(response, 'related_frames')
 
     def test_get_frame_list_filter(self):
         response = self.client.get(