@@ -59,6 +59,7 @@ def setUp(self):
59
59
def test_make_socket_transport (self ):
60
60
m = mock .Mock ()
61
61
self .loop .add_reader = mock .Mock ()
62
+ self .loop .add_reader ._is_coroutine = False
62
63
transport = self .loop ._make_socket_transport (m , asyncio .Protocol ())
63
64
self .assertIsInstance (transport , _SelectorSocketTransport )
64
65
close_transport (transport )
@@ -67,6 +68,7 @@ def test_make_socket_transport(self):
67
68
def test_make_ssl_transport (self ):
68
69
m = mock .Mock ()
69
70
self .loop .add_reader = mock .Mock ()
71
+ self .loop .add_reader ._is_coroutine = False
70
72
self .loop .add_writer = mock .Mock ()
71
73
self .loop .remove_reader = mock .Mock ()
72
74
self .loop .remove_writer = mock .Mock ()
@@ -770,20 +772,24 @@ def socket_transport(self, waiter=None):
770
772
return transport
771
773
772
774
def test_ctor (self ):
773
- tr = self .socket_transport ()
775
+ waiter = asyncio .Future (loop = self .loop )
776
+ tr = self .socket_transport (waiter = waiter )
777
+ self .loop .run_until_complete (waiter )
778
+
774
779
self .loop .assert_reader (7 , tr ._read_ready )
775
780
test_utils .run_briefly (self .loop )
776
781
self .protocol .connection_made .assert_called_with (tr )
777
782
778
783
def test_ctor_with_waiter (self ):
779
- fut = asyncio .Future (loop = self .loop )
784
+ waiter = asyncio .Future (loop = self .loop )
785
+ self .socket_transport (waiter = waiter )
786
+ self .loop .run_until_complete (waiter )
780
787
781
- self .socket_transport (waiter = fut )
782
- test_utils .run_briefly (self .loop )
783
- self .assertIsNone (fut .result ())
788
+ self .assertIsNone (waiter .result ())
784
789
785
790
def test_pause_resume_reading (self ):
786
791
tr = self .socket_transport ()
792
+ test_utils .run_briefly (self .loop )
787
793
self .assertFalse (tr ._paused )
788
794
self .loop .assert_reader (7 , tr ._read_ready )
789
795
tr .pause_reading ()
0 commit comments