ogn-python/tests/backend/test_backends.py

import unittest
from unittest import mock
import os
from datetime import datetime

from xmlunittest import XmlTestMixin

from ogn.model import Receiver, Device

from ogn.backend.liveglidernet import rec, lxml
from ogn.backend.ognrange import stations2_filtered_pl


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')

        self.d01 = Device(address='DD4711')
        self.d02 = Device(address='DD0815')

        session.add(self.r01)
        session.add(self.r02)
        session.add(self.r03)

        session.add(self.d01)
        session.add(self.d02)
        session.commit()

    def tearDown(self):
        session = self.session
        session.execute("DELETE FROM receiver")
        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"]'))

        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')

        # Check the complete document
        self.assertXmlEquivalentOutputs(data, expected)

        print(data)

    @unittest.skip("not finished yet")
    @mock.patch('ogn.backend.liveglidernet.datetime')
    def test_lxml(self, datetime_mock):
        session = self.session

        datetime_mock.utcnow.return_value = datetime(2017, 12, 20, 10, 0)

        data = lxml(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"]'))

        print(data)

    @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)
        print(result)

        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()