@@ -2048,6 +2048,11 @@ def setUpTestData(cls):
20482048 )
20492049 Device .objects .bulk_create (devices )
20502050
2051+ # VirtualChassis assignment for filtering
2052+ virtual_chassis = VirtualChassis .objects .create (master = devices [0 ])
2053+ Device .objects .filter (pk = devices [0 ].pk ).update (virtual_chassis = virtual_chassis , vc_position = 1 , vc_priority = 1 )
2054+ Device .objects .filter (pk = devices [1 ].pk ).update (virtual_chassis = virtual_chassis , vc_position = 2 , vc_priority = 2 )
2055+
20512056 interfaces = (
20522057 Interface (device = devices [0 ], name = 'Interface 1' , label = 'A' , type = InterfaceTypeChoices .TYPE_1GE_SFP , enabled = True , mgmt_only = True , mtu = 100 , mode = InterfaceModeChoices .MODE_ACCESS , mac_address = '00-00-00-00-00-01' , description = 'First' ),
20532058 Interface (device = devices [1 ], name = 'Interface 2' , label = 'B' , type = InterfaceTypeChoices .TYPE_1GE_GBIC , enabled = True , mgmt_only = True , mtu = 200 , mode = InterfaceModeChoices .MODE_TAGGED , mac_address = '00-00-00-00-00-02' , description = 'Second' ),
@@ -2157,6 +2162,10 @@ def test_location(self):
21572162 params = {'location' : [locations [0 ].slug , locations [1 ].slug ]}
21582163 self .assertEqual (self .filterset (params , self .queryset ).qs .count (), 2 )
21592164
2165+ def test_virtual_chassis_id (self ):
2166+ params = {'virtual_chassis_id' : [VirtualChassis .objects .first ().pk ]}
2167+ self .assertEqual (self .filterset (params , self .queryset ).qs .count (), 2 )
2168+
21602169 def test_device (self ):
21612170 devices = Device .objects .all ()[:2 ]
21622171 params = {'device_id' : [devices [0 ].pk , devices [1 ].pk ]}
0 commit comments