def setUp(self, mock_eos_device, mock_native_node, mock_native_vlans): mock_eos_instance = mock_eos_device.return_value mock_eos_instance.native = mock_native_node mock_native_vlans.get.side_effect = get mock_native_vlans.getall.side_effect = getall self.vlans = EOSVlans(mock_eos_instance) self.vlans.native_vlans = mock_native_vlans