kopia lustrzana https://github.com/glidernet/ogn-python
149 wiersze
6.2 KiB
Python
149 wiersze
6.2 KiB
Python
import unittest
|
|
from unittest import mock
|
|
import os
|
|
from datetime import datetime
|
|
|
|
from xmlunittest import XmlTestMixin
|
|
|
|
from ogn.model import AircraftBeacon, Receiver, Device, DeviceInfo
|
|
|
|
from ogn.backend.liveglidernet import rec, lxml
|
|
from ogn.backend.ognrange import stations2_filtered_pl
|
|
from ogn.model.aircraft_type import AircraftType
|
|
|
|
|
|
class TestDB(unittest.TestCase, XmlTestMixin):
|
|
session = None
|
|
engine = None
|
|
app = None
|
|
|
|
def setUp(self):
|
|
os.environ['OGN_CONFIG_MODULE'] = 'config.test'
|
|
from ogn.commands.dbutils import engine, session
|
|
self.session = session
|
|
self.engine = engine
|
|
|
|
from ogn.commands.database import init
|
|
init()
|
|
|
|
# Prepare Beacons
|
|
self.r01 = Receiver(name='Koenigsdf', location_wkt='0101000020E610000061E8FED7A6EE26407F20661C10EA4740', lastseen='2017-12-20 10:00:00', altitude=601, version='0.2.5', platform='ARM')
|
|
self.r02 = Receiver(name='Bene', location_wkt='0101000020E6100000D5E76A2BF6C72640D4063A6DA0DB4740', lastseen='2017-12-20 09:45:00', altitude=609, version='0.2.7', platform='x64')
|
|
self.r03 = Receiver(name='Ohlstadt', location_wkt='0101000020E6100000057E678EBF772640A142883E32D44740', lastseen='2017-12-20 10:05:00', altitude=655, version='0.2.6', platform='ARM')
|
|
session.add(self.r01)
|
|
session.add(self.r02)
|
|
session.add(self.r03)
|
|
session.commit()
|
|
|
|
self.d01 = Device(address='DD4711', lastseen='2017-12-20 10:00:02')
|
|
self.d02 = Device(address='DD0815', lastseen='2017-12-20 09:56:00')
|
|
session.add(self.d01)
|
|
session.add(self.d02)
|
|
session.commit()
|
|
|
|
self.di01 = DeviceInfo(registration='D-4711', competition='Hi', tracked=True, identified=True, device_id=self.d01.id)
|
|
session.add(self.di01)
|
|
session.commit()
|
|
|
|
self.ab11 = AircraftBeacon(location_wkt='0101000020E6100000211FF46C56ED26402650D7EDC6E94740', aircraft_type=AircraftType.glider_or_motor_glider, receiver_name='Koenigsdf', timestamp='2017-12-20 10:00:01', track=105, ground_speed=57, climb_rate=-0.5, device_id=self.d01.id)
|
|
self.ab12 = AircraftBeacon(location_wkt='0101000020E6100000806DEA295FED2640347D898BB6E94740', aircraft_type=AircraftType.glider_or_motor_glider, receiver_name='Koenigsdf', timestamp='2017-12-20 10:00:02', track=123, ground_speed=55, climb_rate=-0.4, altitude=209, device_id=self.d01.id)
|
|
self.ab21 = AircraftBeacon(location_wkt='0101000020E6100000F38B25BF58F22640448B6CE7FBE94740', aircraft_type=AircraftType.powered_aircraft, receiver_name='Koenigsdf', timestamp='2017-12-20 09:54:30', track=280, ground_speed=80, climb_rate=-2.9, device_id=self.d02.id)
|
|
self.ab22 = AircraftBeacon(location_wkt='0101000020E6100000A5E8482EFFF12640DC1EAA16FEE94740', aircraft_type=AircraftType.powered_aircraft, receiver_name='Bene', timestamp='2017-12-20 09:56:00', track=270, ground_speed=77, climb_rate=-1.5, altitude=543, device_id=self.d02.id)
|
|
session.add(self.ab11)
|
|
session.add(self.ab12)
|
|
session.add(self.ab21)
|
|
session.add(self.ab22)
|
|
session.commit()
|
|
|
|
def tearDown(self):
|
|
session = self.session
|
|
session.execute("DELETE FROM device_infos")
|
|
session.execute("DELETE FROM receivers")
|
|
session.execute("DELETE FROM devices")
|
|
session.execute("DELETE FROM aircraft_beacons")
|
|
session.commit()
|
|
|
|
@mock.patch('ogn.backend.liveglidernet.datetime')
|
|
def test_rec(self, datetime_mock):
|
|
session = self.session
|
|
|
|
datetime_mock.utcnow.return_value = datetime(2017, 12, 20, 10, 0)
|
|
|
|
data = rec(session).encode(encoding='utf-8')
|
|
|
|
# Check the document
|
|
root = self.assertXmlDocument(data)
|
|
self.assertXmlNode(root, tag='markers')
|
|
self.assertXpathsOnlyOne(root, ('./m[@a="Koenigsdf"]', './m[@a="Bene"]', './m[@a="Ohlstadt"]'))
|
|
|
|
# Check the complete document
|
|
expected = """<?xml version="1.0" encoding="UTF-8"?>
|
|
<markers>
|
|
<m e="0"/>
|
|
<m a="Bene" b="47.7158333" c="11.3905500" d="0"/>
|
|
<m a="Koenigsdf" b="47.8286167" c="11.4661167" d="1"/>
|
|
<m a="Ohlstadt" b="47.6577833" c="11.2338833" d="1"/>
|
|
</markers>
|
|
""".encode(encoding='utf-8')
|
|
|
|
self.assertXmlEquivalentOutputs(data, expected)
|
|
|
|
@mock.patch('ogn.backend.liveglidernet.utc_to_local', side_effect=lambda x: x)
|
|
@mock.patch('ogn.backend.liveglidernet.datetime')
|
|
def test_lxml(self, datetime_mock, utc_to_local_mock):
|
|
session = self.session
|
|
|
|
datetime_mock.utcnow.return_value = datetime(2017, 12, 20, 10, 0, 5)
|
|
|
|
data = lxml(session).encode(encoding='utf-8')
|
|
|
|
# Check the complete document
|
|
expected = """<?xml version="1.0" encoding="UTF-8"?>
|
|
<markers>
|
|
<m a="47.8258833,11.4636167,Hi,D-4711,209,10:00:02,3,123,55,-0.4,1,Koenigsdf,DD4711,xxDD4711"/>
|
|
<m a="47.8280667,11.4726500,_15,xxDD0815,543,09:56:00,245,270,77,-1.5,8,Bene,0,xxDD0815"/>
|
|
</markers>
|
|
""".encode(encoding='utf-8')
|
|
|
|
self.assertXmlEquivalentOutputs(data, expected)
|
|
|
|
@mock.patch('ogn.backend.ognrange.datetime')
|
|
def test_stations2_filtered_pl(self, datetime_mock):
|
|
session = self.session
|
|
|
|
datetime_mock.utcnow.return_value = datetime(2017, 12, 20, 10, 0)
|
|
|
|
import json
|
|
|
|
result = stations2_filtered_pl(session)
|
|
|
|
data = json.loads(result)
|
|
|
|
stations = data["stations"]
|
|
self.assertEqual(len(stations), 3)
|
|
s1 = stations[0]
|
|
s2 = stations[1]
|
|
s3 = stations[2]
|
|
|
|
self.assertEqual(s1["s"], 'Bene')
|
|
self.assertEqual(s1["lt"], 47.7158)
|
|
self.assertEqual(s1["lg"], 11.3906)
|
|
self.assertEqual(s1["u"], "D") # Down, because last beacon > 10min. ago
|
|
self.assertEqual(s1["ut"], "2017-12-20 09:45")
|
|
# self.assertEqual(s1["b"], 0)
|
|
self.assertEqual(s1["v"], "0.2.7.x64")
|
|
|
|
self.assertEqual(s2["s"], 'Koenigsdf')
|
|
self.assertEqual(s2["lt"], 47.8286)
|
|
self.assertEqual(s2["lg"], 11.4661)
|
|
self.assertEqual(s2["u"], "U")
|
|
self.assertEqual(s2["ut"], "2017-12-20 10:00")
|
|
# self.assertEqual(s2["b"], 0)
|
|
self.assertEqual(s2["v"], "0.2.5.ARM")
|
|
|
|
self.assertEqual(s3["s"], 'Ohlstadt')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|