mirror of
https://github.com/craigerl/aprsd.git
synced 2026-06-11 02:18:40 -04:00
Added unit tests for packets.
Also did some code cleanup.
This commit is contained in:
+50
-50
@@ -20,18 +20,18 @@ class TestPacketBase(unittest.TestCase):
|
||||
message_format=core.PACKET_TYPE_MESSAGE,
|
||||
):
|
||||
packet_dict = {
|
||||
"from": from_call,
|
||||
"addresse": to_call,
|
||||
"to": to_call,
|
||||
"format": message_format,
|
||||
"raw": "",
|
||||
'from': from_call,
|
||||
'addresse': to_call,
|
||||
'to': to_call,
|
||||
'format': message_format,
|
||||
'raw': '',
|
||||
}
|
||||
|
||||
if message:
|
||||
packet_dict["message_text"] = message
|
||||
packet_dict['message_text'] = message
|
||||
|
||||
if msg_number:
|
||||
packet_dict["msgNo"] = str(msg_number)
|
||||
packet_dict['msgNo'] = str(msg_number)
|
||||
|
||||
return packet_dict
|
||||
|
||||
@@ -52,7 +52,7 @@ class TestPacketBase(unittest.TestCase):
|
||||
|
||||
self.assertEqual(
|
||||
fake.FAKE_FROM_CALLSIGN,
|
||||
pkt.get("from_call"),
|
||||
pkt.get('from_call'),
|
||||
)
|
||||
|
||||
def test_packet_factory(self):
|
||||
@@ -64,21 +64,21 @@ class TestPacketBase(unittest.TestCase):
|
||||
self.assertEqual(fake.FAKE_TO_CALLSIGN, pkt.to_call)
|
||||
self.assertEqual(fake.FAKE_TO_CALLSIGN, pkt.addresse)
|
||||
|
||||
pkt_dict["symbol"] = "_"
|
||||
pkt_dict["weather"] = {
|
||||
"wind_gust": 1.11,
|
||||
"temperature": 32.01,
|
||||
"humidity": 85,
|
||||
"pressure": 1095.12,
|
||||
"comment": "Home!",
|
||||
pkt_dict['symbol'] = '_'
|
||||
pkt_dict['weather'] = {
|
||||
'wind_gust': 1.11,
|
||||
'temperature': 32.01,
|
||||
'humidity': 85,
|
||||
'pressure': 1095.12,
|
||||
'comment': 'Home!',
|
||||
}
|
||||
pkt_dict["format"] = core.PACKET_TYPE_UNCOMPRESSED
|
||||
pkt_dict['format'] = core.PACKET_TYPE_UNCOMPRESSED
|
||||
pkt = packets.factory(pkt_dict)
|
||||
self.assertIsInstance(pkt, packets.WeatherPacket)
|
||||
|
||||
@mock.patch("aprsd.packets.core.GPSPacket._build_time_zulu")
|
||||
@mock.patch('aprsd.packets.core.GPSPacket._build_time_zulu')
|
||||
def test_packet_format_rain_1h(self, mock_time_zulu):
|
||||
mock_time_zulu.return_value = "221450"
|
||||
mock_time_zulu.return_value = '221450'
|
||||
|
||||
wx = packets.WeatherPacket(
|
||||
from_call=fake.FAKE_FROM_CALLSIGN,
|
||||
@@ -87,58 +87,58 @@ class TestPacketBase(unittest.TestCase):
|
||||
)
|
||||
wx.prepare()
|
||||
|
||||
expected = "KFAKE>KMINE,WIDE1-1,WIDE2-1:@221450z0.0/0.0_000/000g000t000r000p000P000h00b00000"
|
||||
expected = 'KFAKE>KMINE,WIDE1-1,WIDE2-1:@221450z0.0/0.0_000/000g000t000r000p000P000h00b00000'
|
||||
self.assertEqual(expected, wx.raw)
|
||||
rain_location = 59
|
||||
self.assertEqual(rain_location, wx.raw.find("r000"))
|
||||
self.assertEqual(rain_location, wx.raw.find('r000'))
|
||||
|
||||
wx.rain_1h = 1.11
|
||||
wx.prepare()
|
||||
expected = "KFAKE>KMINE,WIDE1-1,WIDE2-1:@221450z0.0/0.0_000/000g000t000r111p000P000h00b00000"
|
||||
expected = 'KFAKE>KMINE,WIDE1-1,WIDE2-1:@221450z0.0/0.0_000/000g000t000r111p000P000h00b00000'
|
||||
self.assertEqual(expected, wx.raw)
|
||||
|
||||
wx.rain_1h = 0.01
|
||||
wx.prepare()
|
||||
expected = "KFAKE>KMINE,WIDE1-1,WIDE2-1:@221450z0.0/0.0_000/000g000t000r001p000P000h00b00000"
|
||||
expected = 'KFAKE>KMINE,WIDE1-1,WIDE2-1:@221450z0.0/0.0_000/000g000t000r001p000P000h00b00000'
|
||||
self.assertEqual(expected, wx.raw)
|
||||
|
||||
def test_beacon_factory(self):
|
||||
"""Test to ensure a beacon packet is created."""
|
||||
packet_raw = (
|
||||
"WB4BOR-12>APZ100,WIDE2-1:@161647z3724.15N107847.58W$ APRSD WebChat"
|
||||
'WB4BOR-12>APZ100,WIDE2-1:@161647z3724.15N107847.58W$ APRSD WebChat'
|
||||
)
|
||||
packet_dict = aprslib.parse(packet_raw)
|
||||
packet = packets.factory(packet_dict)
|
||||
self.assertIsInstance(packet, packets.BeaconPacket)
|
||||
|
||||
packet_raw = "kd8mey-10>APRS,TCPIP*,qAC,T2SYDNEY:=4247.80N/08539.00WrPHG1210/Making 220 Great Again Allstar# 552191"
|
||||
packet_raw = 'kd8mey-10>APRS,TCPIP*,qAC,T2SYDNEY:=4247.80N/08539.00WrPHG1210/Making 220 Great Again Allstar# 552191'
|
||||
packet_dict = aprslib.parse(packet_raw)
|
||||
packet = packets.factory(packet_dict)
|
||||
self.assertIsInstance(packet, packets.BeaconPacket)
|
||||
|
||||
def test_reject_factory(self):
|
||||
"""Test to ensure a reject packet is created."""
|
||||
packet_raw = "HB9FDL-1>APK102,HB9FM-4*,WIDE2,qAR,HB9FEF-11::REPEAT :rej4139"
|
||||
packet_raw = 'HB9FDL-1>APK102,HB9FM-4*,WIDE2,qAR,HB9FEF-11::REPEAT :rej4139'
|
||||
packet_dict = aprslib.parse(packet_raw)
|
||||
packet = packets.factory(packet_dict)
|
||||
self.assertIsInstance(packet, packets.RejectPacket)
|
||||
|
||||
self.assertEqual("4139", packet.msgNo)
|
||||
self.assertEqual("HB9FDL-1", packet.from_call)
|
||||
self.assertEqual("REPEAT", packet.to_call)
|
||||
self.assertEqual("reject", packet.packet_type)
|
||||
self.assertEqual('4139', packet.msgNo)
|
||||
self.assertEqual('HB9FDL-1', packet.from_call)
|
||||
self.assertEqual('REPEAT', packet.to_call)
|
||||
self.assertEqual('reject', packet.packet_type)
|
||||
self.assertIsNone(packet.payload)
|
||||
|
||||
def test_thirdparty_factory(self):
|
||||
"""Test to ensure a third party packet is created."""
|
||||
packet_raw = "GTOWN>APDW16,WIDE1-1,WIDE2-1:}KM6LYW-9>APZ100,TCPIP,GTOWN*::KM6LYW :KM6LYW: 19 Miles SW"
|
||||
packet_raw = 'GTOWN>APDW16,WIDE1-1,WIDE2-1:}KM6LYW-9>APZ100,TCPIP,GTOWN*::KM6LYW :KM6LYW: 19 Miles SW'
|
||||
packet_dict = aprslib.parse(packet_raw)
|
||||
packet = packets.factory(packet_dict)
|
||||
self.assertIsInstance(packet, packets.ThirdPartyPacket)
|
||||
|
||||
def test_weather_factory(self):
|
||||
"""Test to ensure a weather packet is created."""
|
||||
packet_raw = "FW9222>APRS,TCPXX*,qAX,CWOP-6:@122025z2953.94N/08423.77W_232/003g006t084r000p032P000h80b10157L745.DsWLL"
|
||||
packet_raw = 'FW9222>APRS,TCPXX*,qAX,CWOP-6:@122025z2953.94N/08423.77W_232/003g006t084r000p032P000h80b10157L745.DsWLL'
|
||||
packet_dict = aprslib.parse(packet_raw)
|
||||
packet = packets.factory(packet_dict)
|
||||
self.assertIsInstance(packet, packets.WeatherPacket)
|
||||
@@ -178,7 +178,7 @@ class TestPacketBase(unittest.TestCase):
|
||||
)
|
||||
|
||||
expected = (
|
||||
f"{fake.FAKE_FROM_CALLSIGN}>APZ100::{fake.FAKE_TO_CALLSIGN:<9}:ack123"
|
||||
f'{fake.FAKE_FROM_CALLSIGN}>APZ100::{fake.FAKE_TO_CALLSIGN:<9}:ack123'
|
||||
)
|
||||
self.assertEqual(expected, str(ack))
|
||||
|
||||
@@ -191,7 +191,7 @@ class TestPacketBase(unittest.TestCase):
|
||||
)
|
||||
|
||||
expected = (
|
||||
f"{fake.FAKE_FROM_CALLSIGN}>APZ100::{fake.FAKE_TO_CALLSIGN:<9}:rej123"
|
||||
f'{fake.FAKE_FROM_CALLSIGN}>APZ100::{fake.FAKE_TO_CALLSIGN:<9}:rej123'
|
||||
)
|
||||
self.assertEqual(expected, str(reject))
|
||||
|
||||
@@ -200,20 +200,20 @@ class TestPacketBase(unittest.TestCase):
|
||||
lat = 28.123456
|
||||
lon = -80.123456
|
||||
ts = 1711219496.6426
|
||||
comment = "My Beacon Comment"
|
||||
comment = 'My Beacon Comment'
|
||||
packet = packets.BeaconPacket(
|
||||
from_call=fake.FAKE_FROM_CALLSIGN,
|
||||
to_call=fake.FAKE_TO_CALLSIGN,
|
||||
latitude=lat,
|
||||
longitude=lon,
|
||||
timestamp=ts,
|
||||
symbol=">",
|
||||
symbol='>',
|
||||
comment=comment,
|
||||
)
|
||||
|
||||
expected_lat = aprslib_util.latitude_to_ddm(lat)
|
||||
expected_lon = aprslib_util.longitude_to_ddm(lon)
|
||||
expected = f"KFAKE>APZ100:@231844z{expected_lat}/{expected_lon}>{comment}"
|
||||
expected = f'KFAKE>APZ100:@231844z{expected_lat}/{expected_lon}>{comment}'
|
||||
self.assertEqual(expected, str(packet))
|
||||
|
||||
def test_beacon_format_no_comment(self):
|
||||
@@ -227,13 +227,13 @@ class TestPacketBase(unittest.TestCase):
|
||||
latitude=lat,
|
||||
longitude=lon,
|
||||
timestamp=ts,
|
||||
symbol=">",
|
||||
symbol='>',
|
||||
)
|
||||
empty_comment = "APRSD Beacon"
|
||||
empty_comment = 'APRSD Beacon'
|
||||
|
||||
expected_lat = aprslib_util.latitude_to_ddm(lat)
|
||||
expected_lon = aprslib_util.longitude_to_ddm(lon)
|
||||
expected = f"KFAKE>APZ100:@231844z{expected_lat}/{expected_lon}>{empty_comment}"
|
||||
expected = f'KFAKE>APZ100:@231844z{expected_lat}/{expected_lon}>{empty_comment}'
|
||||
self.assertEqual(expected, str(packet))
|
||||
|
||||
def test_bulletin_format(self):
|
||||
@@ -242,32 +242,32 @@ class TestPacketBase(unittest.TestCase):
|
||||
bid = 0
|
||||
packet = packets.BulletinPacket(
|
||||
from_call=fake.FAKE_FROM_CALLSIGN,
|
||||
message_text="My Bulletin Message",
|
||||
message_text='My Bulletin Message',
|
||||
bid=0,
|
||||
)
|
||||
|
||||
expected = (
|
||||
f"{fake.FAKE_FROM_CALLSIGN}>APZ100::BLN{bid:<9}:{packet.message_text}"
|
||||
f'{fake.FAKE_FROM_CALLSIGN}>APZ100::BLN{bid:<9}:{packet.message_text}'
|
||||
)
|
||||
self.assertEqual(expected, str(packet))
|
||||
|
||||
# bulletin id = 1
|
||||
bid = 1
|
||||
txt = "((((((( CX2SA - Salto Uruguay ))))))) http://www.cx2sa.org"
|
||||
txt = '((((((( CX2SA - Salto Uruguay ))))))) http://www.cx2sa.org'
|
||||
packet = packets.BulletinPacket(
|
||||
from_call=fake.FAKE_FROM_CALLSIGN,
|
||||
message_text=txt,
|
||||
bid=1,
|
||||
)
|
||||
|
||||
expected = f"{fake.FAKE_FROM_CALLSIGN}>APZ100::BLN{bid:<9}:{txt}"
|
||||
expected = f'{fake.FAKE_FROM_CALLSIGN}>APZ100::BLN{bid:<9}:{txt}'
|
||||
self.assertEqual(expected, str(packet))
|
||||
|
||||
def test_message_format(self):
|
||||
"""Test the message packet format."""
|
||||
|
||||
message = "My Message"
|
||||
msgno = "ABX"
|
||||
message = 'My Message'
|
||||
msgno = 'ABX'
|
||||
packet = packets.MessagePacket(
|
||||
from_call=fake.FAKE_FROM_CALLSIGN,
|
||||
to_call=fake.FAKE_TO_CALLSIGN,
|
||||
@@ -275,19 +275,19 @@ class TestPacketBase(unittest.TestCase):
|
||||
msgNo=msgno,
|
||||
)
|
||||
|
||||
expected = f"{fake.FAKE_FROM_CALLSIGN}>APZ100::{fake.FAKE_TO_CALLSIGN:<9}:{message}{{{msgno}"
|
||||
expected = f'{fake.FAKE_FROM_CALLSIGN}>APZ100::{fake.FAKE_TO_CALLSIGN:<9}:{message}{{{msgno}'
|
||||
self.assertEqual(expected, str(packet))
|
||||
|
||||
# test with bad words
|
||||
# Currently fails with mixed case
|
||||
message = "My cunt piss fuck shIt text"
|
||||
exp_msg = "My **** **** **** **** text"
|
||||
msgno = "ABX"
|
||||
message = 'My cunt piss fuck shIt text'
|
||||
exp_msg = 'My **** **** **** **** text'
|
||||
msgno = 'ABX'
|
||||
packet = packets.MessagePacket(
|
||||
from_call=fake.FAKE_FROM_CALLSIGN,
|
||||
to_call=fake.FAKE_TO_CALLSIGN,
|
||||
message_text=message,
|
||||
msgNo=msgno,
|
||||
)
|
||||
expected = f"{fake.FAKE_FROM_CALLSIGN}>APZ100::{fake.FAKE_TO_CALLSIGN:<9}:{exp_msg}{{{msgno}"
|
||||
expected = f'{fake.FAKE_FROM_CALLSIGN}>APZ100::{fake.FAKE_TO_CALLSIGN:<9}:{exp_msg}{{{msgno}'
|
||||
self.assertEqual(expected, str(packet))
|
||||
|
||||
Reference in New Issue
Block a user