kopia lustrzana https://github.com/glidernet/ogn-python
54 wiersze
1.6 KiB
Python
54 wiersze
1.6 KiB
Python
import unittest
|
|
import os
|
|
|
|
from ogn.commands.bulkimport import convert_logfile, import_csv_logfile
|
|
from ogn.model import AircraftBeacon, ReceiverBeacon
|
|
|
|
|
|
class TestDB(unittest.TestCase):
|
|
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()
|
|
|
|
def tearDown(self):
|
|
session = self.session
|
|
session.execute("DELETE FROM aircraft_beacon")
|
|
session.execute("DELETE FROM receiver_beacon")
|
|
session.commit()
|
|
|
|
def test_convert_logfile(self):
|
|
path = os.path.dirname(__file__)
|
|
|
|
convert_logfile(path=path)
|
|
|
|
os.remove(os.path.join(path, 'aircraft_beacons.csv_2016-09-21.gz'))
|
|
os.remove(os.path.join(path, 'receiver_beacons.csv_2016-09-21.gz'))
|
|
|
|
def test_import_csv_logfile(self):
|
|
session = self.session
|
|
|
|
path = os.path.dirname(__file__)
|
|
|
|
convert_logfile(path=path)
|
|
|
|
import_csv_logfile(path, 'aircraft_beacons.csv_2016-09-21.gz')
|
|
import_csv_logfile(path, 'receiver_beacons.csv_2016-09-21.gz')
|
|
|
|
os.remove(os.path.join(path, 'aircraft_beacons.csv_2016-09-21.gz'))
|
|
os.remove(os.path.join(path, 'receiver_beacons.csv_2016-09-21.gz'))
|
|
|
|
aircraft_beacons = session.query(AircraftBeacon).all()
|
|
receiver_beacons = session.query(ReceiverBeacon).all()
|
|
|
|
self.assertGreater(len(aircraft_beacons), 1)
|
|
self.assertGreater(len(receiver_beacons), 1)
|