def test_close(self, mock_socket): conn = TcpConnection('localhost', 2000) conn.open() conn.close() conn.socket.close.assert_called_once_with() ok_(not conn.is_connected())