@@ -384,13 +384,15 @@ def test_create_with_tags(self):
384384 def test_create_with_no_tag (self ):
385385 self ._test_create_with_tag (add_tags = False )
386386
387- def test_create_with_flavor_id (self ):
387+ def test_create_with_flavor_id_or_name (self ):
388388 _flavor = network_fakes .create_one_network_flavor ()
389+ self .network_client .find_flavor = mock .Mock (return_value = _flavor )
389390 arglist = [
390391 self .new_router .name ,
391392 '--flavor-id' ,
392393 _flavor .id ,
393394 ]
395+ arglist_with_name = [self .new_router .name , '--flavor-id' , _flavor .name ]
394396 verifylist = [
395397 ('name' , self .new_router .name ),
396398 ('enable' , True ),
@@ -410,6 +412,28 @@ def test_create_with_flavor_id(self):
410412 self .assertEqual (self .columns , columns )
411413 self .assertCountEqual (self .data , data )
412414
415+ self .network_client .create_router .reset_mock ()
416+ verifylist_w_name = [
417+ ('name' , self .new_router .name ),
418+ ('enable' , True ),
419+ ('distributed' , False ),
420+ ('ha' , False ),
421+ ('flavor_id' , _flavor .name ),
422+ ]
423+ parsed_args_w_name = self .check_parser (
424+ self .cmd , arglist_with_name , verifylist_w_name
425+ )
426+ columns , data = self .cmd .take_action (parsed_args_w_name )
427+ self .network_client .create_router .assert_called_once_with (
428+ ** {
429+ 'admin_state_up' : True ,
430+ 'name' : self .new_router .name ,
431+ 'flavor_id' : _flavor .id ,
432+ }
433+ )
434+ self .assertEqual (self .columns , columns )
435+ self .assertCountEqual (self .data , data )
436+
413437 def test_create_with_enable_default_route_bfd (self ):
414438 self .network_client .find_extension = mock .Mock (
415439 return_value = network_fakes .create_one_extension (
0 commit comments