Add missing reject_connection_test
This commit is contained in:
parent
6d7575a2db
commit
d9406420e3
@ -192,3 +192,17 @@ class ServerTest(unittest.TestCase):
|
|||||||
network.Connection.assert_called_once_with(sentinel.protocol,
|
network.Connection.assert_called_once_with(sentinel.protocol,
|
||||||
sentinel.sock, sentinel.addr, sentinel.timeout)
|
sentinel.sock, sentinel.addr, sentinel.timeout)
|
||||||
|
|
||||||
|
def test_reject_connection(self):
|
||||||
|
sock = Mock(spec=socket.SocketType)
|
||||||
|
|
||||||
|
network.Server.reject_connection(self.mock, sock,
|
||||||
|
(sentinel.host, sentinel.port))
|
||||||
|
sock.close.assert_called_once_with()
|
||||||
|
|
||||||
|
def test_reject_connection_error(self):
|
||||||
|
sock = Mock(spec=socket.SocketType)
|
||||||
|
sock.close.side_effect = socket.error()
|
||||||
|
|
||||||
|
network.Server.reject_connection(self.mock, sock,
|
||||||
|
(sentinel.host, sentinel.port))
|
||||||
|
sock.close.assert_called_once_with()
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user