1
1
"""Tests for selector_events.py"""
2
2
3
3
import collections
4
- import sys
5
4
import selectors
6
5
import socket
6
+ import sys
7
7
import unittest
8
+ from asyncio import selector_events
8
9
from unittest import mock
10
+
9
11
try :
10
12
import ssl
11
13
except ImportError :
12
14
ssl = None
13
15
14
16
import asyncio
15
- from asyncio .selector_events import BaseSelectorEventLoop
16
- from asyncio . selector_events import _SelectorTransport
17
- from asyncio . selector_events import _SelectorSocketTransport
18
- from asyncio . selector_events import _SelectorDatagramTransport
17
+ from asyncio .selector_events import ( BaseSelectorEventLoop ,
18
+ _SelectorDatagramTransport ,
19
+ _SelectorSocketTransport ,
20
+ _SelectorTransport )
19
21
from test .test_asyncio import utils as test_utils
20
22
21
-
22
23
MOCK_ANY = mock .ANY
23
24
24
25
@@ -746,7 +747,7 @@ def test_write_sendmsg_no_data(self):
746
747
self .assertFalse (self .sock .sendmsg .called )
747
748
self .assertEqual (list_to_buffer ([b'data' ]), transport ._buffer )
748
749
749
- @unittest .skipUnless (hasattr ( socket . socket , 'sendmsg' ) , 'no sendmsg' )
750
+ @unittest .skipUnless (selector_events . HAVE_SENDMSG , 'no sendmsg' )
750
751
def test_write_sendmsg_full (self ):
751
752
data = memoryview (b'data' )
752
753
self .sock .sendmsg = mock .Mock ()
@@ -759,7 +760,7 @@ def test_write_sendmsg_full(self):
759
760
self .assertTrue (self .sock .sendmsg .called )
760
761
self .assertFalse (self .loop .writers )
761
762
762
- @unittest .skipUnless (hasattr ( socket . socket , 'sendmsg' ) , 'no sendmsg' )
763
+ @unittest .skipUnless (selector_events . HAVE_SENDMSG , 'no sendmsg' )
763
764
def test_write_sendmsg_partial (self ):
764
765
765
766
data = memoryview (b'data' )
0 commit comments