Request application/json when getting status

MITLibraries · Jul 14, 2016 · f20f8e4 · f20f8e4
1 parent 7aebef4
commit f20f8e4
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/slingshot/app.py b/slingshot/app.py
@@ -26,7 +26,8 @@ def url(self, url):
         self._url = url.rstrip('/')
 
     def status(self, layer):
-        r = self.session.get('{}/{}'.format(self.url, layer))
+        r = self.session.get('{}/{}'.format(self.url, layer),
+                             headers={'Accept': 'application/json'})
         if r.status_code == 404:
             return None
         r.raise_for_status()

diff --git a/tests/test_app.py b/tests/test_app.py
@@ -55,6 +55,13 @@ def test_tmp_archive_removes_archive(layer):
     assert not os.path.exists(arxiv)
 
 
+def test_kepler_requests_json_for_status(kepler):
+    k = Kepler('mock://example.com/completed/')
+    k.status('47458e22-8e50-5b43-ac80-b662a1077af1')
+    assert kepler.request_history[0].headers.get('Accept') == \
+        'application/json'
+
+
 def test_kepler_returns_status(kepler):
     k = Kepler('mock://example.com/completed/')
     assert k.status('47458e22-8e50-5b43-ac80-b662a1077af1') == 'COMPLETED'