Show mentors a link to download the current submissions

PeterJCLaw · PeterJCLaw · commit 30c149992fa8 · 2020-07-17T14:01:17.000+01:00
diff --git a/templates/index.html b/templates/index.html
@@ -31,6 +31,15 @@
   <body>
     <div class="container">
       <h1>Virtual Competition Code Submission</h1>
+      {% if 'blueshirt' in request.auth.scopes %}
+      <div class="row">
+        <div class="col-sm-6">
+          <a href="{{ url_for('download_submissions') }}">
+            Download current chosen submissions
+          </a>
+        </div>
+      </div>
+      {% endif %}
       <div class="row">
         {% if request.user.team %}
         <form
diff --git a/tests/tests_app.py b/tests/tests_app.py
@@ -276,6 +276,23 @@ def test_upload_archive_without_robot_py(self) -> None:
         )
         self.assertEqual([], choices, "Should not have created a choice")
 
+    def test_no_download_link_for_non_blueshirt(self) -> None:
+        download_url = self.url_for('download_submissions')
+
+        response = self.session.get(self.url_for('homepage'))
+
+        html = response.text
+        self.assertNotIn(download_url, html)
+
+    def test_shows_download_link_for_blueshirt(self) -> None:
+        self.session.auth = ('blueshirt', 'blueshirt')
+
+        download_url = self.url_for('download_submissions')
+
+        response = self.session.get(self.url_for('homepage'))
+        html = response.text
+        self.assertIn(download_url, html)
+
     def test_download_submissions_requires_blueshirt(self) -> None:
         response = self.session.get(self.url_for('download_submissions'))
         self.assertEqual(403, response.status_code)
