from unittest import mock from oslo_config import cfg from aprsd import conf # noqa: F401 from aprsd.plugins import ping as ping_plugin from .. import fake, test_plugin CONF = cfg.CONF class TestPingPlugin(test_plugin.TestPlugin): @mock.patch('time.localtime') def test_ping(self, mock_time): fake_time = mock.MagicMock() h = fake_time.tm_hour = 16 m = fake_time.tm_min = 12 s = fake_time.tm_sec = 55 mock_time.return_value = fake_time CONF.callsign = fake.FAKE_TO_CALLSIGN ping = ping_plugin.PingPlugin() packet = fake.fake_packet( message='location', msg_number=1, ) result = ping.filter(packet) self.assertEqual(None, result) def ping_str(h, m, s): return ( 'Pong! ' + str(h).zfill(2) + ':' + str(m).zfill(2) + ':' + str(s).zfill(2) ) packet = fake.fake_packet( message='Ping', msg_number=1, ) actual = ping.filter(packet) expected = ping_str(h, m, s) self.assertEqual(expected, actual) packet = fake.fake_packet( message='ping', msg_number=1, ) actual = ping.filter(packet) self.assertEqual(expected, actual)