kopia lustrzana https://github.com/glidernet/ogn-python
31 wiersze
960 B
Python
31 wiersze
960 B
Python
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()
|