import unittest from unittest import mock import aprslib from aprslib import util as aprslib_util from aprsd import packets from aprsd.packets import core from . import fake class TestPacketBase(unittest.TestCase): def _fake_dict( self, from_call=fake.FAKE_FROM_CALLSIGN, to_call=fake.FAKE_TO_CALLSIGN, message=None, msg_number=None, message_format=core.PACKET_TYPE_MESSAGE, ): packet_dict = { "from": from_call, "addresse": to_call, "to": to_call, "format": message_format, "raw": "", } if message: packet_dict["message_text"] = message if msg_number: packet_dict["msgNo"] = str(msg_number) return packet_dict def test_packet_construct(self): pkt = packets.Packet( from_call=fake.FAKE_FROM_CALLSIGN, to_call=fake.FAKE_TO_CALLSIGN, ) self.assertEqual(fake.FAKE_FROM_CALLSIGN, pkt.from_call) self.assertEqual(fake.FAKE_TO_CALLSIGN, pkt.to_call) def test_packet_get_attr(self): pkt = packets.Packet( from_call=fake.FAKE_FROM_CALLSIGN, to_call=fake.FAKE_TO_CALLSIGN, ) self.assertEqual( fake.FAKE_FROM_CALLSIGN, pkt.get("from_call"), ) def test_packet_factory(self): pkt_dict = self._fake_dict() pkt = packets.factory(pkt_dict) self.assertIsInstance(pkt, packets.MessagePacket) self.assertEqual(fake.FAKE_FROM_CALLSIGN, pkt.from_call) 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["format"] = core.PACKET_TYPE_UNCOMPRESSED pkt = packets.factory(pkt_dict) self.assertIsInstance(pkt, packets.WeatherPacket) @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" wx = packets.WeatherPacket( from_call=fake.FAKE_FROM_CALLSIGN, to_call=fake.FAKE_TO_CALLSIGN, timestamp=1671721164.1112509, ) wx.prepare() 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")) wx.rain_1h = 1.11 wx.prepare() 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" 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" 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_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_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.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_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_dict = aprslib.parse(packet_raw) packet = packets.factory(packet_dict) self.assertIsInstance(packet, packets.WeatherPacket) self.assertEqual(28.88888888888889, packet.temperature) self.assertEqual(0.0, packet.rain_1h) self.assertEqual(1015.7, packet.pressure) self.assertEqual(80, packet.humidity) self.assertEqual(745, packet.luminosity) self.assertEqual(3.0, packet.wind_speed) self.assertEqual(232, packet.wind_direction) self.assertEqual(6.0, packet.wind_gust) self.assertEqual(29.899, packet.latitude) self.assertEqual(-84.39616666666667, packet.longitude) def test_mice_factory(self): packet_raw = 'kh2sr-15>S7TSYR,WIDE1-1,WIDE2-1,qAO,KO6KL-1:`1`7\x1c\x1c.#/`"4,}QuirkyQRP 4.6V 35.3C S06' packet_dict = aprslib.parse(packet_raw) packet = packets.factory(packet_dict) self.assertIsInstance(packet, packets.MicEPacket) # Packet with telemetry and DAO # http://www.aprs.org/datum.txt packet_raw = 'KD9YIL>T0PX9W,WIDE1-1,WIDE2-1,qAO,NU9R-10:`sB,l#P>/\'"6+}|#*%U\'a|!whl!|3' packet_dict = aprslib.parse(packet_raw) packet = packets.factory(packet_dict) self.assertIsInstance(packet, packets.MicEPacket) def test_ack_format(self): """Test the ack packet format.""" ack = packets.AckPacket( from_call=fake.FAKE_FROM_CALLSIGN, to_call=fake.FAKE_TO_CALLSIGN, msgNo=123, ) expected = f"{fake.FAKE_FROM_CALLSIGN}>APZ100::{fake.FAKE_TO_CALLSIGN:<9}:ack123" self.assertEqual(expected, str(ack)) def test_reject_format(self): """Test the reject packet format.""" reject = packets.RejectPacket( from_call=fake.FAKE_FROM_CALLSIGN, to_call=fake.FAKE_TO_CALLSIGN, msgNo=123, ) expected = f"{fake.FAKE_FROM_CALLSIGN}>APZ100::{fake.FAKE_TO_CALLSIGN:<9}:rej123" self.assertEqual(expected, str(reject)) def test_beacon_format(self): """Test the beacon packet format.""" lat = 28.123456 lon = -80.123456 ts = 1711219496.6426 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=">", 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}" self.assertEqual(expected, str(packet)) def test_beacon_format_no_comment(self): """Test the beacon packet format.""" lat = 28.123456 lon = -80.123456 ts = 1711219496.6426 packet = packets.BeaconPacket( from_call=fake.FAKE_FROM_CALLSIGN, to_call=fake.FAKE_TO_CALLSIGN, latitude=lat, longitude=lon, timestamp=ts, symbol=">", ) 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}" self.assertEqual(expected, str(packet)) def test_bulletin_format(self): """Test the bulletin packet format.""" # bulletin id = 0 bid = 0 packet = packets.BulletinPacket( from_call=fake.FAKE_FROM_CALLSIGN, message_text="My Bulletin Message", bid=0, ) expected = 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" 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}" self.assertEqual(expected, str(packet)) def test_message_format(self): """Test the message packet format.""" message = "My Message" 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}:{message}{{{msgno}" self.assertEqual(expected, str(packet)) # test with bad words # Currently fails with mixed case message = "My cunt piss fuck 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}" self.assertEqual(expected, str(packet))