Add test cases for download with access token

Author: helenst

weblab/conftest.py

@@ -249,6 +249,22 @@ def queued_experiment(model_with_version, protocol_with_version):
     return version
 
 
+@pytest.fixture
+def queued_fittingresult(public_model, public_protocol, public_fittingspec, public_dataset):
+    version = recipes.fittingresult_version.make(
+        status='QUEUED',
+        fittingresult__model=public_model,
+        fittingresult__model_version=public_model.repocache.latest_version,
+        fittingresult__protocol=public_protocol,
+        fittingresult__protocol_version=public_protocol.repocache.latest_version,
+        fittingresult__fittingspec=public_fittingspec,
+        fittingresult__fittingspec_version=public_fittingspec.repocache.latest_version,
+        fittingresult__dataset=public_dataset,
+    )
+    recipes.running_experiment.make(runnable=version)
+    return version
+
+
 @pytest.fixture
 def experiment_with_result(model_with_version, protocol_with_version):
     version = recipes.experiment_version.make(

weblab/core/recipes.py

@@ -48,7 +48,6 @@
 
 running_experiment = Recipe('RunningExperiment', runnable=foreign_key(runnable))
 
-
 dataset = Recipe('Dataset',
                  name=seq('my dataset'),
                  protocol=foreign_key(protocol))

weblab/datasets/tests/test_views.py

@@ -346,6 +346,21 @@ def test_returns_404_for_nonexistent_file(self, my_dataset_with_file, client):
 
 @pytest.mark.django_db
 class TestDatasetArchiveView:
+    def test_anonymous_dataset_download_for_running_fittingresult(
+        self, client, queued_fittingresult, my_dataset_with_file,
+    ):
+        queued_fittingresult.fittingresult.dataset = my_dataset_with_file
+        queued_fittingresult.fittingresult.save()
+
+        response = client.get(
+            '/datasets/%d/archive' % my_dataset_with_file.pk,
+            HTTP_AUTHORIZATION='Token {}'.format(queued_fittingresult.signature)
+        )
+
+        assert response.status_code == 200
+        archive = zipfile.ZipFile(BytesIO(response.content))
+        assert archive.filelist[0].filename == 'mydataset.csv'
+
     def test_download_archive(self, my_dataset_with_file, client):
         response = client.get('/datasets/%d/archive' % my_dataset_with_file.pk)
         assert response.status_code == 200

weblab/entities/tests/test_views.py

@@ -1702,6 +1702,27 @@ def test_anonymous_protocol_download_for_running_experiment(self, client, queued
         archive = zipfile.ZipFile(BytesIO(response.content))
         assert archive.filelist[0].filename == 'file1.txt'
 
+    @pytest.mark.parametrize("entity_type,url_fragment", [
+        ('model', '/entities/models'),
+        ('protocol', '/entities/protocols'),
+        ('fittingspec', '/fitting/specs'),
+    ])
+    def test_anonymous_entity_download_for_running_fittingresult(
+        self, client, queued_fittingresult, entity_type, url_fragment
+    ):
+        entity = getattr(queued_fittingresult.fittingresult, entity_type)
+        sha = entity.repo.latest_commit.sha
+        entity.set_version_visibility(sha, 'private')
+
+        response = client.get(
+            '%s/%d/versions/latest/archive' % (url_fragment, entity.pk),
+            HTTP_AUTHORIZATION='Token {}'.format(queued_fittingresult.signature)
+        )
+
+        assert response.status_code == 200
+        archive = zipfile.ZipFile(BytesIO(response.content))
+        assert archive.filelist[0].filename == 'file1.txt'
+
     def test_anonymous_protocol_download_for_analysis_task(self, client, analysis_task):
         protocol = analysis_task.entity
         protocol.set_version_visibility('latest', 'private')