保留響應上下文測試燒瓶應用與pytest

Question

我測試的燒瓶中的應用與py.test用下面的代碼：

response = flask_app_test_client.post('/users', data=json.dumps(user)) 
assert response.status_code == 201 
assert response.content_type == 'application/json' 
assert isinstance(response.json, dict) 
assert set(response.json.keys()) >= {'id', 'status', 'created_at', 'updated_at'} 
assert response.json['name'] == user['name'] 
assert response.json['status'] == 'pending' 

當某些斷言失敗，我得到這樣的：

  response = test_client.post('/users', data=json.dumps(user)) 
    >  assert response.status_code == 201 
    E  assert 400 == 201 
    E  + where 400 = <JSONResponse streamed [400 BAD REQUEST]>.status_code 
    ============== 1 failed, 3 passed in 0.10 seconds =================== 

我做了很多TDD，所以我期望我的測試在開發過程中經常失敗。我的問題是斷言錯誤消息是沒有其餘的響應數據（正文，標題等）沒用。

如何打印每個失敗斷言的回覆摘要？

Answer 1

Assert statement graamar

assert response.status_code == 201, "Anything you want"

你可以只要你想詳細。您也可以使用UnitTest的helper methods - 套件，但不包含通過此位濫用的測試用例類 - https://github.com/nose-devs/nose2/blob/master/nose2/tools/such.py#L34

Answer 2

我想出了兩種不同的解決方案。

解決方案＃1：try/catch語句

try: 
    assert response.status_code == 201 
    assert response.content_type == 'application/json' 
    assert isinstance(response.json, dict) 
    assert set(response.json.keys()) >= {'id', 'status', 'created_at', 'updated_at'} 
    assert response.json['name'] == user['name'] 
    assert response.json['status'] == 'pending' 
except AssertionError as e: 
    except AssertionError as e: 
    raise ResponseAssertionError(e, response) 

class ResponseAssertionError(AssertionError): 
    def __init__(self, e, response): 
     response_dump = "\n + where full response was:\n" \ 
         "HTTP/1.1 {}\n" \ 
         "{}{}\n".format(response.status, response.headers, response.json) 

     self.args = (e.args[0] + response_dump,) 

解決方案2：不需要的try/catch（如果再版太長，有時被切斷......）

擴展和覆蓋瓶響應對象

import json 
class JSONResponse(Response): 

    def __repr__(self): 
     headers = {} 
     while len(self.headers) > 0: 
      tuple_ = self.headers.popitem() 
      headers[tuple_[0]] = tuple_[1] 

     data = { 
      'status': self.status, 
      'headers': headers, 
      'body': self.json 
     } 
     return json.dumps(data) 

和

@pytest.fixture(scope='session') 
    def test_client(flask_app): 
     flask_app.response_class = JSONResponse 
     return flask_app.test_client()