def setUp(self): super().setUp() self.sock = test_utils.mock_nonblocking_socket() self.proactor = mock.Mock() self.ssock, self.csock = mock.Mock(), mock.Mock() with mock.patch('asyncio.proactor_events.socket.socketpair', return_value=(self.ssock, self.csock)): self.loop = BaseProactorEventLoop(self.proactor) self.set_event_loop(self.loop)
def test_sock_connect_resolve_using_socket_params(self, m_gai): addr = ('need-resolution.com', 8080) for sock_type in [socket.SOCK_STREAM, socket.SOCK_DGRAM]: with self.subTest(sock_type): sock = test_utils.mock_nonblocking_socket(type=sock_type) m_gai.side_effect = \ lambda *args: [(None, None, None, None, ('127.0.0.1', 0))] con = self.loop.create_task(self.loop.sock_connect(sock, addr)) self.loop.run_until_complete(con) m_gai.assert_called_with(addr[0], addr[1], sock.family, sock.type, sock.proto, 0) self.loop.run_until_complete(con) sock.connect.assert_called_with(('127.0.0.1', 0))