kopia lustrzana https://github.com/glidernet/ogn-python
144 wiersze
5.8 KiB
Python
144 wiersze
5.8 KiB
Python
import datetime
|
|
import unittest
|
|
|
|
from tests.base import TestBaseDB, db
|
|
|
|
from app.model import Logbook, Airport, Sender, TakeoffLanding
|
|
from app.collect.logbook import update_logbook
|
|
|
|
|
|
class TestLogbook(TestBaseDB):
|
|
def setUp(self):
|
|
super().setUp()
|
|
|
|
# Create basic data and insert
|
|
self.dd0815 = Sender(name="FLRDD0815", address="DD0815")
|
|
self.dd4711 = Sender(name="FLRDD4711", address="DD4711")
|
|
|
|
self.koenigsdorf = Airport(name="Koenigsdorf")
|
|
self.ohlstadt = Airport(name="Ohlstadt")
|
|
|
|
db.session.add(self.dd0815)
|
|
db.session.add(self.dd4711)
|
|
db.session.add(self.koenigsdorf)
|
|
db.session.add(self.ohlstadt)
|
|
|
|
db.session.commit()
|
|
|
|
# Prepare takeoff and landings
|
|
self.takeoff_koenigsdorf_dd0815 = TakeoffLanding(is_takeoff=True, timestamp="2016-06-01 10:00:00", airport_id=self.koenigsdorf.id, sender_id=self.dd0815.id)
|
|
self.landing_koenigsdorf_dd0815 = TakeoffLanding(is_takeoff=False, timestamp="2016-06-01 10:05:00", airport_id=self.koenigsdorf.id, sender_id=self.dd0815.id)
|
|
self.landing_koenigsdorf_dd0815_later = TakeoffLanding(is_takeoff=False, timestamp="2016-06-02 10:05:00", airport_id=self.koenigsdorf.id, sender_id=self.dd0815.id)
|
|
self.takeoff_ohlstadt_dd4711 = TakeoffLanding(is_takeoff=True, timestamp="2016-06-01 10:00:00", airport_id=self.ohlstadt.id, sender_id=self.dd4711.id)
|
|
|
|
def get_logbook_entries(self):
|
|
return db.session.query(Logbook).order_by(Logbook.takeoff_airport_id, Logbook.reference_timestamp).all()
|
|
|
|
def test_single_takeoff(self):
|
|
db.session.add(self.takeoff_koenigsdorf_dd0815)
|
|
db.session.commit()
|
|
|
|
update_logbook()
|
|
entries = self.get_logbook_entries()
|
|
self.assertEqual(len(entries), 1)
|
|
self.assertEqual(entries[0].takeoff_airport_id, self.koenigsdorf.id)
|
|
self.assertEqual(entries[0].landing_airport_id, None)
|
|
|
|
def test_single_landing(self):
|
|
db.session.add(self.landing_koenigsdorf_dd0815)
|
|
db.session.commit()
|
|
|
|
update_logbook()
|
|
entries = self.get_logbook_entries()
|
|
self.assertEqual(len(entries), 1)
|
|
self.assertEqual(entries[0].takeoff_airport_id, None)
|
|
self.assertEqual(entries[0].landing_airport_id, self.koenigsdorf.id)
|
|
|
|
def test_different_takeoffs(self):
|
|
db.session.add(self.takeoff_koenigsdorf_dd0815)
|
|
db.session.add(self.takeoff_ohlstadt_dd4711)
|
|
db.session.commit()
|
|
|
|
update_logbook()
|
|
entries = self.get_logbook_entries()
|
|
self.assertEqual(len(entries), 2)
|
|
self.assertEqual(entries[0].takeoff_airport_id, self.koenigsdorf.id)
|
|
self.assertEqual(entries[1].takeoff_airport_id, self.ohlstadt.id)
|
|
|
|
def test_takeoff_and_landing(self):
|
|
db.session.add(self.takeoff_koenigsdorf_dd0815)
|
|
db.session.add(self.landing_koenigsdorf_dd0815)
|
|
db.session.commit()
|
|
|
|
update_logbook()
|
|
entries = self.get_logbook_entries()
|
|
self.assertEqual(len(entries), 1)
|
|
self.assertEqual(entries[0].takeoff_airport_id, self.koenigsdorf.id)
|
|
self.assertEqual(entries[0].landing_airport_id, self.koenigsdorf.id)
|
|
|
|
@unittest.skip('needs information about airport timezone')
|
|
def test_takeoff_and_landing_on_different_days(self):
|
|
db.session.add(self.takeoff_koenigsdorf_dd0815)
|
|
db.session.add(self.landing_koenigsdorf_dd0815_later)
|
|
db.session.commit()
|
|
|
|
update_logbook()
|
|
entries = self.get_logbook_entries()
|
|
self.assertEqual(len(entries), 2)
|
|
self.assertEqual(entries[0].takeoff_airport_id, self.koenigsdorf.id)
|
|
self.assertEqual(entries[0].reftime, self.takeoff_koenigsdorf_dd0815.timestamp)
|
|
self.assertEqual(entries[1].landing_airport_id, self.koenigsdorf.id)
|
|
self.assertEqual(entries[1].reftime, self.landing_koenigsdorf_dd0815_later.timestamp)
|
|
|
|
def test_update(self):
|
|
db.session.add(self.takeoff_koenigsdorf_dd0815)
|
|
db.session.commit()
|
|
|
|
update_logbook(0)
|
|
entries = self.get_logbook_entries()
|
|
self.assertEqual(len(entries), 1)
|
|
self.assertEqual(entries[0].takeoff_airport_id, self.koenigsdorf.id)
|
|
|
|
db.session.add(self.landing_koenigsdorf_dd0815)
|
|
db.session.commit()
|
|
|
|
update_logbook()
|
|
entries = self.get_logbook_entries()
|
|
self.assertEqual(len(entries), 1)
|
|
self.assertEqual(entries[0].takeoff_airport_id, self.koenigsdorf.id)
|
|
self.assertEqual(entries[0].landing_airport_id, self.koenigsdorf.id)
|
|
|
|
db.session.add(self.takeoff_ohlstadt_dd4711)
|
|
db.session.commit()
|
|
|
|
update_logbook(0)
|
|
entries = self.get_logbook_entries()
|
|
self.assertEqual(len(entries), 2)
|
|
self.assertEqual(entries[0].takeoff_airport_id, self.koenigsdorf.id)
|
|
self.assertEqual(entries[1].takeoff_airport_id, self.ohlstadt.id)
|
|
|
|
def test_update_wrong_order(self):
|
|
db.session.add(self.landing_koenigsdorf_dd0815)
|
|
db.session.commit()
|
|
|
|
update_logbook()
|
|
entries = self.get_logbook_entries()
|
|
self.assertEqual(len(entries), 1)
|
|
self.assertEqual(entries[0].takeoff_airport_id, None)
|
|
self.assertEqual(entries[0].landing_airport_id, self.koenigsdorf.id)
|
|
self.assertEqual(entries[0].reference_timestamp, self.landing_koenigsdorf_dd0815.timestamp)
|
|
|
|
db.session.add(self.takeoff_koenigsdorf_dd0815)
|
|
db.session.commit()
|
|
|
|
update_logbook()
|
|
entries = self.get_logbook_entries()
|
|
self.assertEqual(len(entries), 1)
|
|
self.assertEqual(entries[0].takeoff_airport_id, self.koenigsdorf.id)
|
|
self.assertEqual(entries[0].landing_airport_id, self.koenigsdorf.id)
|
|
self.assertEqual(entries[0].reference_timestamp, self.takeoff_koenigsdorf_dd0815.timestamp)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|