diff --git a/tests/client/test_TelnetClient.py b/tests/client/test_TelnetClient.py
index 5d8eb9c..c26a19f 100644
--- a/tests/client/test_TelnetClient.py
+++ b/tests/client/test_TelnetClient.py
@@ -6,9 +6,21 @@ from ogn.client.client import TelnetClient
 
 class TelnetClientTest(unittest.TestCase):
     @mock.patch('ogn.client.client.socket')
-    def test_connect(self, socket_mock):
+    def test_connect_disconnect(self, socket_mock):
+        client = TelnetClient()
+        client.connect()
+        client.sock.connect.assert_called_once()
+
+        client.disconnect()
+        client.sock.shutdown.assert_called_once()
+        client.sock.close.assert_called_once()
+
+    @mock.patch('ogn.client.client.socket')
+    def test_run(self, socket_mock):
         def callback(raw_message):
-            pass
+            raise ConnectionRefusedError
 
         client = TelnetClient()
+        client.connect()
+
         client.run(callback=callback)