test: plots in csv, html, html standalone formats (#52)

* test: plots in csv, html, html standalone formats new unit tests for plots in non-json formats. * test: invalid plot format should throw HTTP 400
alubbock · Aug 27, 2024 · ff48cc1 · ff48cc1
1 parent 43e4fc0
commit ff48cc1
Showing 1 changed file with 63 additions and 0 deletions.
diff --git a/thunorweb/tests/test_plots.py b/thunorweb/tests/test_plots.py
@@ -89,6 +89,69 @@ def test_time_course(self):
         )
         self.assertEqual(resp.status_code, HTTP_OK)
 
+    def test_time_course_csv(self):
+        self.client.force_login(self.user)
+        url = reverse('thunorweb:ajax_plot', args=['csv'])
+        resp = self.client.get(
+            url,
+            {'plotType': 'tc',
+             'datasetId': self.d.id,
+             'c': self.groupings['cellLines'][0]['id'],
+             'd': self.groupings['drugs'][0]['id'],
+             'assay': self.groupings['dipAssay'] or '',
+             'overlayDipFit': 'true',
+             'logTransform': 'log2'
+             }
+        )
+        self.assertEqual(resp.status_code, HTTP_OK)
+
+    def test_time_course_html(self):
+        self.client.force_login(self.user)
+        url = reverse('thunorweb:ajax_plot', args=['html'])
+        resp = self.client.get(
+            url,
+            {'plotType': 'tc',
+             'datasetId': self.d.id,
+             'c': self.groupings['cellLines'][0]['id'],
+             'd': self.groupings['drugs'][0]['id'],
+             'assay': self.groupings['dipAssay'] or '',
+             'overlayDipFit': 'true',
+             'logTransform': 'log2'
+             }
+        )
+        self.assertEqual(resp.status_code, HTTP_OK)
+
+    def test_time_course_html_standalone(self):
+        self.client.force_login(self.user)
+        url = reverse('thunorweb:ajax_plot', args=['html'])
+        resp = self.client.get(
+            url,
+            {'plotType': 'tc',
+             'datasetId': self.d.id,
+             'c': self.groupings['cellLines'][0]['id'],
+             'd': self.groupings['drugs'][0]['id'],
+             'assay': self.groupings['dipAssay'] or '',
+             'overlayDipFit': 'true',
+             'logTransform': 'log2',
+             'download': '1'
+             }
+        )
+        self.assertEqual(resp.status_code, HTTP_OK)
+
+    def test_invalid_format(self):
+        self.client.force_login(self.user)
+        url = reverse('thunorweb:ajax_plot', args=['pdf'])
+        resp = self.client.get(
+            url,
+            {'plotType': 'tc',
+             'datasetId': self.d.id,
+             'c': self.groupings['cellLines'][0]['id'],
+             'd': self.groupings['drugs'][0]['id'],
+             'assay': self.groupings['dipAssay'] or ''
+             }
+        )
+        self.assertEqual(resp.status_code, HTTP_INVALID_REQUEST)
+
     def test_time_course_invalid_cell_line(self):
         self.client.force_login(self.user)
         url = reverse('thunorweb:ajax_plot', args=['json'])