Incorrect response code.

mozilla-services · Feb 25, 2016 · dc21c44 · dc21c44
1 parent 4d38784
commit dc21c44
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/cliquet/tests/resource/test_views.py b/cliquet/tests/resource/test_views.py
@@ -393,6 +393,7 @@ def test_invalid_accept_header_on_collections_returns_406(self):
                              '',
                              headers=headers,
                              status=406)
+        self.assertEqual(resp.json['code'], 406)
         message = "Accept header should be one of ['application/json']"
         self.assertEqual(resp.json['message'], message)
 
@@ -403,6 +404,7 @@ def test_invalid_content_type_header_on_collections_returns_415(self):
                              '',
                              headers=headers,
                              status=415)
+        self.assertEqual(resp.json['code'], 415)
         message = "Content-Type header should be one of ['application/json']"
         self.assertEqual(resp.json['message'], message)
 
@@ -412,16 +414,20 @@ def test_invalid_accept_header_on_record_returns_406(self):
         resp = self.app.get(self.get_item_url(),
                             headers=headers,
                             status=406)
+        self.assertEqual(resp.json['code'], 406)
         message = "Accept header should be one of ['application/json']"
         self.assertEqual(resp.json['message'], message)
 
     def test_invalid_content_type_header_on_record_returns_415(self):
         headers = self.headers.copy()
         headers['Content-Type'] = 'text/plain'
-        self.app.patch_json(self.get_item_url(),
-                            '',
-                            headers=headers,
-                            status=415)
+        resp = self.app.patch_json(self.get_item_url(),
+                                   '',
+                                   headers=headers,
+                                   status=415)
+        self.assertEqual(resp.json['code'], 415)
+        message = "Accept header should be one of ['application/json']"
+        self.assertEqual(resp.json['message'], message)
 
 
 class IgnoredFieldsTest(BaseWebTest, unittest.TestCase):