From 74bd6d7b34de7f7f0b89aee428c9e1ce81047bf0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Konstantin=20Gru=CC=88ndger?= Date: Tue, 10 Apr 2018 22:37:29 +0200 Subject: [PATCH] Fixed test --- tests/client/test_TelnetClient.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/client/test_TelnetClient.py b/tests/client/test_TelnetClient.py index c26a19f..3e39d76 100644 --- a/tests/client/test_TelnetClient.py +++ b/tests/client/test_TelnetClient.py @@ -9,11 +9,11 @@ class TelnetClientTest(unittest.TestCase): def test_connect_disconnect(self, socket_mock): client = TelnetClient() client.connect() - client.sock.connect.assert_called_once() + client.sock.connect.assert_called_once_with(('localhost', 50001)) client.disconnect() - client.sock.shutdown.assert_called_once() - client.sock.close.assert_called_once() + client.sock.shutdown.assert_called_once_with(0) + client.sock.close.assert_called_once_with() @mock.patch('ogn.client.client.socket') def test_run(self, socket_mock):