|
@@ -1,3 +1,4 @@
|
|
|
|
|
+from django.conf import settings
|
|
|
from django.test import TestCase
|
|
from django.test import TestCase
|
|
|
|
|
|
|
|
from circuits.models import Circuit, CircuitTermination, CircuitType, Provider
|
|
from circuits.models import Circuit, CircuitTermination, CircuitType, Provider
|
|
@@ -3085,6 +3086,11 @@ class ModuleTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Manufacturer(name='Manufacturer 3', slug='manufacturer-3'),
|
|
Manufacturer(name='Manufacturer 3', slug='manufacturer-3'),
|
|
|
)
|
|
)
|
|
|
Manufacturer.objects.bulk_create(manufacturers)
|
|
Manufacturer.objects.bulk_create(manufacturers)
|
|
|
|
|
+ module_type_profiles = (
|
|
|
|
|
+ ModuleTypeProfile(name='Test CPU'),
|
|
|
|
|
+ ModuleTypeProfile(name='Test Hard disk'),
|
|
|
|
|
+ )
|
|
|
|
|
+ ModuleTypeProfile.objects.bulk_create(module_type_profiles)
|
|
|
|
|
|
|
|
device_types = (
|
|
device_types = (
|
|
|
DeviceType(manufacturer=manufacturers[0], model='Device Type 1', slug='device-type-1'),
|
|
DeviceType(manufacturer=manufacturers[0], model='Device Type 1', slug='device-type-1'),
|
|
@@ -3148,8 +3154,8 @@ class ModuleTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Device.objects.bulk_create(devices)
|
|
Device.objects.bulk_create(devices)
|
|
|
|
|
|
|
|
module_types = (
|
|
module_types = (
|
|
|
- ModuleType(manufacturer=manufacturers[0], model='Module Type 1'),
|
|
|
|
|
- ModuleType(manufacturer=manufacturers[1], model='Module Type 2'),
|
|
|
|
|
|
|
+ ModuleType(manufacturer=manufacturers[0], model='Module Type 1', profile=module_type_profiles[0]),
|
|
|
|
|
+ ModuleType(manufacturer=manufacturers[1], model='Module Type 2', profile=module_type_profiles[1]),
|
|
|
ModuleType(manufacturer=manufacturers[2], model='Module Type 3'),
|
|
ModuleType(manufacturer=manufacturers[2], model='Module Type 3'),
|
|
|
)
|
|
)
|
|
|
ModuleType.objects.bulk_create(module_types)
|
|
ModuleType.objects.bulk_create(module_types)
|
|
@@ -3265,6 +3271,19 @@ class ModuleTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
params = {'module_type': [module_types[0].model, module_types[1].model]}
|
|
params = {'module_type': [module_types[0].model, module_types[1].model]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 6)
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 6)
|
|
|
|
|
|
|
|
|
|
+ def test_profile(self):
|
|
|
|
|
+ profiles = ModuleTypeProfile.objects.filter(name__startswith='Test').order_by('name')
|
|
|
|
|
+ params = {'profile_id': [profiles[0].pk]}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 3)
|
|
|
|
|
+ params = {'profile': [profiles[0].name]}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 3)
|
|
|
|
|
+ params = {'profile_id': [profiles[1].pk]}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 3)
|
|
|
|
|
+ params = {'profile': [profiles[1].name]}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 3)
|
|
|
|
|
+ params = {'profile_id': [settings.FILTERS_NULL_CHOICE_VALUE]}
|
|
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 3)
|
|
|
|
|
+
|
|
|
def test_description(self):
|
|
def test_description(self):
|
|
|
params = {'description': ['foobar1', 'foobar2']}
|
|
params = {'description': ['foobar1', 'foobar2']}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|