Update to receiver version 0.2.6

Update to receiver version 0.2.6

Better testing
pull/51/head
Konstantin Gründger 2017-04-03 08:08:03 +02:00
rodzic c58739705f
commit bc3e31838f
1 zmienionych plików z 14 dodań i 2 usunięć

Wyświetl plik

@ -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)