@@ -814,6 +814,53 @@ def test_volume_list_long(self):
814814 self .assertEqual (datalist , tuple (data ))
815815
816816
817+ class TestVolumeSet (TestVolume ):
818+
819+ def setUp (self ):
820+ super (TestVolumeSet , self ).setUp ()
821+
822+ self .new_volume = volume_fakes .FakeVolume .create_one_volume ()
823+ self .volumes_mock .get .return_value = self .new_volume
824+
825+ # Get the command object to test
826+ self .cmd = volume .SetVolume (self .app , None )
827+
828+ def test_volume_set_image_property (self ):
829+ arglist = [
830+ '--image-property' , 'Alpha=a' ,
831+ '--image-property' , 'Beta=b' ,
832+ self .new_volume .id ,
833+ ]
834+ verifylist = [
835+ ('image_property' , {'Alpha' : 'a' , 'Beta' : 'b' }),
836+ ('volume' , self .new_volume .id ),
837+ ]
838+ parsed_args = self .check_parser (self .cmd , arglist , verifylist )
839+
840+ # In base command class ShowOne in cliff, abstract method take_action()
841+ # returns nothing
842+ self .cmd .take_action (parsed_args )
843+ self .volumes_mock .set_image_metadata .assert_called_with (
844+ self .volumes_mock .get ().id , parsed_args .image_property )
845+
846+ def test_volume_set_state (self ):
847+ arglist = [
848+ '--state' , 'error' ,
849+ self .new_volume .id
850+ ]
851+ verifylist = [
852+ ('state' , 'error' ),
853+ ('volume' , self .new_volume .id )
854+ ]
855+
856+ parsed_args = self .check_parser (self .cmd , arglist , verifylist )
857+
858+ result = self .cmd .take_action (parsed_args )
859+ self .volumes_mock .reset_state .assert_called_with (
860+ self .new_volume .id , 'error' )
861+ self .assertIsNone (result )
862+
863+
817864class TestVolumeShow (TestVolume ):
818865
819866 def setUp (self ):
@@ -845,36 +892,6 @@ def test_volume_show(self):
845892 data )
846893
847894
848- class TestVolumeSet (TestVolume ):
849-
850- def setUp (self ):
851- super (TestVolumeSet , self ).setUp ()
852-
853- self .new_volume = volume_fakes .FakeVolume .create_one_volume ()
854- self .volumes_mock .create .return_value = self .new_volume
855-
856- # Get the command object to test
857- self .cmd = volume .SetVolume (self .app , None )
858-
859- def test_volume_set_image_property (self ):
860- arglist = [
861- '--image-property' , 'Alpha=a' ,
862- '--image-property' , 'Beta=b' ,
863- self .new_volume .id ,
864- ]
865- verifylist = [
866- ('image_property' , {'Alpha' : 'a' , 'Beta' : 'b' }),
867- ('volume' , self .new_volume .id ),
868- ]
869- parsed_args = self .check_parser (self .cmd , arglist , verifylist )
870-
871- # In base command class ShowOne in cliff, abstract method take_action()
872- # returns nothing
873- self .cmd .take_action (parsed_args )
874- self .volumes_mock .set_image_metadata .assert_called_with (
875- self .volumes_mock .get ().id , parsed_args .image_property )
876-
877-
878895class TestVolumeUnset (TestVolume ):
879896
880897 def setUp (self ):
0 commit comments