From d9406420e3ce6462e2eb0b878b00dc06a135ef9b Mon Sep 17 00:00:00 2001 From: Thomas Adamcik Date: Tue, 12 Jul 2011 01:34:02 +0200 Subject: [PATCH] Add missing reject_connection_test --- tests/utils/network_test.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/tests/utils/network_test.py b/tests/utils/network_test.py index b686e20c..00032884 100644 --- a/tests/utils/network_test.py +++ b/tests/utils/network_test.py @@ -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()