kopia lustrzana https://github.com/glidernet/ogn-python
30 wiersze
862 B
Python
30 wiersze
862 B
Python
import datetime
|
|
|
|
import unittest
|
|
|
|
from tests.base import TestBaseDB, db
|
|
from ogn_python.model import Device, DeviceInfo
|
|
|
|
|
|
class TestStringMethods(TestBaseDB):
|
|
def test_device_info(self):
|
|
device = Device(name='FLRDD0815', address='DD0815')
|
|
device_info1 = DeviceInfo(address='DD0815', address_origin=1, registration='D-0815')
|
|
device_info2 = DeviceInfo(address='DD0815', address_origin=2, 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()
|