kopia lustrzana https://github.com/glidernet/python-ogn-client
27 wiersze
753 B
Python
27 wiersze
753 B
Python
import unittest
|
|
import unittest.mock as mock
|
|
|
|
from ogn.client.client import TelnetClient
|
|
|
|
|
|
class TelnetClientTest(unittest.TestCase):
|
|
@mock.patch('ogn.client.client.socket')
|
|
def test_connect_disconnect(self, socket_mock):
|
|
client = TelnetClient()
|
|
client.connect()
|
|
client.sock.connect.assert_called_once_with(('localhost', 50001))
|
|
|
|
client.disconnect()
|
|
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):
|
|
def callback(raw_message):
|
|
raise ConnectionRefusedError
|
|
|
|
client = TelnetClient()
|
|
client.connect()
|
|
|
|
client.run(callback=callback)
|