Added comparison operator for status codes

  Provided a convenience for quickly checking the status code of a
  response. Now, comparing a result to an integer will compare the
  status code against that integer

  If the comparison fails, a useful failure message is provided that
  also shows the full status string from the response

  Furthermore, the object being compared against could be a string. It
  just has to be a type that *can* be converted to an integer

  Also added some tests to ensure good behavior such as using the
  default comparator if the code comparison doesn't work (by virtue of
  raising an exception -- usually caused by failure to convert
  right-hand-side to an int.

Author: dusktreader

pytest_flask/plugin.py

@@ -29,6 +29,29 @@ def json(self):
         """
         return json.loads(self.data)
 
+    def __eq__(self, other):
+        try:
+            status_code = int(other)
+            return self.status_code == status_code
+        except:
+            return super(JSONResponse, self).__eq__(other)
+
+
+def pytest_assertrepr_compare(op, left, right):
+    if isinstance(left, JSONResponse) and op == '==':
+        try:
+            right = int(right)
+            return [
+                'Mismatch in status code for response: {} != {}'.format(
+                    left.status_code,
+                    right,
+                ),
+                'Response status: {}'.format(left.status),
+            ]
+        except:
+            pass
+    return None
+
 
 def _make_test_response_class(response_class):
     """Extends the response class with special attribute to test JSON

tests/test_fixtures.py

@@ -33,6 +33,19 @@ def test_json_response(self, client, accept_json):
         res = client.get(url_for('ping'), headers=accept_json)
         assert res.json == {'ping': 'pong'}
 
+    def test_json_response_compare_to_status_code(self, client, accept_json):
+        assert client.get(url_for('ping'), headers=accept_json) == 200
+        assert client.get('fake-route', headers=accept_json) == 404
+        res = client.get(url_for('ping'), headers=accept_json)
+        assert res == res
+
+    @pytest.mark.xfail(
+        strict=True,
+        reason='Assertion comparing result to non-matching int should fail',
+    )
+    def test_mismatching_eq_comparison(self, client, accept_json):
+        assert client.get('fake-route', headers=accept_json) == 200
+
     def test_dont_rewrite_existing_implementation(self, app, accept_json):
         class MyResponse(app.response_class):
             @property