Skip to content
This repository was archived by the owner on Nov 23, 2017. It is now read-only.

Commit b2810ae

Browse files
author
Vincent Michel
committed
Avoid warnings in test_create_datagram_endpoint_allow_broadcast
1 parent d79f52c commit b2810ae

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

tests/test_base_events.py

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1463,18 +1463,23 @@ def test_create_datagram_endpoint_connect_err(self):
14631463
self.assertRaises(
14641464
OSError, self.loop.run_until_complete, coro)
14651465

1466-
def test_create_datagram_endpoint_no_connect_when_broadcast_allowed(self):
1466+
def test_create_datagram_endpoint_allow_broadcast(self):
1467+
protocol = MyDatagramProto(create_future=True, loop=self.loop)
14671468
self.loop.sock_connect = sock_connect = mock.Mock()
14681469
sock_connect.return_value = []
14691470

14701471
coro = self.loop.create_datagram_endpoint(
1471-
asyncio.DatagramProtocol,
1472+
lambda: protocol,
14721473
remote_addr=('127.0.0.1', 0),
14731474
allow_broadcast=True)
14741475

1475-
self.loop.run_until_complete(coro)
1476+
transport, _ = self.loop.run_until_complete(coro)
14761477
self.assertFalse(sock_connect.called)
14771478

1479+
transport.close()
1480+
self.loop.run_until_complete(protocol.done)
1481+
self.assertEqual('CLOSED', protocol.state)
1482+
14781483
@patch_socket
14791484
def test_create_datagram_endpoint_socket_err(self, m_socket):
14801485
m_socket.getaddrinfo = socket.getaddrinfo

0 commit comments

Comments
 (0)