mirror of
https://github.com/craigerl/aprsd.git
synced 2025-05-29 04:32:29 -04:00
Hemna
1c39546bb9
This patch includes a completely reworked client structure. There is now only 1 client object, that loads the appropriate drivers. The drivers are fake, aprsis and tcpkiss. The TCPKISS client was written from scratch to avoid using asyncio. Asyncion is nothing but a pain in the ass.
65 lines
1.8 KiB
Python
65 lines
1.8 KiB
Python
from unittest import mock
|
|
|
|
from oslo_config import cfg
|
|
|
|
import aprsd
|
|
from aprsd.client.drivers.fake import APRSDFakeDriver
|
|
from aprsd.plugins import version as version_plugin
|
|
|
|
from .. import fake, test_plugin
|
|
|
|
CONF = cfg.CONF
|
|
|
|
|
|
class TestVersionPlugin(test_plugin.TestPlugin):
|
|
def setUp(self):
|
|
# make sure the fake client driver is enabled
|
|
# Mock CONF for testing
|
|
super().setUp()
|
|
self.conf_patcher = mock.patch('aprsd.client.drivers.fake.CONF')
|
|
self.mock_conf = self.conf_patcher.start()
|
|
|
|
# Configure fake_client.enabled
|
|
self.mock_conf.fake_client.enabled = True
|
|
|
|
# Create an instance of the driver
|
|
self.driver = APRSDFakeDriver()
|
|
self.fromcall = fake.FAKE_FROM_CALLSIGN
|
|
|
|
def tearDown(self):
|
|
self.conf_patcher.stop()
|
|
super().tearDown()
|
|
|
|
@mock.patch('aprsd.stats.collector.Collector')
|
|
def test_version(self, mock_collector_class):
|
|
# Set up the mock collector instance
|
|
mock_collector_instance = mock_collector_class.return_value
|
|
mock_collector_instance.collect.return_value = {
|
|
'APRSDStats': {
|
|
'uptime': '00:00:00',
|
|
}
|
|
}
|
|
|
|
expected = f'APRSD ver:{aprsd.__version__} uptime:00:00:00'
|
|
CONF.callsign = fake.FAKE_TO_CALLSIGN
|
|
version = version_plugin.VersionPlugin()
|
|
version.enabled = True
|
|
|
|
packet = fake.fake_packet(
|
|
message='No',
|
|
msg_number=1,
|
|
)
|
|
|
|
actual = version.filter(packet)
|
|
self.assertEqual(None, actual)
|
|
|
|
packet = fake.fake_packet(
|
|
message='version',
|
|
msg_number=1,
|
|
)
|
|
actual = version.filter(packet)
|
|
self.assertEqual(expected, actual)
|
|
|
|
# Verify the mock was called exactly once
|
|
mock_collector_instance.collect.assert_called_once()
|