@@ -6323,11 +6323,12 @@ def test_get_segment_ids_for_network_passes(self, mock_client):
6323
6323
res = self .api .get_segment_ids_for_network (
6324
6324
self .context , uuids .network_id )
6325
6325
self .assertEqual ([uuids .segment_id ], res )
6326
+ mock_client .assert_called_once_with (self .context , admin = True )
6326
6327
mocked_client .list_subnets .assert_called_once_with (
6327
6328
network_id = uuids .network_id , fields = 'segment_id' )
6328
6329
6329
6330
@mock .patch .object (neutronapi , 'get_client' )
6330
- def test_get_segment_ids_for_network_with_no_segments (self , mock_client ):
6331
+ def test_get_segment_ids_for_network_with_segments_none (self , mock_client ):
6331
6332
subnets = {'subnets' : [{'segment_id' : None }]}
6332
6333
mocked_client = mock .create_autospec (client .Client )
6333
6334
mock_client .return_value = mocked_client
@@ -6338,6 +6339,23 @@ def test_get_segment_ids_for_network_with_no_segments(self, mock_client):
6338
6339
res = self .api .get_segment_ids_for_network (
6339
6340
self .context , uuids .network_id )
6340
6341
self .assertEqual ([], res )
6342
+ mock_client .assert_called_once_with (self .context , admin = True )
6343
+ mocked_client .list_subnets .assert_called_once_with (
6344
+ network_id = uuids .network_id , fields = 'segment_id' )
6345
+
6346
+ @mock .patch .object (neutronapi , 'get_client' )
6347
+ def test_get_segment_ids_for_network_with_no_segments (self , mock_client ):
6348
+ subnets = {'subnets' : [{}]}
6349
+ mocked_client = mock .create_autospec (client .Client )
6350
+ mock_client .return_value = mocked_client
6351
+ mocked_client .list_subnets .return_value = subnets
6352
+ with mock .patch .object (
6353
+ self .api , '_has_segment_extension' , return_value = True ,
6354
+ ):
6355
+ res = self .api .get_segment_ids_for_network (
6356
+ self .context , uuids .network_id )
6357
+ self .assertEqual ([], res )
6358
+ mock_client .assert_called_once_with (self .context , admin = True )
6341
6359
mocked_client .list_subnets .assert_called_once_with (
6342
6360
network_id = uuids .network_id , fields = 'segment_id' )
6343
6361
@@ -6353,6 +6371,7 @@ def test_get_segment_ids_for_network_fails(self, mock_client):
6353
6371
self .assertRaises (exception .InvalidRoutedNetworkConfiguration ,
6354
6372
self .api .get_segment_ids_for_network ,
6355
6373
self .context , uuids .network_id )
6374
+ mock_client .assert_called_once_with (self .context , admin = True )
6356
6375
6357
6376
def test_get_segment_id_for_subnet_no_segment_ext (self ):
6358
6377
with mock .patch .object (
@@ -6374,6 +6393,7 @@ def test_get_segment_id_for_subnet_passes(self, mock_client):
6374
6393
res = self .api .get_segment_id_for_subnet (
6375
6394
self .context , uuids .subnet_id )
6376
6395
self .assertEqual (uuids .segment_id , res )
6396
+ mock_client .assert_called_once_with (self .context , admin = True )
6377
6397
mocked_client .show_subnet .assert_called_once_with (uuids .subnet_id )
6378
6398
6379
6399
@mock .patch .object (neutronapi , 'get_client' )
@@ -6388,6 +6408,7 @@ def test_get_segment_id_for_subnet_with_no_segment(self, mock_client):
6388
6408
self .assertIsNone (
6389
6409
self .api .get_segment_id_for_subnet (self .context ,
6390
6410
uuids .subnet_id ))
6411
+ mock_client .assert_called_once_with (self .context , admin = True )
6391
6412
6392
6413
@mock .patch .object (neutronapi , 'get_client' )
6393
6414
def test_get_segment_id_for_subnet_fails (self , mock_client ):
@@ -6401,6 +6422,7 @@ def test_get_segment_id_for_subnet_fails(self, mock_client):
6401
6422
self .assertRaises (exception .InvalidRoutedNetworkConfiguration ,
6402
6423
self .api .get_segment_id_for_subnet ,
6403
6424
self .context , uuids .subnet_id )
6425
+ mock_client .assert_called_once_with (self .context , admin = True )
6404
6426
6405
6427
@mock .patch .object (neutronapi .LOG , 'debug' )
6406
6428
def test_get_port_pci_slot (self , mock_debug ):
0 commit comments