From 9c4c0829a96d6654151ff7e988e0195362ca3200 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Konstantin=20Gru=CC=88ndger?= Date: Sun, 15 Sep 2019 21:36:06 +0200 Subject: [PATCH] Fixed DeviceInfo test --- tests/model/test_device.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tests/model/test_device.py b/tests/model/test_device.py index 436589f..bbcb71b 100644 --- a/tests/model/test_device.py +++ b/tests/model/test_device.py @@ -4,13 +4,14 @@ 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=1, registration="D-0815") - device_info2 = DeviceInfo(address="DD0815", address_origin=2, registration="15") + 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)