@@ -2016,17 +2016,12 @@ def test_plug_vifs_with_port(self, mock_vatt):
20162016 @mock .patch .object (ironic_driver .IronicDriver , '_plug_vifs' )
20172017 def test_plug_vifs (self , mock__plug_vifs ):
20182018 node_id = 'aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee'
2019- node = _get_cached_node (id = node_id )
2020-
2021- self .mock_conn .get_node .return_value = node
20222019 instance = fake_instance .fake_instance_obj (self .ctx ,
20232020 node = node_id )
20242021 network_info = utils .get_test_network_info ()
20252022 self .driver .plug_vifs (instance , network_info )
20262023
2027- self .mock_conn .get_node .assert_called_once_with (
2028- node_id , fields = ironic_driver ._NODE_FIELDS )
2029- mock__plug_vifs .assert_called_once_with (node , instance , network_info )
2024+ mock__plug_vifs .assert_not_called ()
20302025
20312026 @mock .patch .object (FAKE_CLIENT .node , 'vif_attach' )
20322027 def test_plug_vifs_multiple_ports (self , mock_vatt ):
@@ -2160,11 +2155,17 @@ def test_unplug_vifs_no_network_info(self, mock_vdet):
21602155 self .driver .unplug_vifs (instance , network_info )
21612156 self .assertFalse (mock_vdet .called )
21622157
2163- @mock .patch .object (ironic_driver .IronicDriver , 'plug_vifs ' )
2158+ @mock .patch .object (ironic_driver .IronicDriver , '_plug_vifs ' )
21642159 def test_attach_interface (self , mock_pv ):
2165- self .driver .attach_interface ('fake_context' , 'fake_instance' ,
2160+ node_uuid = 'aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee'
2161+ node = _get_cached_node (uuid = node_uuid )
2162+ instance = fake_instance .fake_instance_obj (self .ctx ,
2163+ node = node_uuid )
2164+ self .mock_conn .get_node .return_value = node
2165+
2166+ self .driver .attach_interface ('fake_context' , instance ,
21662167 'fake_image_meta' , 'fake_vif' )
2167- mock_pv .assert_called_once_with ('fake_instance' , ['fake_vif' ])
2168+ mock_pv .assert_called_once_with (node , instance , ['fake_vif' ])
21682169
21692170 @mock .patch .object (ironic_driver .IronicDriver , 'unplug_vifs' )
21702171 def test_detach_interface (self , mock_uv ):
@@ -2440,25 +2441,31 @@ def test_get_volume_connector_no_ip_without_mac(self):
24402441 def test_get_volume_connector_no_ip_no_fixed_ip (self ):
24412442 self ._test_get_volume_connector_no_ip (False , no_fixed_ip = True )
24422443
2443- @mock .patch .object (ironic_driver .IronicDriver , 'plug_vifs ' )
2444+ @mock .patch .object (ironic_driver .IronicDriver , '_plug_vifs ' )
24442445 def test_prepare_networks_before_block_device_mapping (self , mock_pvifs ):
2446+ node_uuid = 'aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee'
2447+ node = _get_cached_node (uuid = node_uuid )
2448+ self .mock_conn .get_node .return_value = node
24452449 instance = fake_instance .fake_instance_obj (self .ctx )
24462450 network_info = utils .get_test_network_info ()
24472451 self .driver .prepare_networks_before_block_device_mapping (instance ,
24482452 network_info )
2449- mock_pvifs .assert_called_once_with (instance , network_info )
2453+ mock_pvifs .assert_called_once_with (node , instance , network_info )
24502454
2451- @mock .patch .object (ironic_driver .IronicDriver , 'plug_vifs ' )
2455+ @mock .patch .object (ironic_driver .IronicDriver , '_plug_vifs ' )
24522456 def test_prepare_networks_before_block_device_mapping_error (self ,
24532457 mock_pvifs ):
2458+ node_uuid = 'aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeeeee'
2459+ node = _get_cached_node (uuid = node_uuid )
2460+ self .mock_conn .get_node .return_value = node
24542461 instance = fake_instance .fake_instance_obj (self .ctx )
24552462 network_info = utils .get_test_network_info ()
24562463 mock_pvifs .side_effect = ironic_exception .BadRequest ('fake error' )
24572464 self .assertRaises (
24582465 ironic_exception .BadRequest ,
24592466 self .driver .prepare_networks_before_block_device_mapping ,
24602467 instance , network_info )
2461- mock_pvifs .assert_called_once_with (instance , network_info )
2468+ mock_pvifs .assert_called_once_with (node , instance , network_info )
24622469
24632470 @mock .patch .object (ironic_driver .IronicDriver , 'unplug_vifs' )
24642471 def test_clean_networks_preparation (self , mock_upvifs ):
0 commit comments