Add test cases to support no-tracking flag

glidernet · Apr 5, 2021 · f11e0ef · f11e0ef
1 parent 029b7ad
commit f11e0ef
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/tests/parser/test_parse_flarm.py b/tests/parser/test_parse_flarm.py
@@ -11,6 +11,7 @@ def test_position_comment(self):
         self.assertEqual(message['address_type'], 1)
         self.assertEqual(message['aircraft_type'], 8)
         self.assertFalse(message['stealth'])
+        self.assertFalse(message['no-tracking'])
         self.assertEqual(message['address'], "A8CBA8")
         self.assertAlmostEqual(message['climb_rate'], -39 * FPM_TO_MS, 2)
         self.assertEqual(message['turn_rate'], 0.1 * HPM_TO_DEGS)
@@ -27,7 +28,7 @@ def test_position_comment_relevant_keys_only(self):
         message = FlarmParser().parse_position("id21A8CBA8")
 
         self.assertIsNotNone(message)
-        self.assertEqual(sorted(message.keys()), sorted(['address_type', 'aircraft_type', 'stealth', 'address']))
+        self.assertEqual(sorted(message.keys()), sorted(['address_type', 'aircraft_type', 'stealth', 'address', 'no-tracking']))
 
 
 if __name__ == '__main__':

diff --git a/tests/parser/test_parse_ogn_aircraft.py b/tests/parser/test_parse_ogn_aircraft.py
@@ -14,6 +14,7 @@ def test_basic(self):
         self.assertEqual(message['address_type'], 2)
         self.assertEqual(message['aircraft_type'], 2)
         self.assertFalse(message['stealth'])
+        self.assertFalse(message['no-tracking'])
         self.assertEqual(message['address'], "DDA5BA")
         self.assertAlmostEqual(message['climb_rate'], -454 * FPM_TO_MS, 2)
         self.assertEqual(message['turn_rate'], -1.1 * HPM_TO_DEGS)
@@ -26,6 +27,13 @@ def test_basic(self):
         self.assertEqual(message['proximity'][1], 'B597')
         self.assertEqual(message['proximity'][2], 'B598')
 
+    def test_no_tracking(self):
+        message = OgnParser().parse_aircraft_beacon("id0ADD1234 -454fpm -1.1rot 8.8dB 0e +51.2kHz gps4x5 hear1084 hearB597 hearB598")
+        self.assertFalse(message['no-tracking'])
+
+        message = OgnParser().parse_aircraft_beacon("id4ADD1234 -454fpm -1.1rot 8.8dB 0e +51.2kHz gps4x5 hear1084 hearB597 hearB598")
+        self.assertTrue(message['no-tracking'])
+
     def test_stealth(self):
         message = OgnParser().parse_aircraft_beacon("id0ADD1234 -454fpm -1.1rot 8.8dB 0e +51.2kHz gps4x5 hear1084 hearB597 hearB598")
         self.assertFalse(message['stealth'])
@@ -63,7 +71,7 @@ def test_relevant_keys_only(self):
         message = OgnParser().parse_aircraft_beacon("id093D0930")
 
         self.assertIsNotNone(message)
-        self.assertEqual(sorted(message.keys()), sorted(['address_type', 'aircraft_type', 'stealth', 'address']))
+        self.assertEqual(sorted(message.keys()), sorted(['address_type', 'aircraft_type', 'stealth', 'address', 'no-tracking']))
 
 
 if __name__ == '__main__':