mirror of
https://github.com/craigerl/aprsd.git
synced 2026-01-14 01:25:32 -05:00
54 lines
1.4 KiB
Python
54 lines
1.4 KiB
Python
from unittest import mock
|
|
|
|
import pytz
|
|
from oslo_config import cfg
|
|
|
|
from aprsd.plugins import time as time_plugin
|
|
from aprsd.utils import fuzzy
|
|
|
|
from .. import fake, test_plugin
|
|
|
|
CONF = cfg.CONF
|
|
|
|
|
|
class TestTimePlugins(test_plugin.TestPlugin):
|
|
@mock.patch('aprsd.plugins.time.TimePlugin._get_local_tz')
|
|
@mock.patch('aprsd.plugins.time.TimePlugin._get_utcnow')
|
|
def test_time(self, mock_utcnow, mock_localtz):
|
|
utcnow = pytz.datetime.datetime.utcnow()
|
|
mock_utcnow.return_value = utcnow
|
|
tz = pytz.timezone('US/Pacific')
|
|
mock_localtz.return_value = tz
|
|
|
|
gmt_t = pytz.utc.localize(utcnow)
|
|
local_t = gmt_t.astimezone(tz)
|
|
|
|
fake_time = mock.MagicMock()
|
|
h = int(local_t.strftime('%H'))
|
|
m = int(local_t.strftime('%M'))
|
|
fake_time.tm_sec = 13
|
|
CONF.callsign = fake.FAKE_TO_CALLSIGN
|
|
time = time_plugin.TimePlugin()
|
|
|
|
packet = fake.fake_packet(
|
|
message='location',
|
|
msg_number=1,
|
|
)
|
|
|
|
actual = time.filter(packet)
|
|
self.assertEqual(None, actual)
|
|
|
|
cur_time = fuzzy(h, m, 1)
|
|
|
|
packet = fake.fake_packet(
|
|
message='time',
|
|
msg_number=1,
|
|
)
|
|
local_short_str = local_t.strftime('%H:%M %Z')
|
|
expected = '{} ({})'.format(
|
|
cur_time,
|
|
local_short_str,
|
|
)
|
|
actual = time.filter(packet)
|
|
self.assertEqual(expected, actual)
|