mirror of
https://github.com/craigerl/aprsd.git
synced 2024-11-22 08:04:53 -05:00
82 lines
3.1 KiB
Python
82 lines
3.1 KiB
Python
|
import datetime
|
||
|
|
import unittest
|
||
|
|
from unittest import mock
|
||
|
|
|
||
|
|
from aprsd import exception
|
||
|
|
from aprsd.client.aprsis import APRSISClient
|
||
|
|
|
||
|
|
|
||
|
|
class TestAPRSISClient(unittest.TestCase):
|
||
|
|
"""Test cases for APRSISClient."""
|
||
|
|
|
||
|
|
def setUp(self):
|
||
|
|
"""Set up test fixtures."""
|
||
|
|
super().setUp()
|
||
|
|
|
||
|
|
# Mock the config
|
||
|
|
self.mock_conf = mock.MagicMock()
|
||
|
|
self.mock_conf.aprs_network.enabled = True
|
||
|
|
self.mock_conf.aprs_network.login = "TEST"
|
||
|
|
self.mock_conf.aprs_network.password = "12345"
|
||
|
|
self.mock_conf.aprs_network.host = "localhost"
|
||
|
|
self.mock_conf.aprs_network.port = 14580
|
||
|
|
|
||
|
|
@mock.patch("aprsd.client.base.APRSClient")
|
||
|
|
@mock.patch("aprsd.client.drivers.aprsis.Aprsdis")
|
||
|
|
def test_stats_not_configured(self, mock_aprsdis, mock_base):
|
||
|
|
"""Test stats when client is not configured."""
|
||
|
|
mock_client = mock.MagicMock()
|
||
|
|
mock_aprsdis.return_value = mock_client
|
||
|
|
|
||
|
|
with mock.patch("aprsd.client.aprsis.cfg.CONF", self.mock_conf):
|
||
|
|
self.client = APRSISClient()
|
||
|
|
|
||
|
|
with mock.patch.object(APRSISClient, "is_configured", return_value=False):
|
||
|
|
stats = self.client.stats()
|
||
|
|
self.assertEqual({}, stats)
|
||
|
|
|
||
|
|
@mock.patch("aprsd.client.base.APRSClient")
|
||
|
|
@mock.patch("aprsd.client.drivers.aprsis.Aprsdis")
|
||
|
|
def test_stats_configured(self, mock_aprsdis, mock_base):
|
||
|
|
"""Test stats when client is configured."""
|
||
|
|
mock_client = mock.MagicMock()
|
||
|
|
mock_aprsdis.return_value = mock_client
|
||
|
|
|
||
|
|
with mock.patch("aprsd.client.aprsis.cfg.CONF", self.mock_conf):
|
||
|
|
self.client = APRSISClient()
|
||
|
|
|
||
|
|
mock_client = mock.MagicMock()
|
||
|
|
mock_client.server_string = "test.server:14580"
|
||
|
|
mock_client.aprsd_keepalive = datetime.datetime.now()
|
||
|
|
self.client._client = mock_client
|
||
|
|
self.client.filter = "m/50"
|
||
|
|
|
||
|
|
with mock.patch.object(APRSISClient, "is_configured", return_value=True):
|
||
|
|
stats = self.client.stats()
|
||
|
|
self.assertEqual(
|
||
|
|
{
|
||
|
|
"server_string": mock_client.server_string,
|
||
|
|
"sever_keepalive": mock_client.aprsd_keepalive,
|
||
|
|
"filter": "m/50",
|
||
|
|
}, stats,
|
||
|
|
)
|
||
|
|
|
||
|
|
def test_is_configured_missing_login(self):
|
||
|
|
"""Test is_configured with missing login."""
|
||
|
|
self.mock_conf.aprs_network.login = None
|
||
|
|
with self.assertRaises(exception.MissingConfigOptionException):
|
||
|
|
APRSISClient.is_configured()
|
||
|
|
|
||
|
|
def test_is_configured_missing_password(self):
|
||
|
|
"""Test is_configured with missing password."""
|
||
|
|
self.mock_conf.aprs_network.password = None
|
||
|
|
with self.assertRaises(exception.MissingConfigOptionException):
|
||
|
|
APRSISClient.is_configured()
|
||
|
|
|
||
|
|
def test_is_configured_missing_host(self):
|
||
|
|
"""Test is_configured with missing host."""
|
||
|
|
self.mock_conf.aprs_network.host = None
|
||
|
|
with mock.patch("aprsd.client.aprsis.cfg.CONF", self.mock_conf):
|
||
|
|
with self.assertRaises(exception.MissingConfigOptionException):
|
||
|
|
APRSISClient.is_configured()
|