Source code for apps.core.tests.test_device_type
from django.test import TestCase
from ..models.device_type import DeviceType
[docs]class DeviceTypeTest(TestCase):
[docs] def setUp(self):
DeviceType.objects.create(
type_name="type1", default_version="2020", description="test description 1"
)
DeviceType.objects.create(
type_name="type2", default_version="2019", description="test description 2"
)
[docs] def test_device_type_name(self):
type1 = DeviceType.objects.get(type_name="type1")
type2 = DeviceType.objects.get(type_name="type2")
self.assertEqual(
type1.type_name, "type1"
)
self.assertEqual(
type2.type_name, "type2"
)
[docs] def test_device_type_description(self):
type1 = DeviceType.objects.get(type_name="type1")
type2 = DeviceType.objects.get(type_name="type2")
self.assertEqual(
type1.description, "test description 1"
)
self.assertEqual(
type2.description, "test description 2"
)