Skip to content

Commit 5b866c5

Browse files
committed
test: fix VC member mocks to match select_related rack call chain
1 parent e36d323 commit 5b866c5

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

tests/test_utils.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)