kopia lustrzana https://github.com/glidernet/ogn-python
36 wiersze
1.1 KiB
Python
36 wiersze
1.1 KiB
Python
import unittest
|
|
import unittest.mock as mock
|
|
|
|
from ogn.gateway.manage import run
|
|
# from ogn.gateway.manage import import_logfile
|
|
|
|
|
|
class GatewayManagerTest(unittest.TestCase):
|
|
# try simple user interrupt
|
|
@mock.patch('ogn.gateway.manage.AprsClient')
|
|
def test_run_user_interruption(self, mock_aprs_client):
|
|
instance = mock_aprs_client.return_value
|
|
instance.run.side_effect = KeyboardInterrupt()
|
|
|
|
run(aprs_user="testuser")
|
|
|
|
instance.connect.assert_called_once_with()
|
|
self.assertEqual(instance.run.call_count, 1)
|
|
instance.disconnect.assert_called_once_with()
|
|
|
|
# try to import stored OGN logfile
|
|
@mock.patch('ogn.gateway.manage.import_logfile')
|
|
def test_run_import_logfile(self, mock_import_logfile):
|
|
# instance = mock_import_logfile.return_value
|
|
|
|
# import_logfile(ogn_logfile="tests/OGN_log.txt_2016-09-21", reference_date="2016-09-21")
|
|
|
|
# instance.connect.assert_called_once_with()
|
|
# self.assertEqual(import_logfile.call_count, 1)
|
|
# instance.disconnect.assert_called_once_with()
|
|
pass
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|