@@ -7026,13 +7026,17 @@ def test_get_requested_resource_for_instance_with_multiple_ports_extended(
7026
7026
req_lvl_params .same_subtree ,
7027
7027
)
7028
7028
7029
- def test_get_segment_ids_for_network_no_segment_ext (self ):
7029
+ @mock .patch .object (neutronapi , 'get_client' )
7030
+ def test_get_segment_ids_for_network_no_segment_ext (self , mock_client ):
7031
+ mocked_client = mock .create_autospec (client .Client )
7032
+ mock_client .return_value = mocked_client
7030
7033
with mock .patch .object (
7031
7034
self .api , 'has_segment_extension' , return_value = False ,
7032
7035
):
7033
7036
self .assertEqual (
7034
7037
[], self .api .get_segment_ids_for_network (self .context ,
7035
7038
uuids .network_id ))
7039
+ mock_client .assert_called_once_with (self .context , admin = True )
7036
7040
7037
7041
@mock .patch .object (neutronapi , 'get_client' )
7038
7042
def test_get_segment_ids_for_network_passes (self , mock_client ):
@@ -7046,6 +7050,7 @@ def test_get_segment_ids_for_network_passes(self, mock_client):
7046
7050
res = self .api .get_segment_ids_for_network (
7047
7051
self .context , uuids .network_id )
7048
7052
self .assertEqual ([uuids .segment_id ], res )
7053
+ mock_client .assert_called_once_with (self .context , admin = True )
7049
7054
mocked_client .list_subnets .assert_called_once_with (
7050
7055
network_id = uuids .network_id , fields = 'segment_id' )
7051
7056
@@ -7061,6 +7066,7 @@ def test_get_segment_ids_for_network_with_no_segments(self, mock_client):
7061
7066
res = self .api .get_segment_ids_for_network (
7062
7067
self .context , uuids .network_id )
7063
7068
self .assertEqual ([], res )
7069
+ mock_client .assert_called_once_with (self .context , admin = True )
7064
7070
mocked_client .list_subnets .assert_called_once_with (
7065
7071
network_id = uuids .network_id , fields = 'segment_id' )
7066
7072
@@ -7076,14 +7082,19 @@ def test_get_segment_ids_for_network_fails(self, mock_client):
7076
7082
self .assertRaises (exception .InvalidRoutedNetworkConfiguration ,
7077
7083
self .api .get_segment_ids_for_network ,
7078
7084
self .context , uuids .network_id )
7085
+ mock_client .assert_called_once_with (self .context , admin = True )
7079
7086
7080
- def test_get_segment_id_for_subnet_no_segment_ext (self ):
7087
+ @mock .patch .object (neutronapi , 'get_client' )
7088
+ def test_get_segment_id_for_subnet_no_segment_ext (self , mock_client ):
7089
+ mocked_client = mock .create_autospec (client .Client )
7090
+ mock_client .return_value = mocked_client
7081
7091
with mock .patch .object (
7082
7092
self .api , 'has_segment_extension' , return_value = False ,
7083
7093
):
7084
7094
self .assertIsNone (
7085
7095
self .api .get_segment_id_for_subnet (self .context ,
7086
7096
uuids .subnet_id ))
7097
+ mock_client .assert_called_once_with (self .context , admin = True )
7087
7098
7088
7099
@mock .patch .object (neutronapi , 'get_client' )
7089
7100
def test_get_segment_id_for_subnet_passes (self , mock_client ):
@@ -7097,6 +7108,7 @@ def test_get_segment_id_for_subnet_passes(self, mock_client):
7097
7108
res = self .api .get_segment_id_for_subnet (
7098
7109
self .context , uuids .subnet_id )
7099
7110
self .assertEqual (uuids .segment_id , res )
7111
+ mock_client .assert_called_once_with (self .context , admin = True )
7100
7112
mocked_client .show_subnet .assert_called_once_with (uuids .subnet_id )
7101
7113
7102
7114
@mock .patch .object (neutronapi , 'get_client' )
@@ -7111,6 +7123,7 @@ def test_get_segment_id_for_subnet_with_no_segment(self, mock_client):
7111
7123
self .assertIsNone (
7112
7124
self .api .get_segment_id_for_subnet (self .context ,
7113
7125
uuids .subnet_id ))
7126
+ mock_client .assert_called_once_with (self .context , admin = True )
7114
7127
7115
7128
@mock .patch .object (neutronapi , 'get_client' )
7116
7129
def test_get_segment_id_for_subnet_fails (self , mock_client ):
@@ -7124,6 +7137,7 @@ def test_get_segment_id_for_subnet_fails(self, mock_client):
7124
7137
self .assertRaises (exception .InvalidRoutedNetworkConfiguration ,
7125
7138
self .api .get_segment_id_for_subnet ,
7126
7139
self .context , uuids .subnet_id )
7140
+ mock_client .assert_called_once_with (self .context , admin = True )
7127
7141
7128
7142
@mock .patch .object (neutronapi .LOG , 'debug' )
7129
7143
def test_get_port_pci_dev (self , mock_debug ):
0 commit comments