|
1 | 1 | import errno |
2 | | -import socket |
3 | 2 | import sys |
4 | 3 | import unittest |
5 | 4 | from shutil import rmtree |
@@ -121,13 +120,13 @@ def test_no_last_error_on_successful_emit(self): |
121 | 120 |
|
122 | 121 | def test_last_error_property(self): |
123 | 122 | EXCEPTION_MSG = "custom exception for testing last_error property" |
124 | | - self._sender.last_error = socket.error(EXCEPTION_MSG) |
| 123 | + self._sender.last_error = OSError(EXCEPTION_MSG) |
125 | 124 |
|
126 | 125 | self.assertEqual(self._sender.last_error.args[0], EXCEPTION_MSG) |
127 | 126 |
|
128 | 127 | def test_clear_last_error(self): |
129 | 128 | EXCEPTION_MSG = "custom exception for testing clear_last_error" |
130 | | - self._sender.last_error = socket.error(EXCEPTION_MSG) |
| 129 | + self._sender.last_error = OSError(EXCEPTION_MSG) |
131 | 130 | self._sender.clear_last_error() |
132 | 131 |
|
133 | 132 | self.assertEqual(self._sender.last_error, None) |
@@ -225,12 +224,12 @@ def __init__(self): |
225 | 224 | self.send_side_effects = [3, 0, 9] |
226 | 225 | self.send_idx = 0 |
227 | 226 | self.recv_side_effects = [ |
228 | | - socket.error(errno.EWOULDBLOCK, "Blah"), |
| 227 | + OSError(errno.EWOULDBLOCK, "Blah"), |
229 | 228 | b"this data is going to be ignored", |
230 | 229 | b"", |
231 | | - socket.error(errno.EWOULDBLOCK, "Blah"), |
232 | | - socket.error(errno.EWOULDBLOCK, "Blah"), |
233 | | - socket.error(errno.EACCES, "This error will never happen"), |
| 230 | + OSError(errno.EWOULDBLOCK, "Blah"), |
| 231 | + OSError(errno.EWOULDBLOCK, "Blah"), |
| 232 | + OSError(errno.EACCES, "This error will never happen"), |
234 | 233 | ] |
235 | 234 | self.recv_idx = 0 |
236 | 235 |
|
|
0 commit comments