Ejemplo n.º 1
0
async def test_requesthandleradapter_write_response_to_client(handleradapter,
                                                              webapp):

    resp_msg = MagicMock()
    resp_msg.keep_alive = MagicMock(return_value='wololo')

    aiothttp_req = MagicMock()
    aiohttp_res = AsyncMock()
    aiohttp_res.prepare = AsyncMock(return_value=resp_msg)

    req = HttpRequestController(webapp, aiothttp_req)
    res = req.response
    res._build_aiohttp_response = MagicMock(return_value=aiohttp_res)

    handleradapter.keep_alive = MagicMock()

    await handleradapter._write_response_to_client(req, res)

    aiohttp_res.prepare.assert_called_once_with(aiothttp_req)
    aiohttp_res.write_eof.assert_called_once_with()
    handleradapter.keep_alive.assert_called_once_with("wololo")
    resp_msg.keep_alive.assert_called_once_with()