mirror of https://github.com/python/cpython
1977 lines
71 KiB
Python
1977 lines
71 KiB
Python
"""Tests for base_events.py"""
|
|
|
|
import errno
|
|
import logging
|
|
import math
|
|
import os
|
|
import socket
|
|
import sys
|
|
import threading
|
|
import time
|
|
import unittest
|
|
from unittest import mock
|
|
|
|
import asyncio
|
|
from asyncio import base_events
|
|
from asyncio import constants
|
|
from asyncio import events
|
|
from test.test_asyncio import utils as test_utils
|
|
from test import support
|
|
from test.support.script_helper import assert_python_ok
|
|
|
|
|
|
MOCK_ANY = mock.ANY
|
|
PY34 = sys.version_info >= (3, 4)
|
|
|
|
|
|
def tearDownModule():
|
|
asyncio.set_event_loop_policy(None)
|
|
|
|
|
|
def mock_socket_module():
|
|
m_socket = mock.MagicMock(spec=socket)
|
|
for name in (
|
|
'AF_INET', 'AF_INET6', 'AF_UNSPEC', 'IPPROTO_TCP', 'IPPROTO_UDP',
|
|
'SOCK_STREAM', 'SOCK_DGRAM', 'SOL_SOCKET', 'SO_REUSEADDR', 'inet_pton'
|
|
):
|
|
if hasattr(socket, name):
|
|
setattr(m_socket, name, getattr(socket, name))
|
|
else:
|
|
delattr(m_socket, name)
|
|
|
|
m_socket.socket = mock.MagicMock()
|
|
m_socket.socket.return_value = test_utils.mock_nonblocking_socket()
|
|
m_socket.getaddrinfo._is_coroutine = False
|
|
|
|
return m_socket
|
|
|
|
|
|
def patch_socket(f):
|
|
return mock.patch('asyncio.base_events.socket',
|
|
new_callable=mock_socket_module)(f)
|
|
|
|
|
|
class BaseEventTests(test_utils.TestCase):
|
|
|
|
def test_ipaddr_info(self):
|
|
UNSPEC = socket.AF_UNSPEC
|
|
INET = socket.AF_INET
|
|
INET6 = socket.AF_INET6
|
|
STREAM = socket.SOCK_STREAM
|
|
DGRAM = socket.SOCK_DGRAM
|
|
TCP = socket.IPPROTO_TCP
|
|
UDP = socket.IPPROTO_UDP
|
|
|
|
self.assertEqual(
|
|
(INET, STREAM, TCP, '', ('1.2.3.4', 1)),
|
|
base_events._ipaddr_info('1.2.3.4', 1, INET, STREAM, TCP))
|
|
|
|
self.assertEqual(
|
|
(INET, STREAM, TCP, '', ('1.2.3.4', 1)),
|
|
base_events._ipaddr_info(b'1.2.3.4', 1, INET, STREAM, TCP))
|
|
|
|
self.assertEqual(
|
|
(INET, STREAM, TCP, '', ('1.2.3.4', 1)),
|
|
base_events._ipaddr_info('1.2.3.4', 1, UNSPEC, STREAM, TCP))
|
|
|
|
self.assertEqual(
|
|
(INET, DGRAM, UDP, '', ('1.2.3.4', 1)),
|
|
base_events._ipaddr_info('1.2.3.4', 1, UNSPEC, DGRAM, UDP))
|
|
|
|
# Socket type STREAM implies TCP protocol.
|
|
self.assertEqual(
|
|
(INET, STREAM, TCP, '', ('1.2.3.4', 1)),
|
|
base_events._ipaddr_info('1.2.3.4', 1, UNSPEC, STREAM, 0))
|
|
|
|
# Socket type DGRAM implies UDP protocol.
|
|
self.assertEqual(
|
|
(INET, DGRAM, UDP, '', ('1.2.3.4', 1)),
|
|
base_events._ipaddr_info('1.2.3.4', 1, UNSPEC, DGRAM, 0))
|
|
|
|
# No socket type.
|
|
self.assertIsNone(
|
|
base_events._ipaddr_info('1.2.3.4', 1, UNSPEC, 0, 0))
|
|
|
|
# IPv4 address with family IPv6.
|
|
self.assertIsNone(
|
|
base_events._ipaddr_info('1.2.3.4', 1, INET6, STREAM, TCP))
|
|
|
|
self.assertEqual(
|
|
(INET6, STREAM, TCP, '', ('::3', 1)),
|
|
base_events._ipaddr_info('::3', 1, INET6, STREAM, TCP))
|
|
|
|
self.assertEqual(
|
|
(INET6, STREAM, TCP, '', ('::3', 1)),
|
|
base_events._ipaddr_info('::3', 1, UNSPEC, STREAM, TCP))
|
|
|
|
# IPv6 address with family IPv4.
|
|
self.assertIsNone(
|
|
base_events._ipaddr_info('::3', 1, INET, STREAM, TCP))
|
|
|
|
# IPv6 address with zone index.
|
|
self.assertIsNone(
|
|
base_events._ipaddr_info('::3%lo0', 1, INET6, STREAM, TCP))
|
|
|
|
def test_port_parameter_types(self):
|
|
# Test obscure kinds of arguments for "port".
|
|
INET = socket.AF_INET
|
|
STREAM = socket.SOCK_STREAM
|
|
TCP = socket.IPPROTO_TCP
|
|
|
|
self.assertEqual(
|
|
(INET, STREAM, TCP, '', ('1.2.3.4', 0)),
|
|
base_events._ipaddr_info('1.2.3.4', None, INET, STREAM, TCP))
|
|
|
|
self.assertEqual(
|
|
(INET, STREAM, TCP, '', ('1.2.3.4', 0)),
|
|
base_events._ipaddr_info('1.2.3.4', b'', INET, STREAM, TCP))
|
|
|
|
self.assertEqual(
|
|
(INET, STREAM, TCP, '', ('1.2.3.4', 0)),
|
|
base_events._ipaddr_info('1.2.3.4', '', INET, STREAM, TCP))
|
|
|
|
self.assertEqual(
|
|
(INET, STREAM, TCP, '', ('1.2.3.4', 1)),
|
|
base_events._ipaddr_info('1.2.3.4', '1', INET, STREAM, TCP))
|
|
|
|
self.assertEqual(
|
|
(INET, STREAM, TCP, '', ('1.2.3.4', 1)),
|
|
base_events._ipaddr_info('1.2.3.4', b'1', INET, STREAM, TCP))
|
|
|
|
@patch_socket
|
|
def test_ipaddr_info_no_inet_pton(self, m_socket):
|
|
del m_socket.inet_pton
|
|
self.assertIsNone(base_events._ipaddr_info('1.2.3.4', 1,
|
|
socket.AF_INET,
|
|
socket.SOCK_STREAM,
|
|
socket.IPPROTO_TCP))
|
|
|
|
|
|
class BaseEventLoopTests(test_utils.TestCase):
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
self.loop = base_events.BaseEventLoop()
|
|
self.loop._selector = mock.Mock()
|
|
self.loop._selector.select.return_value = ()
|
|
self.set_event_loop(self.loop)
|
|
|
|
def test_not_implemented(self):
|
|
m = mock.Mock()
|
|
self.assertRaises(
|
|
NotImplementedError,
|
|
self.loop._make_socket_transport, m, m)
|
|
self.assertRaises(
|
|
NotImplementedError,
|
|
self.loop._make_ssl_transport, m, m, m, m)
|
|
self.assertRaises(
|
|
NotImplementedError,
|
|
self.loop._make_datagram_transport, m, m)
|
|
self.assertRaises(
|
|
NotImplementedError, self.loop._process_events, [])
|
|
self.assertRaises(
|
|
NotImplementedError, self.loop._write_to_self)
|
|
self.assertRaises(
|
|
NotImplementedError,
|
|
self.loop._make_read_pipe_transport, m, m)
|
|
self.assertRaises(
|
|
NotImplementedError,
|
|
self.loop._make_write_pipe_transport, m, m)
|
|
gen = self.loop._make_subprocess_transport(m, m, m, m, m, m, m)
|
|
with self.assertRaises(NotImplementedError):
|
|
gen.send(None)
|
|
|
|
def test_close(self):
|
|
self.assertFalse(self.loop.is_closed())
|
|
self.loop.close()
|
|
self.assertTrue(self.loop.is_closed())
|
|
|
|
# it should be possible to call close() more than once
|
|
self.loop.close()
|
|
self.loop.close()
|
|
|
|
# operation blocked when the loop is closed
|
|
f = asyncio.Future(loop=self.loop)
|
|
self.assertRaises(RuntimeError, self.loop.run_forever)
|
|
self.assertRaises(RuntimeError, self.loop.run_until_complete, f)
|
|
|
|
def test__add_callback_handle(self):
|
|
h = asyncio.Handle(lambda: False, (), self.loop, None)
|
|
|
|
self.loop._add_callback(h)
|
|
self.assertFalse(self.loop._scheduled)
|
|
self.assertIn(h, self.loop._ready)
|
|
|
|
def test__add_callback_cancelled_handle(self):
|
|
h = asyncio.Handle(lambda: False, (), self.loop, None)
|
|
h.cancel()
|
|
|
|
self.loop._add_callback(h)
|
|
self.assertFalse(self.loop._scheduled)
|
|
self.assertFalse(self.loop._ready)
|
|
|
|
def test_set_default_executor(self):
|
|
executor = mock.Mock()
|
|
self.loop.set_default_executor(executor)
|
|
self.assertIs(executor, self.loop._default_executor)
|
|
|
|
def test_call_soon(self):
|
|
def cb():
|
|
pass
|
|
|
|
h = self.loop.call_soon(cb)
|
|
self.assertEqual(h._callback, cb)
|
|
self.assertIsInstance(h, asyncio.Handle)
|
|
self.assertIn(h, self.loop._ready)
|
|
|
|
def test_call_soon_non_callable(self):
|
|
self.loop.set_debug(True)
|
|
with self.assertRaisesRegex(TypeError, 'a callable object'):
|
|
self.loop.call_soon(1)
|
|
|
|
def test_call_later(self):
|
|
def cb():
|
|
pass
|
|
|
|
h = self.loop.call_later(10.0, cb)
|
|
self.assertIsInstance(h, asyncio.TimerHandle)
|
|
self.assertIn(h, self.loop._scheduled)
|
|
self.assertNotIn(h, self.loop._ready)
|
|
|
|
def test_call_later_negative_delays(self):
|
|
calls = []
|
|
|
|
def cb(arg):
|
|
calls.append(arg)
|
|
|
|
self.loop._process_events = mock.Mock()
|
|
self.loop.call_later(-1, cb, 'a')
|
|
self.loop.call_later(-2, cb, 'b')
|
|
test_utils.run_briefly(self.loop)
|
|
self.assertEqual(calls, ['b', 'a'])
|
|
|
|
def test_time_and_call_at(self):
|
|
def cb():
|
|
self.loop.stop()
|
|
|
|
self.loop._process_events = mock.Mock()
|
|
delay = 0.1
|
|
|
|
when = self.loop.time() + delay
|
|
self.loop.call_at(when, cb)
|
|
t0 = self.loop.time()
|
|
self.loop.run_forever()
|
|
dt = self.loop.time() - t0
|
|
|
|
# 50 ms: maximum granularity of the event loop
|
|
self.assertGreaterEqual(dt, delay - 0.050, dt)
|
|
# tolerate a difference of +800 ms because some Python buildbots
|
|
# are really slow
|
|
self.assertLessEqual(dt, 0.9, dt)
|
|
|
|
def check_thread(self, loop, debug):
|
|
def cb():
|
|
pass
|
|
|
|
loop.set_debug(debug)
|
|
if debug:
|
|
msg = ("Non-thread-safe operation invoked on an event loop other "
|
|
"than the current one")
|
|
with self.assertRaisesRegex(RuntimeError, msg):
|
|
loop.call_soon(cb)
|
|
with self.assertRaisesRegex(RuntimeError, msg):
|
|
loop.call_later(60, cb)
|
|
with self.assertRaisesRegex(RuntimeError, msg):
|
|
loop.call_at(loop.time() + 60, cb)
|
|
else:
|
|
loop.call_soon(cb)
|
|
loop.call_later(60, cb)
|
|
loop.call_at(loop.time() + 60, cb)
|
|
|
|
def test_check_thread(self):
|
|
def check_in_thread(loop, event, debug, create_loop, fut):
|
|
# wait until the event loop is running
|
|
event.wait()
|
|
|
|
try:
|
|
if create_loop:
|
|
loop2 = base_events.BaseEventLoop()
|
|
try:
|
|
asyncio.set_event_loop(loop2)
|
|
self.check_thread(loop, debug)
|
|
finally:
|
|
asyncio.set_event_loop(None)
|
|
loop2.close()
|
|
else:
|
|
self.check_thread(loop, debug)
|
|
except Exception as exc:
|
|
loop.call_soon_threadsafe(fut.set_exception, exc)
|
|
else:
|
|
loop.call_soon_threadsafe(fut.set_result, None)
|
|
|
|
def test_thread(loop, debug, create_loop=False):
|
|
event = threading.Event()
|
|
fut = asyncio.Future(loop=loop)
|
|
loop.call_soon(event.set)
|
|
args = (loop, event, debug, create_loop, fut)
|
|
thread = threading.Thread(target=check_in_thread, args=args)
|
|
thread.start()
|
|
loop.run_until_complete(fut)
|
|
thread.join()
|
|
|
|
self.loop._process_events = mock.Mock()
|
|
self.loop._write_to_self = mock.Mock()
|
|
|
|
# raise RuntimeError if the thread has no event loop
|
|
test_thread(self.loop, True)
|
|
|
|
# check disabled if debug mode is disabled
|
|
test_thread(self.loop, False)
|
|
|
|
# raise RuntimeError if the event loop of the thread is not the called
|
|
# event loop
|
|
test_thread(self.loop, True, create_loop=True)
|
|
|
|
# check disabled if debug mode is disabled
|
|
test_thread(self.loop, False, create_loop=True)
|
|
|
|
def test__run_once(self):
|
|
h1 = asyncio.TimerHandle(time.monotonic() + 5.0, lambda: True, (),
|
|
self.loop, None)
|
|
h2 = asyncio.TimerHandle(time.monotonic() + 10.0, lambda: True, (),
|
|
self.loop, None)
|
|
|
|
h1.cancel()
|
|
|
|
self.loop._process_events = mock.Mock()
|
|
self.loop._scheduled.append(h1)
|
|
self.loop._scheduled.append(h2)
|
|
self.loop._run_once()
|
|
|
|
t = self.loop._selector.select.call_args[0][0]
|
|
self.assertTrue(9.5 < t < 10.5, t)
|
|
self.assertEqual([h2], self.loop._scheduled)
|
|
self.assertTrue(self.loop._process_events.called)
|
|
|
|
def test_set_debug(self):
|
|
self.loop.set_debug(True)
|
|
self.assertTrue(self.loop.get_debug())
|
|
self.loop.set_debug(False)
|
|
self.assertFalse(self.loop.get_debug())
|
|
|
|
@mock.patch('asyncio.base_events.logger')
|
|
def test__run_once_logging(self, m_logger):
|
|
def slow_select(timeout):
|
|
# Sleep a bit longer than a second to avoid timer resolution
|
|
# issues.
|
|
time.sleep(1.1)
|
|
return []
|
|
|
|
# logging needs debug flag
|
|
self.loop.set_debug(True)
|
|
|
|
# Log to INFO level if timeout > 1.0 sec.
|
|
self.loop._selector.select = slow_select
|
|
self.loop._process_events = mock.Mock()
|
|
self.loop._run_once()
|
|
self.assertEqual(logging.INFO, m_logger.log.call_args[0][0])
|
|
|
|
def fast_select(timeout):
|
|
time.sleep(0.001)
|
|
return []
|
|
|
|
self.loop._selector.select = fast_select
|
|
self.loop._run_once()
|
|
self.assertEqual(logging.DEBUG, m_logger.log.call_args[0][0])
|
|
|
|
def test__run_once_schedule_handle(self):
|
|
handle = None
|
|
processed = False
|
|
|
|
def cb(loop):
|
|
nonlocal processed, handle
|
|
processed = True
|
|
handle = loop.call_soon(lambda: True)
|
|
|
|
h = asyncio.TimerHandle(time.monotonic() - 1, cb, (self.loop,),
|
|
self.loop, None)
|
|
|
|
self.loop._process_events = mock.Mock()
|
|
self.loop._scheduled.append(h)
|
|
self.loop._run_once()
|
|
|
|
self.assertTrue(processed)
|
|
self.assertEqual([handle], list(self.loop._ready))
|
|
|
|
def test__run_once_cancelled_event_cleanup(self):
|
|
self.loop._process_events = mock.Mock()
|
|
|
|
self.assertTrue(
|
|
0 < base_events._MIN_CANCELLED_TIMER_HANDLES_FRACTION < 1.0)
|
|
|
|
def cb():
|
|
pass
|
|
|
|
# Set up one "blocking" event that will not be cancelled to
|
|
# ensure later cancelled events do not make it to the head
|
|
# of the queue and get cleaned.
|
|
not_cancelled_count = 1
|
|
self.loop.call_later(3000, cb)
|
|
|
|
# Add less than threshold (base_events._MIN_SCHEDULED_TIMER_HANDLES)
|
|
# cancelled handles, ensure they aren't removed
|
|
|
|
cancelled_count = 2
|
|
for x in range(2):
|
|
h = self.loop.call_later(3600, cb)
|
|
h.cancel()
|
|
|
|
# Add some cancelled events that will be at head and removed
|
|
cancelled_count += 2
|
|
for x in range(2):
|
|
h = self.loop.call_later(100, cb)
|
|
h.cancel()
|
|
|
|
# This test is invalid if _MIN_SCHEDULED_TIMER_HANDLES is too low
|
|
self.assertLessEqual(cancelled_count + not_cancelled_count,
|
|
base_events._MIN_SCHEDULED_TIMER_HANDLES)
|
|
|
|
self.assertEqual(self.loop._timer_cancelled_count, cancelled_count)
|
|
|
|
self.loop._run_once()
|
|
|
|
cancelled_count -= 2
|
|
|
|
self.assertEqual(self.loop._timer_cancelled_count, cancelled_count)
|
|
|
|
self.assertEqual(len(self.loop._scheduled),
|
|
cancelled_count + not_cancelled_count)
|
|
|
|
# Need enough events to pass _MIN_CANCELLED_TIMER_HANDLES_FRACTION
|
|
# so that deletion of cancelled events will occur on next _run_once
|
|
add_cancel_count = int(math.ceil(
|
|
base_events._MIN_SCHEDULED_TIMER_HANDLES *
|
|
base_events._MIN_CANCELLED_TIMER_HANDLES_FRACTION)) + 1
|
|
|
|
add_not_cancel_count = max(base_events._MIN_SCHEDULED_TIMER_HANDLES -
|
|
add_cancel_count, 0)
|
|
|
|
# Add some events that will not be cancelled
|
|
not_cancelled_count += add_not_cancel_count
|
|
for x in range(add_not_cancel_count):
|
|
self.loop.call_later(3600, cb)
|
|
|
|
# Add enough cancelled events
|
|
cancelled_count += add_cancel_count
|
|
for x in range(add_cancel_count):
|
|
h = self.loop.call_later(3600, cb)
|
|
h.cancel()
|
|
|
|
# Ensure all handles are still scheduled
|
|
self.assertEqual(len(self.loop._scheduled),
|
|
cancelled_count + not_cancelled_count)
|
|
|
|
self.loop._run_once()
|
|
|
|
# Ensure cancelled events were removed
|
|
self.assertEqual(len(self.loop._scheduled), not_cancelled_count)
|
|
|
|
# Ensure only uncancelled events remain scheduled
|
|
self.assertTrue(all([not x._cancelled for x in self.loop._scheduled]))
|
|
|
|
def test_run_until_complete_type_error(self):
|
|
self.assertRaises(TypeError,
|
|
self.loop.run_until_complete, 'blah')
|
|
|
|
def test_run_until_complete_loop(self):
|
|
task = asyncio.Future(loop=self.loop)
|
|
other_loop = self.new_test_loop()
|
|
self.addCleanup(other_loop.close)
|
|
self.assertRaises(ValueError,
|
|
other_loop.run_until_complete, task)
|
|
|
|
def test_run_until_complete_loop_orphan_future_close_loop(self):
|
|
class ShowStopper(BaseException):
|
|
pass
|
|
|
|
async def foo(delay):
|
|
await asyncio.sleep(delay, loop=self.loop)
|
|
|
|
def throw():
|
|
raise ShowStopper
|
|
|
|
self.loop._process_events = mock.Mock()
|
|
self.loop.call_soon(throw)
|
|
try:
|
|
self.loop.run_until_complete(foo(0.1))
|
|
except ShowStopper:
|
|
pass
|
|
|
|
# This call fails if run_until_complete does not clean up
|
|
# done-callback for the previous future.
|
|
self.loop.run_until_complete(foo(0.2))
|
|
|
|
def test_subprocess_exec_invalid_args(self):
|
|
args = [sys.executable, '-c', 'pass']
|
|
|
|
# missing program parameter (empty args)
|
|
self.assertRaises(TypeError,
|
|
self.loop.run_until_complete, self.loop.subprocess_exec,
|
|
asyncio.SubprocessProtocol)
|
|
|
|
# expected multiple arguments, not a list
|
|
self.assertRaises(TypeError,
|
|
self.loop.run_until_complete, self.loop.subprocess_exec,
|
|
asyncio.SubprocessProtocol, args)
|
|
|
|
# program arguments must be strings, not int
|
|
self.assertRaises(TypeError,
|
|
self.loop.run_until_complete, self.loop.subprocess_exec,
|
|
asyncio.SubprocessProtocol, sys.executable, 123)
|
|
|
|
# universal_newlines, shell, bufsize must not be set
|
|
self.assertRaises(TypeError,
|
|
self.loop.run_until_complete, self.loop.subprocess_exec,
|
|
asyncio.SubprocessProtocol, *args, universal_newlines=True)
|
|
self.assertRaises(TypeError,
|
|
self.loop.run_until_complete, self.loop.subprocess_exec,
|
|
asyncio.SubprocessProtocol, *args, shell=True)
|
|
self.assertRaises(TypeError,
|
|
self.loop.run_until_complete, self.loop.subprocess_exec,
|
|
asyncio.SubprocessProtocol, *args, bufsize=4096)
|
|
|
|
def test_subprocess_shell_invalid_args(self):
|
|
# expected a string, not an int or a list
|
|
self.assertRaises(TypeError,
|
|
self.loop.run_until_complete, self.loop.subprocess_shell,
|
|
asyncio.SubprocessProtocol, 123)
|
|
self.assertRaises(TypeError,
|
|
self.loop.run_until_complete, self.loop.subprocess_shell,
|
|
asyncio.SubprocessProtocol, [sys.executable, '-c', 'pass'])
|
|
|
|
# universal_newlines, shell, bufsize must not be set
|
|
self.assertRaises(TypeError,
|
|
self.loop.run_until_complete, self.loop.subprocess_shell,
|
|
asyncio.SubprocessProtocol, 'exit 0', universal_newlines=True)
|
|
self.assertRaises(TypeError,
|
|
self.loop.run_until_complete, self.loop.subprocess_shell,
|
|
asyncio.SubprocessProtocol, 'exit 0', shell=True)
|
|
self.assertRaises(TypeError,
|
|
self.loop.run_until_complete, self.loop.subprocess_shell,
|
|
asyncio.SubprocessProtocol, 'exit 0', bufsize=4096)
|
|
|
|
def test_default_exc_handler_callback(self):
|
|
self.loop._process_events = mock.Mock()
|
|
|
|
def zero_error(fut):
|
|
fut.set_result(True)
|
|
1/0
|
|
|
|
# Test call_soon (events.Handle)
|
|
with mock.patch('asyncio.base_events.logger') as log:
|
|
fut = asyncio.Future(loop=self.loop)
|
|
self.loop.call_soon(zero_error, fut)
|
|
fut.add_done_callback(lambda fut: self.loop.stop())
|
|
self.loop.run_forever()
|
|
log.error.assert_called_with(
|
|
test_utils.MockPattern('Exception in callback.*zero'),
|
|
exc_info=(ZeroDivisionError, MOCK_ANY, MOCK_ANY))
|
|
|
|
# Test call_later (events.TimerHandle)
|
|
with mock.patch('asyncio.base_events.logger') as log:
|
|
fut = asyncio.Future(loop=self.loop)
|
|
self.loop.call_later(0.01, zero_error, fut)
|
|
fut.add_done_callback(lambda fut: self.loop.stop())
|
|
self.loop.run_forever()
|
|
log.error.assert_called_with(
|
|
test_utils.MockPattern('Exception in callback.*zero'),
|
|
exc_info=(ZeroDivisionError, MOCK_ANY, MOCK_ANY))
|
|
|
|
def test_default_exc_handler_coro(self):
|
|
self.loop._process_events = mock.Mock()
|
|
|
|
@asyncio.coroutine
|
|
def zero_error_coro():
|
|
yield from asyncio.sleep(0.01, loop=self.loop)
|
|
1/0
|
|
|
|
# Test Future.__del__
|
|
with mock.patch('asyncio.base_events.logger') as log:
|
|
fut = asyncio.ensure_future(zero_error_coro(), loop=self.loop)
|
|
fut.add_done_callback(lambda *args: self.loop.stop())
|
|
self.loop.run_forever()
|
|
fut = None # Trigger Future.__del__ or futures._TracebackLogger
|
|
support.gc_collect()
|
|
if PY34:
|
|
# Future.__del__ in Python 3.4 logs error with
|
|
# an actual exception context
|
|
log.error.assert_called_with(
|
|
test_utils.MockPattern('.*exception was never retrieved'),
|
|
exc_info=(ZeroDivisionError, MOCK_ANY, MOCK_ANY))
|
|
else:
|
|
# futures._TracebackLogger logs only textual traceback
|
|
log.error.assert_called_with(
|
|
test_utils.MockPattern(
|
|
'.*exception was never retrieved.*ZeroDiv'),
|
|
exc_info=False)
|
|
|
|
def test_set_exc_handler_invalid(self):
|
|
with self.assertRaisesRegex(TypeError, 'A callable object or None'):
|
|
self.loop.set_exception_handler('spam')
|
|
|
|
def test_set_exc_handler_custom(self):
|
|
def zero_error():
|
|
1/0
|
|
|
|
def run_loop():
|
|
handle = self.loop.call_soon(zero_error)
|
|
self.loop._run_once()
|
|
return handle
|
|
|
|
self.loop.set_debug(True)
|
|
self.loop._process_events = mock.Mock()
|
|
|
|
self.assertIsNone(self.loop.get_exception_handler())
|
|
mock_handler = mock.Mock()
|
|
self.loop.set_exception_handler(mock_handler)
|
|
self.assertIs(self.loop.get_exception_handler(), mock_handler)
|
|
handle = run_loop()
|
|
mock_handler.assert_called_with(self.loop, {
|
|
'exception': MOCK_ANY,
|
|
'message': test_utils.MockPattern(
|
|
'Exception in callback.*zero_error'),
|
|
'handle': handle,
|
|
'source_traceback': handle._source_traceback,
|
|
})
|
|
mock_handler.reset_mock()
|
|
|
|
self.loop.set_exception_handler(None)
|
|
with mock.patch('asyncio.base_events.logger') as log:
|
|
run_loop()
|
|
log.error.assert_called_with(
|
|
test_utils.MockPattern(
|
|
'Exception in callback.*zero'),
|
|
exc_info=(ZeroDivisionError, MOCK_ANY, MOCK_ANY))
|
|
|
|
assert not mock_handler.called
|
|
|
|
def test_set_exc_handler_broken(self):
|
|
def run_loop():
|
|
def zero_error():
|
|
1/0
|
|
self.loop.call_soon(zero_error)
|
|
self.loop._run_once()
|
|
|
|
def handler(loop, context):
|
|
raise AttributeError('spam')
|
|
|
|
self.loop._process_events = mock.Mock()
|
|
|
|
self.loop.set_exception_handler(handler)
|
|
|
|
with mock.patch('asyncio.base_events.logger') as log:
|
|
run_loop()
|
|
log.error.assert_called_with(
|
|
test_utils.MockPattern(
|
|
'Unhandled error in exception handler'),
|
|
exc_info=(AttributeError, MOCK_ANY, MOCK_ANY))
|
|
|
|
def test_default_exc_handler_broken(self):
|
|
_context = None
|
|
|
|
class Loop(base_events.BaseEventLoop):
|
|
|
|
_selector = mock.Mock()
|
|
_process_events = mock.Mock()
|
|
|
|
def default_exception_handler(self, context):
|
|
nonlocal _context
|
|
_context = context
|
|
# Simulates custom buggy "default_exception_handler"
|
|
raise ValueError('spam')
|
|
|
|
loop = Loop()
|
|
self.addCleanup(loop.close)
|
|
asyncio.set_event_loop(loop)
|
|
|
|
def run_loop():
|
|
def zero_error():
|
|
1/0
|
|
loop.call_soon(zero_error)
|
|
loop._run_once()
|
|
|
|
with mock.patch('asyncio.base_events.logger') as log:
|
|
run_loop()
|
|
log.error.assert_called_with(
|
|
'Exception in default exception handler',
|
|
exc_info=True)
|
|
|
|
def custom_handler(loop, context):
|
|
raise ValueError('ham')
|
|
|
|
_context = None
|
|
loop.set_exception_handler(custom_handler)
|
|
with mock.patch('asyncio.base_events.logger') as log:
|
|
run_loop()
|
|
log.error.assert_called_with(
|
|
test_utils.MockPattern('Exception in default exception.*'
|
|
'while handling.*in custom'),
|
|
exc_info=True)
|
|
|
|
# Check that original context was passed to default
|
|
# exception handler.
|
|
self.assertIn('context', _context)
|
|
self.assertIs(type(_context['context']['exception']),
|
|
ZeroDivisionError)
|
|
|
|
def test_set_task_factory_invalid(self):
|
|
with self.assertRaisesRegex(
|
|
TypeError, 'task factory must be a callable or None'):
|
|
|
|
self.loop.set_task_factory(1)
|
|
|
|
self.assertIsNone(self.loop.get_task_factory())
|
|
|
|
def test_set_task_factory(self):
|
|
self.loop._process_events = mock.Mock()
|
|
|
|
class MyTask(asyncio.Task):
|
|
pass
|
|
|
|
@asyncio.coroutine
|
|
def coro():
|
|
pass
|
|
|
|
factory = lambda loop, coro: MyTask(coro, loop=loop)
|
|
|
|
self.assertIsNone(self.loop.get_task_factory())
|
|
self.loop.set_task_factory(factory)
|
|
self.assertIs(self.loop.get_task_factory(), factory)
|
|
|
|
task = self.loop.create_task(coro())
|
|
self.assertTrue(isinstance(task, MyTask))
|
|
self.loop.run_until_complete(task)
|
|
|
|
self.loop.set_task_factory(None)
|
|
self.assertIsNone(self.loop.get_task_factory())
|
|
|
|
task = self.loop.create_task(coro())
|
|
self.assertTrue(isinstance(task, asyncio.Task))
|
|
self.assertFalse(isinstance(task, MyTask))
|
|
self.loop.run_until_complete(task)
|
|
|
|
def test_env_var_debug(self):
|
|
code = '\n'.join((
|
|
'import asyncio',
|
|
'loop = asyncio.get_event_loop()',
|
|
'print(loop.get_debug())'))
|
|
|
|
# Test with -E to not fail if the unit test was run with
|
|
# PYTHONASYNCIODEBUG set to a non-empty string
|
|
sts, stdout, stderr = assert_python_ok('-E', '-c', code)
|
|
self.assertEqual(stdout.rstrip(), b'False')
|
|
|
|
sts, stdout, stderr = assert_python_ok('-c', code,
|
|
PYTHONASYNCIODEBUG='',
|
|
PYTHONDEVMODE='')
|
|
self.assertEqual(stdout.rstrip(), b'False')
|
|
|
|
sts, stdout, stderr = assert_python_ok('-c', code,
|
|
PYTHONASYNCIODEBUG='1',
|
|
PYTHONDEVMODE='')
|
|
self.assertEqual(stdout.rstrip(), b'True')
|
|
|
|
sts, stdout, stderr = assert_python_ok('-E', '-c', code,
|
|
PYTHONASYNCIODEBUG='1')
|
|
self.assertEqual(stdout.rstrip(), b'False')
|
|
|
|
# -X dev
|
|
sts, stdout, stderr = assert_python_ok('-E', '-X', 'dev',
|
|
'-c', code)
|
|
self.assertEqual(stdout.rstrip(), b'True')
|
|
|
|
def test_create_task(self):
|
|
class MyTask(asyncio.Task):
|
|
pass
|
|
|
|
@asyncio.coroutine
|
|
def test():
|
|
pass
|
|
|
|
class EventLoop(base_events.BaseEventLoop):
|
|
def create_task(self, coro):
|
|
return MyTask(coro, loop=loop)
|
|
|
|
loop = EventLoop()
|
|
self.set_event_loop(loop)
|
|
|
|
coro = test()
|
|
task = asyncio.ensure_future(coro, loop=loop)
|
|
self.assertIsInstance(task, MyTask)
|
|
|
|
# make warnings quiet
|
|
task._log_destroy_pending = False
|
|
coro.close()
|
|
|
|
def test_run_forever_keyboard_interrupt(self):
|
|
# Python issue #22601: ensure that the temporary task created by
|
|
# run_forever() consumes the KeyboardInterrupt and so don't log
|
|
# a warning
|
|
@asyncio.coroutine
|
|
def raise_keyboard_interrupt():
|
|
raise KeyboardInterrupt
|
|
|
|
self.loop._process_events = mock.Mock()
|
|
self.loop.call_exception_handler = mock.Mock()
|
|
|
|
try:
|
|
self.loop.run_until_complete(raise_keyboard_interrupt())
|
|
except KeyboardInterrupt:
|
|
pass
|
|
self.loop.close()
|
|
support.gc_collect()
|
|
|
|
self.assertFalse(self.loop.call_exception_handler.called)
|
|
|
|
def test_run_until_complete_baseexception(self):
|
|
# Python issue #22429: run_until_complete() must not schedule a pending
|
|
# call to stop() if the future raised a BaseException
|
|
@asyncio.coroutine
|
|
def raise_keyboard_interrupt():
|
|
raise KeyboardInterrupt
|
|
|
|
self.loop._process_events = mock.Mock()
|
|
|
|
try:
|
|
self.loop.run_until_complete(raise_keyboard_interrupt())
|
|
except KeyboardInterrupt:
|
|
pass
|
|
|
|
def func():
|
|
self.loop.stop()
|
|
func.called = True
|
|
func.called = False
|
|
try:
|
|
self.loop.call_soon(func)
|
|
self.loop.run_forever()
|
|
except KeyboardInterrupt:
|
|
pass
|
|
self.assertTrue(func.called)
|
|
|
|
def test_single_selecter_event_callback_after_stopping(self):
|
|
# Python issue #25593: A stopped event loop may cause event callbacks
|
|
# to run more than once.
|
|
event_sentinel = object()
|
|
callcount = 0
|
|
doer = None
|
|
|
|
def proc_events(event_list):
|
|
nonlocal doer
|
|
if event_sentinel in event_list:
|
|
doer = self.loop.call_soon(do_event)
|
|
|
|
def do_event():
|
|
nonlocal callcount
|
|
callcount += 1
|
|
self.loop.call_soon(clear_selector)
|
|
|
|
def clear_selector():
|
|
doer.cancel()
|
|
self.loop._selector.select.return_value = ()
|
|
|
|
self.loop._process_events = proc_events
|
|
self.loop._selector.select.return_value = (event_sentinel,)
|
|
|
|
for i in range(1, 3):
|
|
with self.subTest('Loop %d/2' % i):
|
|
self.loop.call_soon(self.loop.stop)
|
|
self.loop.run_forever()
|
|
self.assertEqual(callcount, 1)
|
|
|
|
def test_run_once(self):
|
|
# Simple test for test_utils.run_once(). It may seem strange
|
|
# to have a test for this (the function isn't even used!) but
|
|
# it's a de-factor standard API for library tests. This tests
|
|
# the idiom: loop.call_soon(loop.stop); loop.run_forever().
|
|
count = 0
|
|
|
|
def callback():
|
|
nonlocal count
|
|
count += 1
|
|
|
|
self.loop._process_events = mock.Mock()
|
|
self.loop.call_soon(callback)
|
|
test_utils.run_once(self.loop)
|
|
self.assertEqual(count, 1)
|
|
|
|
def test_run_forever_pre_stopped(self):
|
|
# Test that the old idiom for pre-stopping the loop works.
|
|
self.loop._process_events = mock.Mock()
|
|
self.loop.stop()
|
|
self.loop.run_forever()
|
|
self.loop._selector.select.assert_called_once_with(0)
|
|
|
|
|
|
class MyProto(asyncio.Protocol):
|
|
done = None
|
|
|
|
def __init__(self, create_future=False):
|
|
self.state = 'INITIAL'
|
|
self.nbytes = 0
|
|
if create_future:
|
|
self.done = asyncio.Future()
|
|
|
|
def connection_made(self, transport):
|
|
self.transport = transport
|
|
assert self.state == 'INITIAL', self.state
|
|
self.state = 'CONNECTED'
|
|
transport.write(b'GET / HTTP/1.0\r\nHost: example.com\r\n\r\n')
|
|
|
|
def data_received(self, data):
|
|
assert self.state == 'CONNECTED', self.state
|
|
self.nbytes += len(data)
|
|
|
|
def eof_received(self):
|
|
assert self.state == 'CONNECTED', self.state
|
|
self.state = 'EOF'
|
|
|
|
def connection_lost(self, exc):
|
|
assert self.state in ('CONNECTED', 'EOF'), self.state
|
|
self.state = 'CLOSED'
|
|
if self.done:
|
|
self.done.set_result(None)
|
|
|
|
|
|
class MyDatagramProto(asyncio.DatagramProtocol):
|
|
done = None
|
|
|
|
def __init__(self, create_future=False, loop=None):
|
|
self.state = 'INITIAL'
|
|
self.nbytes = 0
|
|
if create_future:
|
|
self.done = asyncio.Future(loop=loop)
|
|
|
|
def connection_made(self, transport):
|
|
self.transport = transport
|
|
assert self.state == 'INITIAL', self.state
|
|
self.state = 'INITIALIZED'
|
|
|
|
def datagram_received(self, data, addr):
|
|
assert self.state == 'INITIALIZED', self.state
|
|
self.nbytes += len(data)
|
|
|
|
def error_received(self, exc):
|
|
assert self.state == 'INITIALIZED', self.state
|
|
|
|
def connection_lost(self, exc):
|
|
assert self.state == 'INITIALIZED', self.state
|
|
self.state = 'CLOSED'
|
|
if self.done:
|
|
self.done.set_result(None)
|
|
|
|
|
|
class BaseEventLoopWithSelectorTests(test_utils.TestCase):
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
self.loop = asyncio.new_event_loop()
|
|
self.set_event_loop(self.loop)
|
|
|
|
@mock.patch('socket.getnameinfo')
|
|
def test_getnameinfo(self, m_gai):
|
|
m_gai.side_effect = lambda *args: 42
|
|
r = self.loop.run_until_complete(self.loop.getnameinfo(('abc', 123)))
|
|
self.assertEqual(r, 42)
|
|
|
|
@patch_socket
|
|
def test_create_connection_multiple_errors(self, m_socket):
|
|
|
|
class MyProto(asyncio.Protocol):
|
|
pass
|
|
|
|
@asyncio.coroutine
|
|
def getaddrinfo(*args, **kw):
|
|
yield from []
|
|
return [(2, 1, 6, '', ('107.6.106.82', 80)),
|
|
(2, 1, 6, '', ('107.6.106.82', 80))]
|
|
|
|
def getaddrinfo_task(*args, **kwds):
|
|
return asyncio.Task(getaddrinfo(*args, **kwds), loop=self.loop)
|
|
|
|
idx = -1
|
|
errors = ['err1', 'err2']
|
|
|
|
def _socket(*args, **kw):
|
|
nonlocal idx, errors
|
|
idx += 1
|
|
raise OSError(errors[idx])
|
|
|
|
m_socket.socket = _socket
|
|
|
|
self.loop.getaddrinfo = getaddrinfo_task
|
|
|
|
coro = self.loop.create_connection(MyProto, 'example.com', 80)
|
|
with self.assertRaises(OSError) as cm:
|
|
self.loop.run_until_complete(coro)
|
|
|
|
self.assertEqual(str(cm.exception), 'Multiple exceptions: err1, err2')
|
|
|
|
@patch_socket
|
|
def test_create_connection_timeout(self, m_socket):
|
|
# Ensure that the socket is closed on timeout
|
|
sock = mock.Mock()
|
|
m_socket.socket.return_value = sock
|
|
|
|
def getaddrinfo(*args, **kw):
|
|
fut = asyncio.Future(loop=self.loop)
|
|
addr = (socket.AF_INET, socket.SOCK_STREAM, 0, '',
|
|
('127.0.0.1', 80))
|
|
fut.set_result([addr])
|
|
return fut
|
|
self.loop.getaddrinfo = getaddrinfo
|
|
|
|
with mock.patch.object(self.loop, 'sock_connect',
|
|
side_effect=asyncio.TimeoutError):
|
|
coro = self.loop.create_connection(MyProto, '127.0.0.1', 80)
|
|
with self.assertRaises(asyncio.TimeoutError):
|
|
self.loop.run_until_complete(coro)
|
|
self.assertTrue(sock.close.called)
|
|
|
|
def test_create_connection_host_port_sock(self):
|
|
coro = self.loop.create_connection(
|
|
MyProto, 'example.com', 80, sock=object())
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, coro)
|
|
|
|
def test_create_connection_wrong_sock(self):
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
with sock:
|
|
coro = self.loop.create_connection(MyProto, sock=sock)
|
|
with self.assertRaisesRegex(ValueError,
|
|
'A Stream Socket was expected'):
|
|
self.loop.run_until_complete(coro)
|
|
|
|
def test_create_server_wrong_sock(self):
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
with sock:
|
|
coro = self.loop.create_server(MyProto, sock=sock)
|
|
with self.assertRaisesRegex(ValueError,
|
|
'A Stream Socket was expected'):
|
|
self.loop.run_until_complete(coro)
|
|
|
|
def test_create_server_ssl_timeout_for_plain_socket(self):
|
|
coro = self.loop.create_server(
|
|
MyProto, 'example.com', 80, ssl_handshake_timeout=1)
|
|
with self.assertRaisesRegex(
|
|
ValueError,
|
|
'ssl_handshake_timeout is only meaningful with ssl'):
|
|
self.loop.run_until_complete(coro)
|
|
|
|
@unittest.skipUnless(hasattr(socket, 'SOCK_NONBLOCK'),
|
|
'no socket.SOCK_NONBLOCK (linux only)')
|
|
def test_create_server_stream_bittype(self):
|
|
sock = socket.socket(
|
|
socket.AF_INET, socket.SOCK_STREAM | socket.SOCK_NONBLOCK)
|
|
with sock:
|
|
coro = self.loop.create_server(lambda: None, sock=sock)
|
|
srv = self.loop.run_until_complete(coro)
|
|
srv.close()
|
|
self.loop.run_until_complete(srv.wait_closed())
|
|
|
|
def test_create_datagram_endpoint_wrong_sock(self):
|
|
sock = socket.socket(socket.AF_INET)
|
|
with sock:
|
|
coro = self.loop.create_datagram_endpoint(MyProto, sock=sock)
|
|
with self.assertRaisesRegex(ValueError,
|
|
'A UDP Socket was expected'):
|
|
self.loop.run_until_complete(coro)
|
|
|
|
def test_create_connection_no_host_port_sock(self):
|
|
coro = self.loop.create_connection(MyProto)
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, coro)
|
|
|
|
def test_create_connection_no_getaddrinfo(self):
|
|
@asyncio.coroutine
|
|
def getaddrinfo(*args, **kw):
|
|
yield from []
|
|
|
|
def getaddrinfo_task(*args, **kwds):
|
|
return asyncio.Task(getaddrinfo(*args, **kwds), loop=self.loop)
|
|
|
|
self.loop.getaddrinfo = getaddrinfo_task
|
|
coro = self.loop.create_connection(MyProto, 'example.com', 80)
|
|
self.assertRaises(
|
|
OSError, self.loop.run_until_complete, coro)
|
|
|
|
def test_create_connection_connect_err(self):
|
|
async def getaddrinfo(*args, **kw):
|
|
return [(2, 1, 6, '', ('107.6.106.82', 80))]
|
|
|
|
def getaddrinfo_task(*args, **kwds):
|
|
return asyncio.Task(getaddrinfo(*args, **kwds), loop=self.loop)
|
|
|
|
self.loop.getaddrinfo = getaddrinfo_task
|
|
self.loop.sock_connect = mock.Mock()
|
|
self.loop.sock_connect.side_effect = OSError
|
|
|
|
coro = self.loop.create_connection(MyProto, 'example.com', 80)
|
|
self.assertRaises(
|
|
OSError, self.loop.run_until_complete, coro)
|
|
|
|
def test_create_connection_multiple(self):
|
|
@asyncio.coroutine
|
|
def getaddrinfo(*args, **kw):
|
|
return [(2, 1, 6, '', ('0.0.0.1', 80)),
|
|
(2, 1, 6, '', ('0.0.0.2', 80))]
|
|
|
|
def getaddrinfo_task(*args, **kwds):
|
|
return asyncio.Task(getaddrinfo(*args, **kwds), loop=self.loop)
|
|
|
|
self.loop.getaddrinfo = getaddrinfo_task
|
|
self.loop.sock_connect = mock.Mock()
|
|
self.loop.sock_connect.side_effect = OSError
|
|
|
|
coro = self.loop.create_connection(
|
|
MyProto, 'example.com', 80, family=socket.AF_INET)
|
|
with self.assertRaises(OSError):
|
|
self.loop.run_until_complete(coro)
|
|
|
|
@patch_socket
|
|
def test_create_connection_multiple_errors_local_addr(self, m_socket):
|
|
|
|
def bind(addr):
|
|
if addr[0] == '0.0.0.1':
|
|
err = OSError('Err')
|
|
err.strerror = 'Err'
|
|
raise err
|
|
|
|
m_socket.socket.return_value.bind = bind
|
|
|
|
@asyncio.coroutine
|
|
def getaddrinfo(*args, **kw):
|
|
return [(2, 1, 6, '', ('0.0.0.1', 80)),
|
|
(2, 1, 6, '', ('0.0.0.2', 80))]
|
|
|
|
def getaddrinfo_task(*args, **kwds):
|
|
return asyncio.Task(getaddrinfo(*args, **kwds), loop=self.loop)
|
|
|
|
self.loop.getaddrinfo = getaddrinfo_task
|
|
self.loop.sock_connect = mock.Mock()
|
|
self.loop.sock_connect.side_effect = OSError('Err2')
|
|
|
|
coro = self.loop.create_connection(
|
|
MyProto, 'example.com', 80, family=socket.AF_INET,
|
|
local_addr=(None, 8080))
|
|
with self.assertRaises(OSError) as cm:
|
|
self.loop.run_until_complete(coro)
|
|
|
|
self.assertTrue(str(cm.exception).startswith('Multiple exceptions: '))
|
|
self.assertTrue(m_socket.socket.return_value.close.called)
|
|
|
|
def _test_create_connection_ip_addr(self, m_socket, allow_inet_pton):
|
|
# Test the fallback code, even if this system has inet_pton.
|
|
if not allow_inet_pton:
|
|
del m_socket.inet_pton
|
|
|
|
m_socket.getaddrinfo = socket.getaddrinfo
|
|
sock = m_socket.socket.return_value
|
|
|
|
self.loop._add_reader = mock.Mock()
|
|
self.loop._add_reader._is_coroutine = False
|
|
self.loop._add_writer = mock.Mock()
|
|
self.loop._add_writer._is_coroutine = False
|
|
|
|
coro = self.loop.create_connection(asyncio.Protocol, '1.2.3.4', 80)
|
|
t, p = self.loop.run_until_complete(coro)
|
|
try:
|
|
sock.connect.assert_called_with(('1.2.3.4', 80))
|
|
_, kwargs = m_socket.socket.call_args
|
|
self.assertEqual(kwargs['family'], m_socket.AF_INET)
|
|
self.assertEqual(kwargs['type'], m_socket.SOCK_STREAM)
|
|
finally:
|
|
t.close()
|
|
test_utils.run_briefly(self.loop) # allow transport to close
|
|
|
|
sock.family = socket.AF_INET6
|
|
coro = self.loop.create_connection(asyncio.Protocol, '::1', 80)
|
|
t, p = self.loop.run_until_complete(coro)
|
|
try:
|
|
# Without inet_pton we use getaddrinfo, which transforms ('::1', 80)
|
|
# to ('::1', 80, 0, 0). The last 0s are flow info, scope id.
|
|
[address] = sock.connect.call_args[0]
|
|
host, port = address[:2]
|
|
self.assertRegex(host, r'::(0\.)*1')
|
|
self.assertEqual(port, 80)
|
|
_, kwargs = m_socket.socket.call_args
|
|
self.assertEqual(kwargs['family'], m_socket.AF_INET6)
|
|
self.assertEqual(kwargs['type'], m_socket.SOCK_STREAM)
|
|
finally:
|
|
t.close()
|
|
test_utils.run_briefly(self.loop) # allow transport to close
|
|
|
|
@patch_socket
|
|
def test_create_connection_ip_addr(self, m_socket):
|
|
self._test_create_connection_ip_addr(m_socket, True)
|
|
|
|
@patch_socket
|
|
def test_create_connection_no_inet_pton(self, m_socket):
|
|
self._test_create_connection_ip_addr(m_socket, False)
|
|
|
|
@patch_socket
|
|
def test_create_connection_service_name(self, m_socket):
|
|
m_socket.getaddrinfo = socket.getaddrinfo
|
|
sock = m_socket.socket.return_value
|
|
|
|
self.loop._add_reader = mock.Mock()
|
|
self.loop._add_reader._is_coroutine = False
|
|
self.loop._add_writer = mock.Mock()
|
|
self.loop._add_writer._is_coroutine = False
|
|
|
|
for service, port in ('http', 80), (b'http', 80):
|
|
coro = self.loop.create_connection(asyncio.Protocol,
|
|
'127.0.0.1', service)
|
|
|
|
t, p = self.loop.run_until_complete(coro)
|
|
try:
|
|
sock.connect.assert_called_with(('127.0.0.1', port))
|
|
_, kwargs = m_socket.socket.call_args
|
|
self.assertEqual(kwargs['family'], m_socket.AF_INET)
|
|
self.assertEqual(kwargs['type'], m_socket.SOCK_STREAM)
|
|
finally:
|
|
t.close()
|
|
test_utils.run_briefly(self.loop) # allow transport to close
|
|
|
|
for service in 'nonsense', b'nonsense':
|
|
coro = self.loop.create_connection(asyncio.Protocol,
|
|
'127.0.0.1', service)
|
|
|
|
with self.assertRaises(OSError):
|
|
self.loop.run_until_complete(coro)
|
|
|
|
def test_create_connection_no_local_addr(self):
|
|
@asyncio.coroutine
|
|
def getaddrinfo(host, *args, **kw):
|
|
if host == 'example.com':
|
|
return [(2, 1, 6, '', ('107.6.106.82', 80)),
|
|
(2, 1, 6, '', ('107.6.106.82', 80))]
|
|
else:
|
|
return []
|
|
|
|
def getaddrinfo_task(*args, **kwds):
|
|
return asyncio.Task(getaddrinfo(*args, **kwds), loop=self.loop)
|
|
self.loop.getaddrinfo = getaddrinfo_task
|
|
|
|
coro = self.loop.create_connection(
|
|
MyProto, 'example.com', 80, family=socket.AF_INET,
|
|
local_addr=(None, 8080))
|
|
self.assertRaises(
|
|
OSError, self.loop.run_until_complete, coro)
|
|
|
|
@patch_socket
|
|
def test_create_connection_bluetooth(self, m_socket):
|
|
# See http://bugs.python.org/issue27136, fallback to getaddrinfo when
|
|
# we can't recognize an address is resolved, e.g. a Bluetooth address.
|
|
addr = ('00:01:02:03:04:05', 1)
|
|
|
|
def getaddrinfo(host, port, *args, **kw):
|
|
assert (host, port) == addr
|
|
return [(999, 1, 999, '', (addr, 1))]
|
|
|
|
m_socket.getaddrinfo = getaddrinfo
|
|
sock = m_socket.socket()
|
|
coro = self.loop.sock_connect(sock, addr)
|
|
self.loop.run_until_complete(coro)
|
|
|
|
def test_create_connection_ssl_server_hostname_default(self):
|
|
self.loop.getaddrinfo = mock.Mock()
|
|
|
|
def mock_getaddrinfo(*args, **kwds):
|
|
f = asyncio.Future(loop=self.loop)
|
|
f.set_result([(socket.AF_INET, socket.SOCK_STREAM,
|
|
socket.SOL_TCP, '', ('1.2.3.4', 80))])
|
|
return f
|
|
|
|
self.loop.getaddrinfo.side_effect = mock_getaddrinfo
|
|
self.loop.sock_connect = mock.Mock()
|
|
self.loop.sock_connect.return_value = self.loop.create_future()
|
|
self.loop.sock_connect.return_value.set_result(None)
|
|
self.loop._make_ssl_transport = mock.Mock()
|
|
|
|
class _SelectorTransportMock:
|
|
_sock = None
|
|
|
|
def get_extra_info(self, key):
|
|
return mock.Mock()
|
|
|
|
def close(self):
|
|
self._sock.close()
|
|
|
|
def mock_make_ssl_transport(sock, protocol, sslcontext, waiter,
|
|
**kwds):
|
|
waiter.set_result(None)
|
|
transport = _SelectorTransportMock()
|
|
transport._sock = sock
|
|
return transport
|
|
|
|
self.loop._make_ssl_transport.side_effect = mock_make_ssl_transport
|
|
ANY = mock.ANY
|
|
handshake_timeout = object()
|
|
# First try the default server_hostname.
|
|
self.loop._make_ssl_transport.reset_mock()
|
|
coro = self.loop.create_connection(
|
|
MyProto, 'python.org', 80, ssl=True,
|
|
ssl_handshake_timeout=handshake_timeout)
|
|
transport, _ = self.loop.run_until_complete(coro)
|
|
transport.close()
|
|
self.loop._make_ssl_transport.assert_called_with(
|
|
ANY, ANY, ANY, ANY,
|
|
server_side=False,
|
|
server_hostname='python.org',
|
|
ssl_handshake_timeout=handshake_timeout)
|
|
# Next try an explicit server_hostname.
|
|
self.loop._make_ssl_transport.reset_mock()
|
|
coro = self.loop.create_connection(
|
|
MyProto, 'python.org', 80, ssl=True,
|
|
server_hostname='perl.com',
|
|
ssl_handshake_timeout=handshake_timeout)
|
|
transport, _ = self.loop.run_until_complete(coro)
|
|
transport.close()
|
|
self.loop._make_ssl_transport.assert_called_with(
|
|
ANY, ANY, ANY, ANY,
|
|
server_side=False,
|
|
server_hostname='perl.com',
|
|
ssl_handshake_timeout=handshake_timeout)
|
|
# Finally try an explicit empty server_hostname.
|
|
self.loop._make_ssl_transport.reset_mock()
|
|
coro = self.loop.create_connection(
|
|
MyProto, 'python.org', 80, ssl=True,
|
|
server_hostname='',
|
|
ssl_handshake_timeout=handshake_timeout)
|
|
transport, _ = self.loop.run_until_complete(coro)
|
|
transport.close()
|
|
self.loop._make_ssl_transport.assert_called_with(
|
|
ANY, ANY, ANY, ANY,
|
|
server_side=False,
|
|
server_hostname='',
|
|
ssl_handshake_timeout=handshake_timeout)
|
|
|
|
def test_create_connection_no_ssl_server_hostname_errors(self):
|
|
# When not using ssl, server_hostname must be None.
|
|
coro = self.loop.create_connection(MyProto, 'python.org', 80,
|
|
server_hostname='')
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, coro)
|
|
coro = self.loop.create_connection(MyProto, 'python.org', 80,
|
|
server_hostname='python.org')
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, coro)
|
|
|
|
def test_create_connection_ssl_server_hostname_errors(self):
|
|
# When using ssl, server_hostname may be None if host is non-empty.
|
|
coro = self.loop.create_connection(MyProto, '', 80, ssl=True)
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, coro)
|
|
coro = self.loop.create_connection(MyProto, None, 80, ssl=True)
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, coro)
|
|
sock = socket.socket()
|
|
coro = self.loop.create_connection(MyProto, None, None,
|
|
ssl=True, sock=sock)
|
|
self.addCleanup(sock.close)
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, coro)
|
|
|
|
def test_create_connection_ssl_timeout_for_plain_socket(self):
|
|
coro = self.loop.create_connection(
|
|
MyProto, 'example.com', 80, ssl_handshake_timeout=1)
|
|
with self.assertRaisesRegex(
|
|
ValueError,
|
|
'ssl_handshake_timeout is only meaningful with ssl'):
|
|
self.loop.run_until_complete(coro)
|
|
|
|
def test_create_server_empty_host(self):
|
|
# if host is empty string use None instead
|
|
host = object()
|
|
|
|
@asyncio.coroutine
|
|
def getaddrinfo(*args, **kw):
|
|
nonlocal host
|
|
host = args[0]
|
|
yield from []
|
|
|
|
def getaddrinfo_task(*args, **kwds):
|
|
return asyncio.Task(getaddrinfo(*args, **kwds), loop=self.loop)
|
|
|
|
self.loop.getaddrinfo = getaddrinfo_task
|
|
fut = self.loop.create_server(MyProto, '', 0)
|
|
self.assertRaises(OSError, self.loop.run_until_complete, fut)
|
|
self.assertIsNone(host)
|
|
|
|
def test_create_server_host_port_sock(self):
|
|
fut = self.loop.create_server(
|
|
MyProto, '0.0.0.0', 0, sock=object())
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, fut)
|
|
|
|
def test_create_server_no_host_port_sock(self):
|
|
fut = self.loop.create_server(MyProto)
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, fut)
|
|
|
|
def test_create_server_no_getaddrinfo(self):
|
|
getaddrinfo = self.loop.getaddrinfo = mock.Mock()
|
|
getaddrinfo.return_value = self.loop.create_future()
|
|
getaddrinfo.return_value.set_result(None)
|
|
|
|
f = self.loop.create_server(MyProto, 'python.org', 0)
|
|
self.assertRaises(OSError, self.loop.run_until_complete, f)
|
|
|
|
@patch_socket
|
|
def test_create_server_nosoreuseport(self, m_socket):
|
|
m_socket.getaddrinfo = socket.getaddrinfo
|
|
del m_socket.SO_REUSEPORT
|
|
m_socket.socket.return_value = mock.Mock()
|
|
|
|
f = self.loop.create_server(
|
|
MyProto, '0.0.0.0', 0, reuse_port=True)
|
|
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, f)
|
|
|
|
@patch_socket
|
|
def test_create_server_soreuseport_only_defined(self, m_socket):
|
|
m_socket.getaddrinfo = socket.getaddrinfo
|
|
m_socket.socket.return_value = mock.Mock()
|
|
m_socket.SO_REUSEPORT = -1
|
|
|
|
f = self.loop.create_server(
|
|
MyProto, '0.0.0.0', 0, reuse_port=True)
|
|
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, f)
|
|
|
|
@patch_socket
|
|
def test_create_server_cant_bind(self, m_socket):
|
|
|
|
class Err(OSError):
|
|
strerror = 'error'
|
|
|
|
m_socket.getaddrinfo.return_value = [
|
|
(2, 1, 6, '', ('127.0.0.1', 10100))]
|
|
m_socket.getaddrinfo._is_coroutine = False
|
|
m_sock = m_socket.socket.return_value = mock.Mock()
|
|
m_sock.bind.side_effect = Err
|
|
|
|
fut = self.loop.create_server(MyProto, '0.0.0.0', 0)
|
|
self.assertRaises(OSError, self.loop.run_until_complete, fut)
|
|
self.assertTrue(m_sock.close.called)
|
|
|
|
@patch_socket
|
|
def test_create_datagram_endpoint_no_addrinfo(self, m_socket):
|
|
m_socket.getaddrinfo.return_value = []
|
|
m_socket.getaddrinfo._is_coroutine = False
|
|
|
|
coro = self.loop.create_datagram_endpoint(
|
|
MyDatagramProto, local_addr=('localhost', 0))
|
|
self.assertRaises(
|
|
OSError, self.loop.run_until_complete, coro)
|
|
|
|
def test_create_datagram_endpoint_addr_error(self):
|
|
coro = self.loop.create_datagram_endpoint(
|
|
MyDatagramProto, local_addr='localhost')
|
|
self.assertRaises(
|
|
AssertionError, self.loop.run_until_complete, coro)
|
|
coro = self.loop.create_datagram_endpoint(
|
|
MyDatagramProto, local_addr=('localhost', 1, 2, 3))
|
|
self.assertRaises(
|
|
AssertionError, self.loop.run_until_complete, coro)
|
|
|
|
def test_create_datagram_endpoint_connect_err(self):
|
|
self.loop.sock_connect = mock.Mock()
|
|
self.loop.sock_connect.side_effect = OSError
|
|
|
|
coro = self.loop.create_datagram_endpoint(
|
|
asyncio.DatagramProtocol, remote_addr=('127.0.0.1', 0))
|
|
self.assertRaises(
|
|
OSError, self.loop.run_until_complete, coro)
|
|
|
|
@patch_socket
|
|
def test_create_datagram_endpoint_socket_err(self, m_socket):
|
|
m_socket.getaddrinfo = socket.getaddrinfo
|
|
m_socket.socket.side_effect = OSError
|
|
|
|
coro = self.loop.create_datagram_endpoint(
|
|
asyncio.DatagramProtocol, family=socket.AF_INET)
|
|
self.assertRaises(
|
|
OSError, self.loop.run_until_complete, coro)
|
|
|
|
coro = self.loop.create_datagram_endpoint(
|
|
asyncio.DatagramProtocol, local_addr=('127.0.0.1', 0))
|
|
self.assertRaises(
|
|
OSError, self.loop.run_until_complete, coro)
|
|
|
|
@unittest.skipUnless(support.IPV6_ENABLED, 'IPv6 not supported or enabled')
|
|
def test_create_datagram_endpoint_no_matching_family(self):
|
|
coro = self.loop.create_datagram_endpoint(
|
|
asyncio.DatagramProtocol,
|
|
remote_addr=('127.0.0.1', 0), local_addr=('::1', 0))
|
|
self.assertRaises(
|
|
ValueError, self.loop.run_until_complete, coro)
|
|
|
|
@patch_socket
|
|
def test_create_datagram_endpoint_setblk_err(self, m_socket):
|
|
m_socket.socket.return_value.setblocking.side_effect = OSError
|
|
|
|
coro = self.loop.create_datagram_endpoint(
|
|
asyncio.DatagramProtocol, family=socket.AF_INET)
|
|
self.assertRaises(
|
|
OSError, self.loop.run_until_complete, coro)
|
|
self.assertTrue(
|
|
m_socket.socket.return_value.close.called)
|
|
|
|
def test_create_datagram_endpoint_noaddr_nofamily(self):
|
|
coro = self.loop.create_datagram_endpoint(
|
|
asyncio.DatagramProtocol)
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, coro)
|
|
|
|
@patch_socket
|
|
def test_create_datagram_endpoint_cant_bind(self, m_socket):
|
|
class Err(OSError):
|
|
pass
|
|
|
|
m_socket.getaddrinfo = socket.getaddrinfo
|
|
m_sock = m_socket.socket.return_value = mock.Mock()
|
|
m_sock.bind.side_effect = Err
|
|
|
|
fut = self.loop.create_datagram_endpoint(
|
|
MyDatagramProto,
|
|
local_addr=('127.0.0.1', 0), family=socket.AF_INET)
|
|
self.assertRaises(Err, self.loop.run_until_complete, fut)
|
|
self.assertTrue(m_sock.close.called)
|
|
|
|
def test_create_datagram_endpoint_sock(self):
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
sock.bind(('127.0.0.1', 0))
|
|
fut = self.loop.create_datagram_endpoint(
|
|
lambda: MyDatagramProto(create_future=True, loop=self.loop),
|
|
sock=sock)
|
|
transport, protocol = self.loop.run_until_complete(fut)
|
|
transport.close()
|
|
self.loop.run_until_complete(protocol.done)
|
|
self.assertEqual('CLOSED', protocol.state)
|
|
|
|
@unittest.skipUnless(hasattr(socket, 'AF_UNIX'), 'No UNIX Sockets')
|
|
def test_create_datagram_endpoint_sock_unix(self):
|
|
fut = self.loop.create_datagram_endpoint(
|
|
lambda: MyDatagramProto(create_future=True, loop=self.loop),
|
|
family=socket.AF_UNIX)
|
|
transport, protocol = self.loop.run_until_complete(fut)
|
|
assert transport._sock.family == socket.AF_UNIX
|
|
transport.close()
|
|
self.loop.run_until_complete(protocol.done)
|
|
self.assertEqual('CLOSED', protocol.state)
|
|
|
|
def test_create_datagram_endpoint_sock_sockopts(self):
|
|
class FakeSock:
|
|
type = socket.SOCK_DGRAM
|
|
|
|
fut = self.loop.create_datagram_endpoint(
|
|
MyDatagramProto, local_addr=('127.0.0.1', 0), sock=FakeSock())
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, fut)
|
|
|
|
fut = self.loop.create_datagram_endpoint(
|
|
MyDatagramProto, remote_addr=('127.0.0.1', 0), sock=FakeSock())
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, fut)
|
|
|
|
fut = self.loop.create_datagram_endpoint(
|
|
MyDatagramProto, family=1, sock=FakeSock())
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, fut)
|
|
|
|
fut = self.loop.create_datagram_endpoint(
|
|
MyDatagramProto, proto=1, sock=FakeSock())
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, fut)
|
|
|
|
fut = self.loop.create_datagram_endpoint(
|
|
MyDatagramProto, flags=1, sock=FakeSock())
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, fut)
|
|
|
|
fut = self.loop.create_datagram_endpoint(
|
|
MyDatagramProto, reuse_address=True, sock=FakeSock())
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, fut)
|
|
|
|
fut = self.loop.create_datagram_endpoint(
|
|
MyDatagramProto, reuse_port=True, sock=FakeSock())
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, fut)
|
|
|
|
fut = self.loop.create_datagram_endpoint(
|
|
MyDatagramProto, allow_broadcast=True, sock=FakeSock())
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, fut)
|
|
|
|
def test_create_datagram_endpoint_sockopts(self):
|
|
# Socket options should not be applied unless asked for.
|
|
# SO_REUSEADDR defaults to on for UNIX.
|
|
# SO_REUSEPORT is not available on all platforms.
|
|
|
|
coro = self.loop.create_datagram_endpoint(
|
|
lambda: MyDatagramProto(create_future=True, loop=self.loop),
|
|
local_addr=('127.0.0.1', 0))
|
|
transport, protocol = self.loop.run_until_complete(coro)
|
|
sock = transport.get_extra_info('socket')
|
|
|
|
reuse_address_default_on = (
|
|
os.name == 'posix' and sys.platform != 'cygwin')
|
|
reuseport_supported = hasattr(socket, 'SO_REUSEPORT')
|
|
|
|
if reuse_address_default_on:
|
|
self.assertTrue(
|
|
sock.getsockopt(
|
|
socket.SOL_SOCKET, socket.SO_REUSEADDR))
|
|
else:
|
|
self.assertFalse(
|
|
sock.getsockopt(
|
|
socket.SOL_SOCKET, socket.SO_REUSEADDR))
|
|
if reuseport_supported:
|
|
self.assertFalse(
|
|
sock.getsockopt(
|
|
socket.SOL_SOCKET, socket.SO_REUSEPORT))
|
|
self.assertFalse(
|
|
sock.getsockopt(
|
|
socket.SOL_SOCKET, socket.SO_BROADCAST))
|
|
|
|
transport.close()
|
|
self.loop.run_until_complete(protocol.done)
|
|
self.assertEqual('CLOSED', protocol.state)
|
|
|
|
coro = self.loop.create_datagram_endpoint(
|
|
lambda: MyDatagramProto(create_future=True, loop=self.loop),
|
|
local_addr=('127.0.0.1', 0),
|
|
reuse_address=True,
|
|
reuse_port=reuseport_supported,
|
|
allow_broadcast=True)
|
|
transport, protocol = self.loop.run_until_complete(coro)
|
|
sock = transport.get_extra_info('socket')
|
|
|
|
self.assertTrue(
|
|
sock.getsockopt(
|
|
socket.SOL_SOCKET, socket.SO_REUSEADDR))
|
|
if reuseport_supported:
|
|
self.assertTrue(
|
|
sock.getsockopt(
|
|
socket.SOL_SOCKET, socket.SO_REUSEPORT))
|
|
self.assertTrue(
|
|
sock.getsockopt(
|
|
socket.SOL_SOCKET, socket.SO_BROADCAST))
|
|
|
|
transport.close()
|
|
self.loop.run_until_complete(protocol.done)
|
|
self.assertEqual('CLOSED', protocol.state)
|
|
|
|
@patch_socket
|
|
def test_create_datagram_endpoint_nosoreuseport(self, m_socket):
|
|
del m_socket.SO_REUSEPORT
|
|
m_socket.socket.return_value = mock.Mock()
|
|
|
|
coro = self.loop.create_datagram_endpoint(
|
|
lambda: MyDatagramProto(loop=self.loop),
|
|
local_addr=('127.0.0.1', 0),
|
|
reuse_address=False,
|
|
reuse_port=True)
|
|
|
|
self.assertRaises(ValueError, self.loop.run_until_complete, coro)
|
|
|
|
@patch_socket
|
|
def test_create_datagram_endpoint_ip_addr(self, m_socket):
|
|
def getaddrinfo(*args, **kw):
|
|
self.fail('should not have called getaddrinfo')
|
|
|
|
m_socket.getaddrinfo = getaddrinfo
|
|
m_socket.socket.return_value.bind = bind = mock.Mock()
|
|
self.loop._add_reader = mock.Mock()
|
|
self.loop._add_reader._is_coroutine = False
|
|
|
|
reuseport_supported = hasattr(socket, 'SO_REUSEPORT')
|
|
coro = self.loop.create_datagram_endpoint(
|
|
lambda: MyDatagramProto(loop=self.loop),
|
|
local_addr=('1.2.3.4', 0),
|
|
reuse_address=False,
|
|
reuse_port=reuseport_supported)
|
|
|
|
t, p = self.loop.run_until_complete(coro)
|
|
try:
|
|
bind.assert_called_with(('1.2.3.4', 0))
|
|
m_socket.socket.assert_called_with(family=m_socket.AF_INET,
|
|
proto=m_socket.IPPROTO_UDP,
|
|
type=m_socket.SOCK_DGRAM)
|
|
finally:
|
|
t.close()
|
|
test_utils.run_briefly(self.loop) # allow transport to close
|
|
|
|
def test_accept_connection_retry(self):
|
|
sock = mock.Mock()
|
|
sock.accept.side_effect = BlockingIOError()
|
|
|
|
self.loop._accept_connection(MyProto, sock)
|
|
self.assertFalse(sock.close.called)
|
|
|
|
@mock.patch('asyncio.base_events.logger')
|
|
def test_accept_connection_exception(self, m_log):
|
|
sock = mock.Mock()
|
|
sock.fileno.return_value = 10
|
|
sock.accept.side_effect = OSError(errno.EMFILE, 'Too many open files')
|
|
self.loop._remove_reader = mock.Mock()
|
|
self.loop.call_later = mock.Mock()
|
|
|
|
self.loop._accept_connection(MyProto, sock)
|
|
self.assertTrue(m_log.error.called)
|
|
self.assertFalse(sock.close.called)
|
|
self.loop._remove_reader.assert_called_with(10)
|
|
self.loop.call_later.assert_called_with(
|
|
constants.ACCEPT_RETRY_DELAY,
|
|
# self.loop._start_serving
|
|
mock.ANY,
|
|
MyProto, sock, None, None, mock.ANY, mock.ANY)
|
|
|
|
def test_call_coroutine(self):
|
|
@asyncio.coroutine
|
|
def simple_coroutine():
|
|
pass
|
|
|
|
self.loop.set_debug(True)
|
|
coro_func = simple_coroutine
|
|
coro_obj = coro_func()
|
|
self.addCleanup(coro_obj.close)
|
|
for func in (coro_func, coro_obj):
|
|
with self.assertRaises(TypeError):
|
|
self.loop.call_soon(func)
|
|
with self.assertRaises(TypeError):
|
|
self.loop.call_soon_threadsafe(func)
|
|
with self.assertRaises(TypeError):
|
|
self.loop.call_later(60, func)
|
|
with self.assertRaises(TypeError):
|
|
self.loop.call_at(self.loop.time() + 60, func)
|
|
with self.assertRaises(TypeError):
|
|
self.loop.run_until_complete(
|
|
self.loop.run_in_executor(None, func))
|
|
|
|
@mock.patch('asyncio.base_events.logger')
|
|
def test_log_slow_callbacks(self, m_logger):
|
|
def stop_loop_cb(loop):
|
|
loop.stop()
|
|
|
|
@asyncio.coroutine
|
|
def stop_loop_coro(loop):
|
|
yield from ()
|
|
loop.stop()
|
|
|
|
asyncio.set_event_loop(self.loop)
|
|
self.loop.set_debug(True)
|
|
self.loop.slow_callback_duration = 0.0
|
|
|
|
# slow callback
|
|
self.loop.call_soon(stop_loop_cb, self.loop)
|
|
self.loop.run_forever()
|
|
fmt, *args = m_logger.warning.call_args[0]
|
|
self.assertRegex(fmt % tuple(args),
|
|
"^Executing <Handle.*stop_loop_cb.*> "
|
|
"took .* seconds$")
|
|
|
|
# slow task
|
|
asyncio.ensure_future(stop_loop_coro(self.loop), loop=self.loop)
|
|
self.loop.run_forever()
|
|
fmt, *args = m_logger.warning.call_args[0]
|
|
self.assertRegex(fmt % tuple(args),
|
|
"^Executing <Task.*stop_loop_coro.*> "
|
|
"took .* seconds$")
|
|
|
|
|
|
class RunningLoopTests(unittest.TestCase):
|
|
|
|
def test_running_loop_within_a_loop(self):
|
|
@asyncio.coroutine
|
|
def runner(loop):
|
|
loop.run_forever()
|
|
|
|
loop = asyncio.new_event_loop()
|
|
outer_loop = asyncio.new_event_loop()
|
|
try:
|
|
with self.assertRaisesRegex(RuntimeError,
|
|
'while another loop is running'):
|
|
outer_loop.run_until_complete(runner(loop))
|
|
finally:
|
|
loop.close()
|
|
outer_loop.close()
|
|
|
|
|
|
class BaseLoopSockSendfileTests(test_utils.TestCase):
|
|
|
|
DATA = b"12345abcde" * 16 * 1024 # 160 KiB
|
|
|
|
class MyProto(asyncio.Protocol):
|
|
|
|
def __init__(self, loop):
|
|
self.started = False
|
|
self.closed = False
|
|
self.data = bytearray()
|
|
self.fut = loop.create_future()
|
|
self.transport = None
|
|
|
|
def connection_made(self, transport):
|
|
self.started = True
|
|
self.transport = transport
|
|
|
|
def data_received(self, data):
|
|
self.data.extend(data)
|
|
|
|
def connection_lost(self, exc):
|
|
self.closed = True
|
|
self.fut.set_result(None)
|
|
self.transport = None
|
|
|
|
async def wait_closed(self):
|
|
await self.fut
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
cls.__old_bufsize = constants.SENDFILE_FALLBACK_READBUFFER_SIZE
|
|
constants.SENDFILE_FALLBACK_READBUFFER_SIZE = 1024 * 16
|
|
with open(support.TESTFN, 'wb') as fp:
|
|
fp.write(cls.DATA)
|
|
super().setUpClass()
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
constants.SENDFILE_FALLBACK_READBUFFER_SIZE = cls.__old_bufsize
|
|
support.unlink(support.TESTFN)
|
|
super().tearDownClass()
|
|
|
|
def setUp(self):
|
|
from asyncio.selector_events import BaseSelectorEventLoop
|
|
# BaseSelectorEventLoop() has no native implementation
|
|
self.loop = BaseSelectorEventLoop()
|
|
self.set_event_loop(self.loop)
|
|
self.file = open(support.TESTFN, 'rb')
|
|
self.addCleanup(self.file.close)
|
|
super().setUp()
|
|
|
|
def make_socket(self, blocking=False):
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
sock.setblocking(blocking)
|
|
self.addCleanup(sock.close)
|
|
return sock
|
|
|
|
def run_loop(self, coro):
|
|
return self.loop.run_until_complete(coro)
|
|
|
|
def prepare(self):
|
|
sock = self.make_socket()
|
|
proto = self.MyProto(self.loop)
|
|
server = self.run_loop(self.loop.create_server(
|
|
lambda: proto, support.HOST, 0, family=socket.AF_INET))
|
|
addr = server.sockets[0].getsockname()
|
|
|
|
for _ in range(10):
|
|
try:
|
|
self.run_loop(self.loop.sock_connect(sock, addr))
|
|
except OSError:
|
|
self.run_loop(asyncio.sleep(0.5))
|
|
continue
|
|
else:
|
|
break
|
|
else:
|
|
# One last try, so we get the exception
|
|
self.run_loop(self.loop.sock_connect(sock, addr))
|
|
|
|
def cleanup():
|
|
server.close()
|
|
self.run_loop(server.wait_closed())
|
|
sock.close()
|
|
if proto.transport is not None:
|
|
proto.transport.close()
|
|
self.run_loop(proto.wait_closed())
|
|
|
|
self.addCleanup(cleanup)
|
|
|
|
return sock, proto
|
|
|
|
def test__sock_sendfile_native_failure(self):
|
|
sock, proto = self.prepare()
|
|
|
|
with self.assertRaisesRegex(events.SendfileNotAvailableError,
|
|
"sendfile is not available"):
|
|
self.run_loop(self.loop._sock_sendfile_native(sock, self.file,
|
|
0, None))
|
|
|
|
self.assertEqual(proto.data, b'')
|
|
self.assertEqual(self.file.tell(), 0)
|
|
|
|
def test_sock_sendfile_no_fallback(self):
|
|
sock, proto = self.prepare()
|
|
|
|
with self.assertRaisesRegex(events.SendfileNotAvailableError,
|
|
"sendfile is not available"):
|
|
self.run_loop(self.loop.sock_sendfile(sock, self.file,
|
|
fallback=False))
|
|
|
|
self.assertEqual(self.file.tell(), 0)
|
|
self.assertEqual(proto.data, b'')
|
|
|
|
def test_sock_sendfile_fallback(self):
|
|
sock, proto = self.prepare()
|
|
|
|
ret = self.run_loop(self.loop.sock_sendfile(sock, self.file))
|
|
sock.close()
|
|
self.run_loop(proto.wait_closed())
|
|
|
|
self.assertEqual(ret, len(self.DATA))
|
|
self.assertEqual(self.file.tell(), len(self.DATA))
|
|
self.assertEqual(proto.data, self.DATA)
|
|
|
|
def test_sock_sendfile_fallback_offset_and_count(self):
|
|
sock, proto = self.prepare()
|
|
|
|
ret = self.run_loop(self.loop.sock_sendfile(sock, self.file,
|
|
1000, 2000))
|
|
sock.close()
|
|
self.run_loop(proto.wait_closed())
|
|
|
|
self.assertEqual(ret, 2000)
|
|
self.assertEqual(self.file.tell(), 3000)
|
|
self.assertEqual(proto.data, self.DATA[1000:3000])
|
|
|
|
def test_blocking_socket(self):
|
|
self.loop.set_debug(True)
|
|
sock = self.make_socket(blocking=True)
|
|
with self.assertRaisesRegex(ValueError, "must be non-blocking"):
|
|
self.run_loop(self.loop.sock_sendfile(sock, self.file))
|
|
|
|
def test_nonbinary_file(self):
|
|
sock = self.make_socket()
|
|
with open(support.TESTFN, 'r') as f:
|
|
with self.assertRaisesRegex(ValueError, "binary mode"):
|
|
self.run_loop(self.loop.sock_sendfile(sock, f))
|
|
|
|
def test_nonstream_socket(self):
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
sock.setblocking(False)
|
|
self.addCleanup(sock.close)
|
|
with self.assertRaisesRegex(ValueError, "only SOCK_STREAM type"):
|
|
self.run_loop(self.loop.sock_sendfile(sock, self.file))
|
|
|
|
def test_notint_count(self):
|
|
sock = self.make_socket()
|
|
with self.assertRaisesRegex(TypeError,
|
|
"count must be a positive integer"):
|
|
self.run_loop(self.loop.sock_sendfile(sock, self.file, 0, 'count'))
|
|
|
|
def test_negative_count(self):
|
|
sock = self.make_socket()
|
|
with self.assertRaisesRegex(ValueError,
|
|
"count must be a positive integer"):
|
|
self.run_loop(self.loop.sock_sendfile(sock, self.file, 0, -1))
|
|
|
|
def test_notint_offset(self):
|
|
sock = self.make_socket()
|
|
with self.assertRaisesRegex(TypeError,
|
|
"offset must be a non-negative integer"):
|
|
self.run_loop(self.loop.sock_sendfile(sock, self.file, 'offset'))
|
|
|
|
def test_negative_offset(self):
|
|
sock = self.make_socket()
|
|
with self.assertRaisesRegex(ValueError,
|
|
"offset must be a non-negative integer"):
|
|
self.run_loop(self.loop.sock_sendfile(sock, self.file, -1))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|