Exemplo n.º 1
0
async def test_handle_response_content_type_error(nw: Network):
    response = CoroutineMock()
    response.status = 123
    response.text = CoroutineMock()
    response.text.return_value = 'some content'
    response.json = CoroutineMock()
    response.json.side_effect = ContentTypeError(None, None)

    with pytest.raises(IdexClientContentTypeError):
        await nw._handle_response(response)
Exemplo n.º 2
0
async def test_handle_response(nw: Network):
    return_value = {'some': 'response'}

    response = CoroutineMock()
    response.json = CoroutineMock()
    response.json.return_value = return_value

    nw._raise_if_error = Mock()
    nw._raise_if_error.return_value = return_value

    result = await nw._handle_response(response)

    response.json.assert_awaited_once()
    nw._raise_if_error.assert_called_once_with(return_value)

    assert result == {'some': 'response'}