mirror of
https://github.com/craigerl/aprsd.git
synced 2024-11-18 06:11:49 -05:00
51 lines
1.2 KiB
Python
51 lines
1.2 KiB
Python
|
from unittest import mock
|
||
|
|
|
||
|
|
from aprsd.plugins import ping as ping_plugin
|
||
|
|
|
||
|
|
from .. import fake, test_plugin
|
||
|
|
|
||
|
|
|
||
|
|
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
|
||
|
|
|
||
|
|
ping = ping_plugin.PingPlugin(self.config)
|
||
|
|
|
||
|
|
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)
|