sepalate the test of extended_download

c-3lab · Mar 27, 2023 · 9380b43 · 9380b43
1 parent 1c6f7b8
commit 9380b43
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/ckanext/feedback/tests/controllers/test_download.py b/ckanext/feedback/tests/controllers/test_download.py
@@ -38,17 +38,22 @@ def test_extended_download(self, mocker):
         resource = factories.Resource()
         self.app = Flask(__name__)
         download = mocker.patch('ckanext.feedback.controllers.download.download')
-
-        with self.app.test_request_context(headers={'Sec-Fetch-Dest': 'image'}):
+        with self.app.test_request_context(headers={'Sec-Fetch-Dest': 'document'}):
             DownloadController.extended_download(
                 'package_type', resource['package_id'], resource['id'], None
             )
-            assert get_downloads(resource['id']) is None
+            assert get_downloads(resource['id']) == 1
             assert download
 
-        with self.app.test_request_context(headers={'Sec-Fetch-Dest': 'document'}):
+    @patch('ckanext.feedback.controllers.download.download')
+    def test_extended_download_with_preview(self, mocker):
+        resource = factories.Resource()
+        self.app = Flask(__name__)
+        download = mocker.patch('ckanext.feedback.controllers.download.download')
+
+        with self.app.test_request_context(headers={'Sec-Fetch-Dest': 'image'}):
             DownloadController.extended_download(
                 'package_type', resource['package_id'], resource['id'], None
             )
-            assert get_downloads(resource['id']) == 1
+            assert get_downloads(resource['id']) is None
             assert download