kopia lustrzana https://github.com/glidernet/ogn-python
39 wiersze
1.4 KiB
Python
39 wiersze
1.4 KiB
Python
import os
|
|
import unittest
|
|
|
|
from ogn.model import AircraftType
|
|
from ogn.utils import get_ddb, get_trackable, get_country_code, get_airports
|
|
import unittest.mock as mock
|
|
|
|
|
|
class TestStringMethods(unittest.TestCase):
|
|
def test_get_devices(self):
|
|
devices = get_ddb()
|
|
self.assertGreater(len(devices), 1000)
|
|
|
|
def test_get_ddb_from_file(self):
|
|
devices = get_ddb(os.path.dirname(__file__) + '/custom_ddb.txt')
|
|
self.assertEqual(len(devices), 6)
|
|
device = devices[0]
|
|
|
|
self.assertEqual(device.address, 'DD4711')
|
|
self.assertEqual(device.aircraft, 'HK36 TTC')
|
|
self.assertEqual(device.registration, 'D-EULE')
|
|
self.assertEqual(device.competition, 'CU')
|
|
self.assertTrue(device.tracked)
|
|
self.assertTrue(device.identified)
|
|
self.assertEqual(device.aircraft_type, AircraftType.glider_or_motor_glider)
|
|
|
|
def test_get_trackable(self):
|
|
devices = get_ddb(os.path.dirname(__file__) + '/custom_ddb.txt')
|
|
trackable = get_trackable(devices)
|
|
self.assertEqual(len(trackable), 4)
|
|
self.assertIn('FLRDD4711', trackable)
|
|
self.assertIn('FLRDD0815', trackable)
|
|
self.assertIn('OGNDEADBE', trackable)
|
|
self.assertIn('ICA999999', trackable)
|
|
|
|
def test_get_airports(self):
|
|
airports = get_airports(os.path.dirname(__file__) + '/SeeYou.cup')
|
|
self.assertGreater(len(airports), 1000)
|