Add missing reject_connection_test

This commit is contained in:
Thomas Adamcik 2011-07-12 01:34:02 +02:00
parent 6d7575a2db
commit d9406420e3

View File

@ -192,3 +192,17 @@ class ServerTest(unittest.TestCase):
network.Connection.assert_called_once_with(sentinel.protocol,
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()