()
| 33 | |
| 34 | |
| 35 | def mock_socket_module(): |
| 36 | m_socket = mock.MagicMock(spec=socket) |
| 37 | for name in ( |
| 38 | 'AF_INET', 'AF_INET6', 'AF_UNSPEC', 'IPPROTO_TCP', 'IPPROTO_UDP', |
| 39 | 'SOCK_STREAM', 'SOCK_DGRAM', 'SOL_SOCKET', 'SO_REUSEADDR', 'inet_pton' |
| 40 | ): |
| 41 | if hasattr(socket, name): |
| 42 | setattr(m_socket, name, getattr(socket, name)) |
| 43 | else: |
| 44 | delattr(m_socket, name) |
| 45 | |
| 46 | m_socket.socket = mock.MagicMock() |
| 47 | m_socket.socket.return_value = test_utils.mock_nonblocking_socket() |
| 48 | |
| 49 | return m_socket |
| 50 | |
| 51 | |
| 52 | def patch_socket(f): |