|
|
@@ -392,13 +392,12 @@ class IPAddressTestCase(TestCase):
|
|
|
params = {'vrf': [vrfs[0].rd, vrfs[1].rd]}
|
|
|
self.assertEqual(self.filterset(params, self.queryset).qs.count(), 4)
|
|
|
|
|
|
- # TODO: Test for multiple values
|
|
|
def test_device(self):
|
|
|
- device = Device.objects.first()
|
|
|
- params = {'device_id': device.pk}
|
|
|
- self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
|
|
|
- params = {'device': device.name}
|
|
|
- self.assertEqual(self.filterset(params, self.queryset).qs.count(), 1)
|
|
|
+ devices = Device.objects.all()[:2]
|
|
|
+ params = {'device_id': [devices[0].pk, devices[1].pk]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
+ params = {'device': [devices[0].name, devices[1].name]}
|
|
|
+ self.assertEqual(self.filterset(params, self.queryset).qs.count(), 2)
|
|
|
|
|
|
def test_virtual_machine(self):
|
|
|
vms = VirtualMachine.objects.all()[:2]
|