ogn-python/tests/model/test_device.py

import datetime

import unittest

from tests.base import TestBaseDB, db
from app.model import Device, DeviceInfo
from app.model.device_info_origin import DeviceInfoOrigin


class TestStringMethods(TestBaseDB):
    def test_device_info(self):
        device = Device(name="FLRDD0815", address="DD0815")
        device_info1 = DeviceInfo(address="DD0815", address_origin=DeviceInfoOrigin.OGN_DDB, registration="D-0815")
        device_info2 = DeviceInfo(address="DD0815", address_origin=DeviceInfoOrigin.FLARMNET, registration="15")

        db.session.add(device)
        db.session.add(device_info1)
        db.session.add(device_info2)
        db.session.commit()

        self.assertEqual(device.info, device_info1)

    def test_expiry_date(self):
        device = Device(name="FLRDD0815", address="DD0815", software_version=6.42)

        self.assertEqual(device.expiry_date(), datetime.date(2019, 10, 31))


if __name__ == "__main__":
    unittest.main()