Exemple #1
0
def test_consecutive_write_eof():
    req = make_request('GET', '/')
    data = b'data'
    resp = Response(body=data)

    yield from resp.prepare(req)
    with mock.patch('aiohttp.web.StreamResponse.write_eof') as super_write_eof:
        yield from resp.write_eof()
        resp._eof_sent = True
        yield from resp.write_eof()
        super_write_eof.assert_called_once_with(data)
Exemple #2
0
def test_consecutive_write_eof():
    req = make_request('GET', '/')
    data = b'data'
    resp = Response(body=data)

    yield from resp.prepare(req)
    with mock.patch('aiohttp.web.StreamResponse.write_eof') as super_write_eof:
        yield from resp.write_eof()
        resp._eof_sent = True
        yield from resp.write_eof()
        super_write_eof.assert_called_once_with(data)