diff options
Diffstat (limited to 'Lib/test/test_socket.py')
| -rwxr-xr-x | Lib/test/test_socket.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/Lib/test/test_socket.py b/Lib/test/test_socket.py index 394d294248..56cc23dbbb 100755 --- a/Lib/test/test_socket.py +++ b/Lib/test/test_socket.py @@ -1517,9 +1517,11 @@ class GeneralModuleTests(unittest.TestCase): infos = socket.getaddrinfo(HOST, 80, socket.AF_INET, socket.SOCK_STREAM) for family, type, _, _, _ in infos: self.assertEqual(family, socket.AF_INET) - self.assertEqual(str(family), 'AF_INET') + self.assertEqual(repr(family), '<AddressFamily.AF_INET: 2>') + self.assertEqual(str(family), '2') self.assertEqual(type, socket.SOCK_STREAM) - self.assertEqual(str(type), 'SOCK_STREAM') + self.assertEqual(repr(type), '<SocketKind.SOCK_STREAM: 1>') + self.assertEqual(str(type), '1') infos = socket.getaddrinfo(HOST, None, 0, socket.SOCK_STREAM) for _, socktype, _, _, _ in infos: self.assertEqual(socktype, socket.SOCK_STREAM) @@ -1793,8 +1795,10 @@ class GeneralModuleTests(unittest.TestCase): # Make sure that the AF_* and SOCK_* constants have enum-like string # reprs. with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s: - self.assertEqual(str(s.family), 'AF_INET') - self.assertEqual(str(s.type), 'SOCK_STREAM') + self.assertEqual(repr(s.family), '<AddressFamily.AF_INET: 2>') + self.assertEqual(repr(s.type), '<SocketKind.SOCK_STREAM: 1>') + self.assertEqual(str(s.family), '2') + self.assertEqual(str(s.type), '1') def test_socket_consistent_sock_type(self): SOCK_NONBLOCK = getattr(socket, 'SOCK_NONBLOCK', 0) |