@@ -1420,6 +1420,19 @@ def setUpTestData(cls):
1420
1420
)
1421
1421
Device .objects .bulk_create (devices )
1422
1422
1423
+ interface = Interface .objects .create (
1424
+ device = devices [0 ],
1425
+ name = 'eth0' ,
1426
+ type = InterfaceTypeChoices .TYPE_VIRTUAL
1427
+ )
1428
+ interface_ct = ContentType .objects .get_for_model (Interface ).pk
1429
+ ip_addresses = (
1430
+ IPAddress (address = '192.0.2.1/24' , assigned_object_type_id = interface_ct , assigned_object_id = interface .pk ),
1431
+ IPAddress (address = '192.0.2.2/24' , assigned_object_type_id = interface_ct , assigned_object_id = interface .pk ),
1432
+ IPAddress (address = '192.0.2.3/24' , assigned_object_type_id = interface_ct , assigned_object_id = interface .pk ),
1433
+ )
1434
+ IPAddress .objects .bulk_create (ip_addresses )
1435
+
1423
1436
clustertype = ClusterType .objects .create (name = 'Cluster Type 1' , slug = 'cluster-type-1' )
1424
1437
cluster = Cluster .objects .create (type = clustertype , name = 'Cluster 1' )
1425
1438
@@ -1439,6 +1452,9 @@ def setUpTestData(cls):
1439
1452
Service (virtual_machine = virtual_machines [2 ], name = 'Service 6' , protocol = ServiceProtocolChoices .PROTOCOL_UDP , ports = [2003 ]),
1440
1453
)
1441
1454
Service .objects .bulk_create (services )
1455
+ services [0 ].ipaddresses .add (ip_addresses [0 ])
1456
+ services [1 ].ipaddresses .add (ip_addresses [1 ])
1457
+ services [2 ].ipaddresses .add (ip_addresses [2 ])
1442
1458
1443
1459
def test_name (self ):
1444
1460
params = {'name' : ['Service 1' , 'Service 2' ]}
@@ -1470,6 +1486,13 @@ def test_virtual_machine(self):
1470
1486
params = {'virtual_machine' : [vms [0 ].name , vms [1 ].name ]}
1471
1487
self .assertEqual (self .filterset (params , self .queryset ).qs .count (), 2 )
1472
1488
1489
+ def test_ipaddress (self ):
1490
+ ips = IPAddress .objects .all ()[:2 ]
1491
+ params = {'ipaddress_id' : [ips [0 ].pk , ips [1 ].pk ]}
1492
+ self .assertEqual (self .filterset (params , self .queryset ).qs .count (), 2 )
1493
+ params = {'ipaddress' : [str (ips [0 ].address ), str (ips [1 ].address )]}
1494
+ self .assertEqual (self .filterset (params , self .queryset ).qs .count (), 2 )
1495
+
1473
1496
1474
1497
class L2VPNTestCase (TestCase , ChangeLoggedFilterSetTests ):
1475
1498
queryset = L2VPN .objects .all ()
0 commit comments