ogn-python/tests/base.py

38 wiersze
861 B
Python

import unittest
import os
os.environ["OGN_CONFIG_MODULE"] = "config/test.py"
from app import db # noqa: E402
class TestBaseDB(unittest.TestCase):
@classmethod
def setUpClass(cls):
db.session.execute("CREATE EXTENSION IF NOT EXISTS postgis;")
db.session.commit()
db.drop_all()
db.create_all()
def setUp(self):
pass
def tearDown(self):
db.session.execute(
"""
DELETE FROM aircraft_beacons;
DELETE FROM receiver_beacons;
DELETE FROM takeoff_landings;
DELETE FROM logbook;
DELETE FROM receiver_coverages;
DELETE FROM device_stats;
DELETE FROM receiver_stats;
DELETE FROM receivers;
DELETE FROM devices;
"""
)
if __name__ == "__main__":
unittest.main()