Python MagicMock.status_codeの例

コード例 #1

ファイル: test_vault.py プロジェクト: crimv42/saltstack

def _mock_json_response(data, status_code=200, reason=""):
    '''
    Mock helper for http response
    '''
    response = MagicMock()
    response.json = MagicMock(return_value=data)
    response.status_code = status_code
    response.reason = reason
    return Mock(return_value=response)

コード例 #2

def _mock_json_response(data, status_code=200, reason=""):
    """
    Mock helper for http response
    """
    response = MagicMock()
    response.json = MagicMock(return_value=data)
    response.status_code = status_code
    response.reason = reason
    if status_code == 200:
        response.ok = True
    else:
        response.ok = False
    return Mock(return_value=response)