|
|
@@ -648,7 +648,6 @@ class CircuitGroupAssignmentTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
CircuitGroup(name='Circuit Group 1', slug='circuit-group-1'),
|
|
|
CircuitGroup(name='Circuit Group 2', slug='circuit-group-2'),
|
|
|
CircuitGroup(name='Circuit Group 3', slug='circuit-group-3'),
|
|
|
- CircuitGroup(name='Circuit Group 4', slug='circuit-group-4'),
|
|
|
)
|
|
|
CircuitGroup.objects.bulk_create(circuit_groups)
|
|
|
|
|
|
@@ -656,7 +655,6 @@ class CircuitGroupAssignmentTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Provider(name='Provider 1', slug='provider-1'),
|
|
|
Provider(name='Provider 2', slug='provider-2'),
|
|
|
Provider(name='Provider 3', slug='provider-3'),
|
|
|
- Provider(name='Provider 4', slug='provider-4'),
|
|
|
))
|
|
|
circuittype = CircuitType.objects.create(name='Circuit Type 1', slug='circuit-type-1')
|
|
|
|
|
|
@@ -664,35 +662,72 @@ class CircuitGroupAssignmentTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
Circuit(cid='Circuit 1', provider=providers[0], type=circuittype),
|
|
|
Circuit(cid='Circuit 2', provider=providers[1], type=circuittype),
|
|
|
Circuit(cid='Circuit 3', provider=providers[2], type=circuittype),
|
|
|
- Circuit(cid='Circuit 4', provider=providers[3], type=circuittype),
|
|
|
)
|
|
|
Circuit.objects.bulk_create(circuits)
|
|
|
|
|
|
+ provider_networks = (
|
|
|
+ ProviderNetwork(name='Provider Network 1', provider=providers[0]),
|
|
|
+ ProviderNetwork(name='Provider Network 2', provider=providers[1]),
|
|
|
+ ProviderNetwork(name='Provider Network 3', provider=providers[2]),
|
|
|
+ )
|
|
|
+ ProviderNetwork.objects.bulk_create(provider_networks)
|
|
|
+
|
|
|
+ virtual_circuits = (
|
|
|
+ VirtualCircuit(
|
|
|
+ provider_network=provider_networks[0],
|
|
|
+ cid='Virtual Circuit 1'
|
|
|
+ ),
|
|
|
+ VirtualCircuit(
|
|
|
+ provider_network=provider_networks[1],
|
|
|
+ cid='Virtual Circuit 2'
|
|
|
+ ),
|
|
|
+ VirtualCircuit(
|
|
|
+ provider_network=provider_networks[2],
|
|
|
+ cid='Virtual Circuit 3'
|
|
|
+ ),
|
|
|
+ )
|
|
|
+ VirtualCircuit.objects.bulk_create(virtual_circuits)
|
|
|
+
|
|
|
assignments = (
|
|
|
CircuitGroupAssignment(
|
|
|
group=circuit_groups[0],
|
|
|
- circuit=circuits[0],
|
|
|
+ member=circuits[0],
|
|
|
priority=CircuitPriorityChoices.PRIORITY_PRIMARY
|
|
|
),
|
|
|
CircuitGroupAssignment(
|
|
|
group=circuit_groups[1],
|
|
|
- circuit=circuits[1],
|
|
|
+ member=circuits[1],
|
|
|
priority=CircuitPriorityChoices.PRIORITY_SECONDARY
|
|
|
),
|
|
|
CircuitGroupAssignment(
|
|
|
group=circuit_groups[2],
|
|
|
- circuit=circuits[2],
|
|
|
+ member=circuits[2],
|
|
|
+ priority=CircuitPriorityChoices.PRIORITY_TERTIARY
|
|
|
+ ),
|
|
|
+ CircuitGroupAssignment(
|
|
|
+ group=circuit_groups[0],
|
|
|
+ member=virtual_circuits[0],
|
|
|
+ priority=CircuitPriorityChoices.PRIORITY_PRIMARY
|
|
|
+ ),
|
|
|
+ CircuitGroupAssignment(
|
|
|
+ group=circuit_groups[1],
|
|
|
+ member=virtual_circuits[1],
|
|
|
+ priority=CircuitPriorityChoices.PRIORITY_SECONDARY
|
|
|
+ ),
|
|
|
+ CircuitGroupAssignment(
|
|
|
+ group=circuit_groups[2],
|
|
|
+ member=virtual_circuits[2],
|
|
|
priority=CircuitPriorityChoices.PRIORITY_TERTIARY
|
|
|
),
|
|
|
)
|
|
|
CircuitGroupAssignment.objects.bulk_create(assignments)
|
|
|
|
|
|
- def test_group_id(self):
|
|
|
- groups = CircuitGroup.objects.filter(name__in=['Circuit Group 1', 'Circuit Group 2'])
|
|
|
+ def test_group(self):
|
|
|
+ groups = CircuitGroup.objects.all()[:2]
|
|
|
params = {'group_id': [groups[0].pk, groups[1].pk]}
|
|
|
- self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
params = {'group': [groups[0].slug, groups[1].slug]}
|
|
|
- self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
|
|
|
def test_circuit(self):
|
|
|
circuits = Circuit.objects.all()[:2]
|
|
|
@@ -701,12 +736,19 @@ class CircuitGroupAssignmentTestCase(TestCase, ChangeLoggedFilterSetTests):
|
|
|
params = {'circuit': [circuits[0].cid, circuits[1].cid]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
+ def test_virtual_circuit(self):
|
|
|
+ virtual_circuits = VirtualCircuit.objects.all()[:2]
|
|
|
+ params = {'virtual_circuit_id': [virtual_circuits[0].pk, virtual_circuits[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ params = {'virtual_circuit': [virtual_circuits[0].cid, virtual_circuits[1].cid]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+
|
|
|
def test_provider(self):
|
|
|
providers = Provider.objects.all()[:2]
|
|
|
params = {'provider_id': [providers[0].pk, providers[1].pk]}
|
|
|
- self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
params = {'provider': [providers[0].slug, providers[1].slug]}
|
|
|
- self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
|
|
|
|
|
|
class ProviderNetworkTestCase(TestCase, ChangeLoggedFilterSetTests):
|