mirror of
https://github.com/craigerl/aprsd.git
synced 2026-07-26 03:54:06 -04:00
Added unit tests for packets.
Also did some code cleanup.
This commit is contained in:
@@ -18,89 +18,89 @@ class TestUSWeatherPlugin(test_plugin.TestPlugin):
|
||||
CONF.callsign = fake.FAKE_TO_CALLSIGN
|
||||
wx = weather_plugin.USWeatherPlugin()
|
||||
expected = "USWeatherPlugin isn't enabled"
|
||||
packet = fake.fake_packet(message="weather")
|
||||
packet = fake.fake_packet(message='weather')
|
||||
actual = wx.filter(packet)
|
||||
self.assertEqual(expected, actual)
|
||||
|
||||
@mock.patch("aprsd.plugin_utils.get_aprs_fi")
|
||||
@mock.patch('aprsd.plugin_utils.get_aprs_fi')
|
||||
def test_failed_aprs_fi_location(self, mock_check):
|
||||
# When the aprs.fi api key isn't set, then
|
||||
# the Plugin will be disabled.
|
||||
mock_check.side_effect = Exception
|
||||
CONF.aprs_fi.apiKey = "abc123"
|
||||
CONF.aprs_fi.apiKey = 'abc123'
|
||||
CONF.callsign = fake.FAKE_TO_CALLSIGN
|
||||
wx = weather_plugin.USWeatherPlugin()
|
||||
expected = "Failed to fetch aprs.fi location"
|
||||
packet = fake.fake_packet(message="weather")
|
||||
expected = 'Failed to fetch aprs.fi location'
|
||||
packet = fake.fake_packet(message='weather')
|
||||
actual = wx.filter(packet)
|
||||
self.assertEqual(expected, actual)
|
||||
|
||||
@mock.patch("aprsd.plugin_utils.get_aprs_fi")
|
||||
@mock.patch('aprsd.plugin_utils.get_aprs_fi')
|
||||
def test_failed_aprs_fi_location_no_entries(self, mock_check):
|
||||
# When the aprs.fi api key isn't set, then
|
||||
# the Plugin will be disabled.
|
||||
mock_check.return_value = {"entries": []}
|
||||
CONF.aprs_fi.apiKey = "abc123"
|
||||
mock_check.return_value = {'entries': []}
|
||||
CONF.aprs_fi.apiKey = 'abc123'
|
||||
CONF.callsign = fake.FAKE_TO_CALLSIGN
|
||||
wx = weather_plugin.USWeatherPlugin()
|
||||
wx.enabled = True
|
||||
expected = "Failed to fetch aprs.fi location"
|
||||
packet = fake.fake_packet(message="weather")
|
||||
expected = 'Failed to fetch aprs.fi location'
|
||||
packet = fake.fake_packet(message='weather')
|
||||
actual = wx.filter(packet)
|
||||
self.assertEqual(expected, actual)
|
||||
|
||||
@mock.patch("aprsd.plugin_utils.get_aprs_fi")
|
||||
@mock.patch("aprsd.plugin_utils.get_weather_gov_for_gps")
|
||||
@mock.patch('aprsd.plugin_utils.get_aprs_fi')
|
||||
@mock.patch('aprsd.plugin_utils.get_weather_gov_for_gps')
|
||||
def test_unknown_gps(self, mock_weather, mock_check_aprs):
|
||||
# When the aprs.fi api key isn't set, then
|
||||
# the LocationPlugin will be disabled.
|
||||
mock_check_aprs.return_value = {
|
||||
"entries": [
|
||||
'entries': [
|
||||
{
|
||||
"lat": 10,
|
||||
"lng": 11,
|
||||
"lasttime": 10,
|
||||
'lat': 10,
|
||||
'lng': 11,
|
||||
'lasttime': 10,
|
||||
},
|
||||
],
|
||||
}
|
||||
mock_weather.side_effect = Exception
|
||||
CONF.aprs_fi.apiKey = "abc123"
|
||||
CONF.aprs_fi.apiKey = 'abc123'
|
||||
CONF.callsign = fake.FAKE_TO_CALLSIGN
|
||||
wx = weather_plugin.USWeatherPlugin()
|
||||
wx.enabled = True
|
||||
expected = "Unable to get weather"
|
||||
packet = fake.fake_packet(message="weather")
|
||||
expected = 'Unable to get weather'
|
||||
packet = fake.fake_packet(message='weather')
|
||||
actual = wx.filter(packet)
|
||||
self.assertEqual(expected, actual)
|
||||
|
||||
@mock.patch("aprsd.plugin_utils.get_aprs_fi")
|
||||
@mock.patch("aprsd.plugin_utils.get_weather_gov_for_gps")
|
||||
@mock.patch('aprsd.plugin_utils.get_aprs_fi')
|
||||
@mock.patch('aprsd.plugin_utils.get_weather_gov_for_gps')
|
||||
def test_working(self, mock_weather, mock_check_aprs):
|
||||
# When the aprs.fi api key isn't set, then
|
||||
# the LocationPlugin will be disabled.
|
||||
mock_check_aprs.return_value = {
|
||||
"entries": [
|
||||
'entries': [
|
||||
{
|
||||
"lat": 10,
|
||||
"lng": 11,
|
||||
"lasttime": 10,
|
||||
'lat': 10,
|
||||
'lng': 11,
|
||||
'lasttime': 10,
|
||||
},
|
||||
],
|
||||
}
|
||||
mock_weather.return_value = {
|
||||
"currentobservation": {"Temp": "400"},
|
||||
"data": {
|
||||
"temperature": ["10", "11"],
|
||||
"weather": ["test", "another"],
|
||||
'currentobservation': {'Temp': '400'},
|
||||
'data': {
|
||||
'temperature': ['10', '11'],
|
||||
'weather': ['test', 'another'],
|
||||
},
|
||||
"time": {"startPeriodName": ["ignored", "sometime"]},
|
||||
'time': {'startPeriodName': ['ignored', 'sometime']},
|
||||
}
|
||||
CONF.aprs_fi.apiKey = "abc123"
|
||||
CONF.aprs_fi.apiKey = 'abc123'
|
||||
CONF.callsign = fake.FAKE_TO_CALLSIGN
|
||||
wx = weather_plugin.USWeatherPlugin()
|
||||
wx.enabled = True
|
||||
expected = "400F(10F/11F) test. sometime, another."
|
||||
packet = fake.fake_packet(message="weather")
|
||||
expected = '400F(10F/11F) test. sometime, another.'
|
||||
packet = fake.fake_packet(message='weather')
|
||||
actual = wx.filter(packet)
|
||||
self.assertEqual(expected, actual)
|
||||
|
||||
@@ -112,93 +112,93 @@ class TestUSMetarPlugin(test_plugin.TestPlugin):
|
||||
CONF.aprs_fi.apiKey = None
|
||||
wx = weather_plugin.USMetarPlugin()
|
||||
expected = "USMetarPlugin isn't enabled"
|
||||
packet = fake.fake_packet(message="metar")
|
||||
packet = fake.fake_packet(message='metar')
|
||||
actual = wx.filter(packet)
|
||||
self.assertEqual(expected, actual)
|
||||
|
||||
@mock.patch("aprsd.plugin_utils.get_aprs_fi")
|
||||
@mock.patch('aprsd.plugin_utils.get_aprs_fi')
|
||||
def test_failed_aprs_fi_location(self, mock_check):
|
||||
# When the aprs.fi api key isn't set, then
|
||||
# the Plugin will be disabled.
|
||||
mock_check.side_effect = Exception
|
||||
CONF.aprs_fi.apiKey = "abc123"
|
||||
CONF.aprs_fi.apiKey = 'abc123'
|
||||
CONF.callsign = fake.FAKE_TO_CALLSIGN
|
||||
wx = weather_plugin.USMetarPlugin()
|
||||
wx.enabled = True
|
||||
expected = "Failed to fetch aprs.fi location"
|
||||
packet = fake.fake_packet(message="metar")
|
||||
expected = 'Failed to fetch aprs.fi location'
|
||||
packet = fake.fake_packet(message='metar')
|
||||
actual = wx.filter(packet)
|
||||
self.assertEqual(expected, actual)
|
||||
|
||||
@mock.patch("aprsd.plugin_utils.get_aprs_fi")
|
||||
@mock.patch('aprsd.plugin_utils.get_aprs_fi')
|
||||
def test_failed_aprs_fi_location_no_entries(self, mock_check):
|
||||
# When the aprs.fi api key isn't set, then
|
||||
# the Plugin will be disabled.
|
||||
mock_check.return_value = {"entries": []}
|
||||
CONF.aprs_fi.apiKey = "abc123"
|
||||
mock_check.return_value = {'entries': []}
|
||||
CONF.aprs_fi.apiKey = 'abc123'
|
||||
CONF.callsign = fake.FAKE_TO_CALLSIGN
|
||||
wx = weather_plugin.USMetarPlugin()
|
||||
wx.enabled = True
|
||||
expected = "Failed to fetch aprs.fi location"
|
||||
packet = fake.fake_packet(message="metar")
|
||||
expected = 'Failed to fetch aprs.fi location'
|
||||
packet = fake.fake_packet(message='metar')
|
||||
actual = wx.filter(packet)
|
||||
self.assertEqual(expected, actual)
|
||||
|
||||
@mock.patch("aprsd.plugin_utils.get_weather_gov_metar")
|
||||
@mock.patch('aprsd.plugin_utils.get_weather_gov_metar')
|
||||
def test_gov_metar_fetch_fails(self, mock_metar):
|
||||
mock_metar.side_effect = Exception
|
||||
CONF.aprs_fi.apiKey = "abc123"
|
||||
CONF.aprs_fi.apiKey = 'abc123'
|
||||
CONF.callsign = fake.FAKE_TO_CALLSIGN
|
||||
wx = weather_plugin.USMetarPlugin()
|
||||
wx.enabled = True
|
||||
expected = "Unable to find station METAR"
|
||||
packet = fake.fake_packet(message="metar KPAO")
|
||||
expected = 'Unable to find station METAR'
|
||||
packet = fake.fake_packet(message='metar KPAO')
|
||||
actual = wx.filter(packet)
|
||||
self.assertEqual(expected, actual)
|
||||
|
||||
@mock.patch("aprsd.plugin_utils.get_weather_gov_metar")
|
||||
@mock.patch('aprsd.plugin_utils.get_weather_gov_metar')
|
||||
def test_airport_works(self, mock_metar):
|
||||
class Response:
|
||||
text = '{"properties": {"rawMessage": "BOGUSMETAR"}}'
|
||||
|
||||
mock_metar.return_value = Response()
|
||||
|
||||
CONF.aprs_fi.apiKey = "abc123"
|
||||
CONF.aprs_fi.apiKey = 'abc123'
|
||||
CONF.callsign = fake.FAKE_TO_CALLSIGN
|
||||
wx = weather_plugin.USMetarPlugin()
|
||||
wx.enabled = True
|
||||
expected = "BOGUSMETAR"
|
||||
packet = fake.fake_packet(message="metar KPAO")
|
||||
expected = 'BOGUSMETAR'
|
||||
packet = fake.fake_packet(message='metar KPAO')
|
||||
actual = wx.filter(packet)
|
||||
self.assertEqual(expected, actual)
|
||||
|
||||
@mock.patch("aprsd.plugin_utils.get_weather_gov_metar")
|
||||
@mock.patch("aprsd.plugin_utils.get_aprs_fi")
|
||||
@mock.patch("aprsd.plugin_utils.get_weather_gov_for_gps")
|
||||
@mock.patch('aprsd.plugin_utils.get_weather_gov_metar')
|
||||
@mock.patch('aprsd.plugin_utils.get_aprs_fi')
|
||||
@mock.patch('aprsd.plugin_utils.get_weather_gov_for_gps')
|
||||
def test_metar_works(self, mock_wx_for_gps, mock_check_aprs, mock_metar):
|
||||
mock_wx_for_gps.return_value = {
|
||||
"location": {"metar": "BOGUSMETAR"},
|
||||
'location': {'metar': 'BOGUSMETAR'},
|
||||
}
|
||||
|
||||
class Response:
|
||||
text = '{"properties": {"rawMessage": "BOGUSMETAR"}}'
|
||||
|
||||
mock_check_aprs.return_value = {
|
||||
"entries": [
|
||||
'entries': [
|
||||
{
|
||||
"lat": 10,
|
||||
"lng": 11,
|
||||
"lasttime": 10,
|
||||
'lat': 10,
|
||||
'lng': 11,
|
||||
'lasttime': 10,
|
||||
},
|
||||
],
|
||||
}
|
||||
mock_metar.return_value = Response()
|
||||
|
||||
CONF.aprs_fi.apiKey = "abc123"
|
||||
CONF.aprs_fi.apiKey = 'abc123'
|
||||
CONF.callsign = fake.FAKE_TO_CALLSIGN
|
||||
wx = weather_plugin.USMetarPlugin()
|
||||
wx.enabled = True
|
||||
expected = "BOGUSMETAR"
|
||||
packet = fake.fake_packet(message="metar")
|
||||
expected = 'BOGUSMETAR'
|
||||
packet = fake.fake_packet(message='metar')
|
||||
actual = wx.filter(packet)
|
||||
self.assertEqual(expected, actual)
|
||||
|
||||
Reference in New Issue
Block a user