@@ -295,19 +295,23 @@ def test_virtual_chassis_scopes_css(self, MockDV, MockCP):
295295 member1 = MagicMock ()
296296 member1 .vc_position = 1
297297 member1 .id = 1
298+ member1 .position = None
299+ member1 .rack = None
298300 member1 .device_type = MagicMock ()
299301 member1 .interfaces .all .return_value = []
300302 member1 .modules .all .return_value = []
301303
302304 member2 = MagicMock ()
303305 member2 .vc_position = 2
304306 member2 .id = 2
307+ member2 .position = None
308+ member2 .rack = None
305309 member2 .device_type = MagicMock ()
306310 member2 .interfaces .all .return_value = []
307311 member2 .modules .all .return_value = []
308312
309313 dev = self ._make_device (vc = MagicMock ())
310- dev .virtual_chassis .members .all .return_value = [member1 , member2 ]
314+ dev .virtual_chassis .members .select_related . return_value . all .return_value = [member1 , member2 ]
311315
312316 dv , modules , ports_chassis , device_view = prepare (dev )
313317
@@ -328,12 +332,14 @@ def test_virtual_chassis_missing_member_device_view_returns_none(
328332 member = MagicMock ()
329333 member .vc_position = 1
330334 member .id = 1
335+ member .position = None
336+ member .rack = None
331337 member .device_type = MagicMock ()
332338 member .interfaces .all .return_value = []
333339 member .modules .all .return_value = []
334340
335341 dev = self ._make_device (vc = MagicMock ())
336- dev .virtual_chassis .members .all .return_value = [member ]
342+ dev .virtual_chassis .members .select_related . return_value . all .return_value = [member ]
337343
338344 dv , modules , ports_chassis , device_view = prepare (dev )
339345
0 commit comments