@@ -37,6 +37,48 @@ def setUp(self):
3737 self .domains_mock = self .app .client_manager .identity .domains
3838
3939
40+ class TestAddNetworkFlavorToProfile (TestNetworkFlavor ):
41+
42+ network_flavor = \
43+ network_fakes .FakeNetworkFlavor .create_one_network_flavor ()
44+ service_profile = \
45+ network_fakes .FakeNetworkFlavorProfile .create_one_service_profile ()
46+
47+ def setUp (self ):
48+ super (TestAddNetworkFlavorToProfile , self ).setUp ()
49+ self .network .find_flavor = mock .Mock (return_value = self .network_flavor )
50+ self .network .find_service_profile = mock .Mock (
51+ return_value = self .service_profile )
52+ self .network .associate_flavor_with_service_profile = mock .Mock ()
53+
54+ self .cmd = network_flavor .AddNetworkFlavorToProfile (
55+ self .app , self .namespace )
56+
57+ def test_show_no_options (self ):
58+ arglist = []
59+ verifylist = []
60+
61+ # Missing required args should bail here
62+ self .assertRaises (tests_utils .ParserException , self .check_parser ,
63+ self .cmd , arglist , verifylist )
64+
65+ def test_add_flavor_to_service_profile (self ):
66+ arglist = [
67+ self .network_flavor .id ,
68+ self .service_profile .id
69+ ]
70+ verifylist = [
71+ ('flavor' , self .network_flavor .id ),
72+ ('service_profile' , self .service_profile .id ),
73+ ]
74+
75+ parsed_args = self .check_parser (self .cmd , arglist , verifylist )
76+ self .cmd .take_action (parsed_args )
77+
78+ self .network .associate_flavor_with_service_profile .\
79+ assert_called_once_with (self .network_flavor , self .service_profile )
80+
81+
4082class TestCreateNetworkFlavor (TestNetworkFlavor ):
4183
4284 project = identity_fakes_v3 .FakeProject .create_one_project ()
@@ -281,6 +323,48 @@ def test_network_flavor_list(self):
281323 self .assertEqual (self .data , list (data ))
282324
283325
326+ class TestRemoveNetworkFlavorFromProfile (TestNetworkFlavor ):
327+
328+ network_flavor = \
329+ network_fakes .FakeNetworkFlavor .create_one_network_flavor ()
330+ service_profile = \
331+ network_fakes .FakeNetworkFlavorProfile .create_one_service_profile ()
332+
333+ def setUp (self ):
334+ super (TestRemoveNetworkFlavorFromProfile , self ).setUp ()
335+ self .network .find_flavor = mock .Mock (return_value = self .network_flavor )
336+ self .network .find_service_profile = mock .Mock (
337+ return_value = self .service_profile )
338+ self .network .disassociate_flavor_from_service_profile = mock .Mock ()
339+
340+ self .cmd = network_flavor .RemoveNetworkFlavorFromProfile (
341+ self .app , self .namespace )
342+
343+ def test_show_no_options (self ):
344+ arglist = []
345+ verifylist = []
346+
347+ # Missing required args should bail here
348+ self .assertRaises (tests_utils .ParserException , self .check_parser ,
349+ self .cmd , arglist , verifylist )
350+
351+ def test_remove_flavor_from_service_profile (self ):
352+ arglist = [
353+ self .network_flavor .id ,
354+ self .service_profile .id
355+ ]
356+ verifylist = [
357+ ('flavor' , self .network_flavor .id ),
358+ ('service_profile' , self .service_profile .id ),
359+ ]
360+
361+ parsed_args = self .check_parser (self .cmd , arglist , verifylist )
362+ self .cmd .take_action (parsed_args )
363+
364+ self .network .disassociate_flavor_from_service_profile .\
365+ assert_called_once_with (self .network_flavor , self .service_profile )
366+
367+
284368class TestShowNetworkFlavor (TestNetworkFlavor ):
285369
286370 # The network flavor to show.
0 commit comments