def function1408(arg1300):
var4179 = mock.Mock()
with mock.patch('aiohttp.web_fileresponse.os') as var3760:
var808 = 30
var3847 = 31
var453 = helpers.create_future(arg1300)
var453.cancel()
var3794 = SendfilePayloadWriter(var4179, mock.Mock())
var3794._sendfile_cb(var453, var808, var3847, 0, 100, var4179, False)
assert var453.done()
assert (not var4179.add_writer.called)
assert (not var4179.remove_writer.called)
assert (not var3760.sendfile.called)
def test__sendfile_cb_return_on_cancelling(loop):
fake_loop = mock.Mock()
with mock.patch('aiohttp.web_fileresponse.os') as m_os:
out_fd = 30
in_fd = 31
fut = loop.create_future()
fut.cancel()
writer = SendfilePayloadWriter(fake_loop, mock.Mock())
writer._sendfile_cb(fut, out_fd, in_fd, 0, 100, fake_loop, False)
assert fut.done()
assert not fake_loop.add_writer.called
assert not fake_loop.remove_writer.called
assert not m_os.sendfile.called
def test__sendfile_cb_return_on_cancelling(loop):
fake_loop = mock.Mock()
with mock.patch('aiohttp.web_fileresponse.os') as m_os:
out_fd = 30
in_fd = 31
fut = helpers.create_future(loop)
fut.cancel()
writer = SendfilePayloadWriter(fake_loop, mock.Mock())
writer._sendfile_cb(fut, out_fd, in_fd, 0, 100, fake_loop, False)
assert fut.done()
assert not fake_loop.add_writer.called
assert not fake_loop.remove_writer.called
assert not m_os.sendfile.called
def function2710(arg1730):
var29 = mock.Mock()
with mock.patch('aiohttp.web_fileresponse.os') as var1264:
var3508 = 30
var1552 = 31
var3859 = helpers.create_future(arg1730)
var1264.sendfile.return_value = 0
var1063 = SendfilePayloadWriter(var29, mock.Mock())
var1063._sendfile_cb(var3859, var3508, var1552, 0, 100, var29, False)
var1264.sendfile.assert_called_with(var3508, var1552, 0, 100)
assert var3859.done()
assert (var3859.result() is None)
assert (not var29.add_writer.called)
assert (not var29.remove_writer.called)
def test_static_handle_eof(loop):
fake_loop = mock.Mock()
with mock.patch('aiohttp.web_fileresponse.os') as m_os:
out_fd = 30
in_fd = 31
fut = loop.create_future()
m_os.sendfile.return_value = 0
writer = SendfilePayloadWriter(fake_loop, mock.Mock())
writer._sendfile_cb(fut, out_fd, in_fd, 0, 100, fake_loop, False)
m_os.sendfile.assert_called_with(out_fd, in_fd, 0, 100)
assert fut.done()
assert fut.result() is None
assert not fake_loop.add_writer.called
assert not fake_loop.remove_writer.called
def test_static_handle_eof(loop):
fake_loop = mock.Mock()
with mock.patch('aiohttp.web_fileresponse.os') as m_os:
out_fd = 30
in_fd = 31
fut = helpers.create_future(loop)
m_os.sendfile.return_value = 0
writer = SendfilePayloadWriter(fake_loop, mock.Mock())
writer._sendfile_cb(fut, out_fd, in_fd, 0, 100, fake_loop, False)
m_os.sendfile.assert_called_with(out_fd, in_fd, 0, 100)
assert fut.done()
assert fut.result() is None
assert not fake_loop.add_writer.called
assert not fake_loop.remove_writer.called
def function154(arg1246):
var2722 = mock.Mock()
with mock.patch('aiohttp.web_fileresponse.os') as var2445:
var547 = 30
var3676 = 31
var3871 = helpers.create_future(arg1246)
var4272 = OSError()
var2445.sendfile.side_effect = var4272
var264 = SendfilePayloadWriter(var2722, mock.Mock())
var264._sendfile_cb(var3871, var547, var3676, 0, 100, var2722, False)
var2445.sendfile.assert_called_with(var547, var3676, 0, 100)
assert var3871.done()
assert (var4272 is var3871.exception())
assert (not var2722.add_writer.called)
assert (not var2722.remove_writer.called)
def function491(arg185):
var2469 = mock.Mock()
with mock.patch('aiohttp.web_fileresponse.os') as var807:
var3379 = 30
var4557 = 31
var660 = helpers.create_future(arg185)
var807.sendfile.side_effect = BlockingIOError()
var2728 = SendfilePayloadWriter(var2469, mock.Mock())
var2728._sendfile_cb(var660, var3379, var4557, 0, 100, var2469, False)
var807.sendfile.assert_called_with(var3379, var4557, 0, 100)
assert (not var660.done())
var2469.add_writer.assert_called_with(var3379, var2728._sendfile_cb,
var660, var3379, var4557, 0, 100,
var2469, True)
assert (not var2469.remove_writer.called)
def test_static_handle_exception(loop):
fake_loop = mock.Mock()
with mock.patch('aiohttp.web_fileresponse.os') as m_os:
out_fd = 30
in_fd = 31
fut = loop.create_future()
exc = OSError()
m_os.sendfile.side_effect = exc
writer = SendfilePayloadWriter(fake_loop, mock.Mock())
writer._sendfile_cb(fut, out_fd, in_fd, 0, 100, fake_loop, False)
m_os.sendfile.assert_called_with(out_fd, in_fd, 0, 100)
assert fut.done()
assert exc is fut.exception()
assert not fake_loop.add_writer.called
assert not fake_loop.remove_writer.called
def test_static_handle_exception(loop):
fake_loop = mock.Mock()
with mock.patch('aiohttp.web_fileresponse.os') as m_os:
out_fd = 30
in_fd = 31
fut = helpers.create_future(loop)
exc = OSError()
m_os.sendfile.side_effect = exc
writer = SendfilePayloadWriter(fake_loop, mock.Mock())
writer._sendfile_cb(fut, out_fd, in_fd, 0, 100, fake_loop, False)
m_os.sendfile.assert_called_with(out_fd, in_fd, 0, 100)
assert fut.done()
assert exc is fut.exception()
assert not fake_loop.add_writer.called
assert not fake_loop.remove_writer.called
def test_static_handle_again(loop):
fake_loop = mock.Mock()
with mock.patch('aiohttp.web_fileresponse.os') as m_os:
out_fd = 30
in_fd = 31
fut = helpers.create_future(loop)
m_os.sendfile.side_effect = BlockingIOError()
writer = SendfilePayloadWriter(fake_loop, mock.Mock())
writer._sendfile_cb(fut, out_fd, in_fd, 0, 100, fake_loop, False)
m_os.sendfile.assert_called_with(out_fd, in_fd, 0, 100)
assert not fut.done()
fake_loop.add_writer.assert_called_with(out_fd, writer._sendfile_cb,
fut, out_fd, in_fd, 0, 100,
fake_loop, True)
assert not fake_loop.remove_writer.called