@@ -76,34 +76,6 @@ def test_snapshot_create(self):
7676 self .assertEqual (data , volume_fakes .SNAPSHOT_data )
7777
7878
79- class TestSnapshotShow (TestSnapshot ):
80-
81- def setUp (self ):
82- super (TestSnapshotShow , self ).setUp ()
83-
84- self .snapshots_mock .get .return_value = fakes .FakeResource (
85- None ,
86- copy .deepcopy (volume_fakes .SNAPSHOT ),
87- loaded = True )
88- # Get the command object to test
89- self .cmd = snapshot .ShowSnapshot (self .app , None )
90-
91- def test_snapshot_show (self ):
92- arglist = [
93- volume_fakes .snapshot_id
94- ]
95- verifylist = [
96- ("snapshot" , volume_fakes .snapshot_id )
97- ]
98- parsed_args = self .check_parser (self .cmd , arglist , verifylist )
99-
100- columns , data = self .cmd .take_action (parsed_args )
101- self .snapshots_mock .get .assert_called_with (volume_fakes .snapshot_id )
102-
103- self .assertEqual (volume_fakes .SNAPSHOT_columns , columns )
104- self .assertEqual (volume_fakes .SNAPSHOT_data , data )
105-
106-
10779class TestSnapshotDelete (TestSnapshot ):
10880
10981 def setUp (self ):
@@ -132,80 +104,6 @@ def test_snapshot_delete(self):
132104 self .snapshots_mock .delete .assert_called_with (volume_fakes .snapshot_id )
133105
134106
135- class TestSnapshotSet (TestSnapshot ):
136-
137- def setUp (self ):
138- super (TestSnapshotSet , self ).setUp ()
139-
140- self .snapshots_mock .get .return_value = fakes .FakeResource (
141- None ,
142- copy .deepcopy (volume_fakes .SNAPSHOT ),
143- loaded = True
144- )
145- self .snapshots_mock .set_metadata .return_value = None
146- self .snapshots_mock .update .return_value = None
147- # Get the command object to mock
148- self .cmd = snapshot .SetSnapshot (self .app , None )
149-
150- def test_snapshot_set (self ):
151- arglist = [
152- volume_fakes .snapshot_id ,
153- "--name" , "new_snapshot" ,
154- "--property" , "x=y" ,
155- "--property" , "foo=foo"
156- ]
157- new_property = {"x" : "y" , "foo" : "foo" }
158- verifylist = [
159- ("snapshot" , volume_fakes .snapshot_id ),
160- ("name" , "new_snapshot" ),
161- ("property" , new_property )
162- ]
163-
164- kwargs = {
165- "name" : "new_snapshot" ,
166- }
167- parsed_args = self .check_parser (self .cmd , arglist , verifylist )
168- self .cmd .take_action (parsed_args )
169-
170- self .snapshots_mock .update .assert_called_with (
171- volume_fakes .snapshot_id , ** kwargs )
172- self .snapshots_mock .set_metadata .assert_called_with (
173- volume_fakes .snapshot_id , new_property
174- )
175-
176-
177- class TestSnapshotUnset (TestSnapshot ):
178-
179- def setUp (self ):
180- super (TestSnapshotUnset , self ).setUp ()
181-
182- self .snapshots_mock .get .return_value = fakes .FakeResource (
183- None ,
184- copy .deepcopy (volume_fakes .SNAPSHOT ),
185- loaded = True
186- )
187- self .snapshots_mock .delete_metadata .return_value = None
188- # Get the command object to mock
189- self .cmd = snapshot .UnsetSnapshot (self .app , None )
190-
191- def test_snapshot_unset (self ):
192- arglist = [
193- volume_fakes .snapshot_id ,
194- "--property" , "foo"
195- ]
196- verifylist = [
197- ("snapshot" , volume_fakes .snapshot_id ),
198- ("property" , ["foo" ])
199- ]
200-
201- parsed_args = self .check_parser (self .cmd , arglist , verifylist )
202- self .cmd .take_action (parsed_args )
203-
204- self .snapshots_mock .delete_metadata .assert_called_with (
205- volume_fakes .snapshot_id , ["foo" ]
206- )
207-
208-
209107class TestSnapshotList (TestSnapshot ):
210108
211109 columns = [
@@ -303,3 +201,105 @@ def test_snapshot_list_all_projects(self):
303201 volume_fakes .snapshot_size
304202 ), )
305203 self .assertEqual (datalist , tuple (data ))
204+
205+
206+ class TestSnapshotSet (TestSnapshot ):
207+
208+ def setUp (self ):
209+ super (TestSnapshotSet , self ).setUp ()
210+
211+ self .snapshots_mock .get .return_value = fakes .FakeResource (
212+ None ,
213+ copy .deepcopy (volume_fakes .SNAPSHOT ),
214+ loaded = True
215+ )
216+ self .snapshots_mock .set_metadata .return_value = None
217+ self .snapshots_mock .update .return_value = None
218+ # Get the command object to mock
219+ self .cmd = snapshot .SetSnapshot (self .app , None )
220+
221+ def test_snapshot_set (self ):
222+ arglist = [
223+ volume_fakes .snapshot_id ,
224+ "--name" , "new_snapshot" ,
225+ "--property" , "x=y" ,
226+ "--property" , "foo=foo"
227+ ]
228+ new_property = {"x" : "y" , "foo" : "foo" }
229+ verifylist = [
230+ ("snapshot" , volume_fakes .snapshot_id ),
231+ ("name" , "new_snapshot" ),
232+ ("property" , new_property )
233+ ]
234+
235+ kwargs = {
236+ "name" : "new_snapshot" ,
237+ }
238+ parsed_args = self .check_parser (self .cmd , arglist , verifylist )
239+ self .cmd .take_action (parsed_args )
240+
241+ self .snapshots_mock .update .assert_called_with (
242+ volume_fakes .snapshot_id , ** kwargs )
243+ self .snapshots_mock .set_metadata .assert_called_with (
244+ volume_fakes .snapshot_id , new_property
245+ )
246+
247+
248+ class TestSnapshotShow (TestSnapshot ):
249+
250+ def setUp (self ):
251+ super (TestSnapshotShow , self ).setUp ()
252+
253+ self .snapshots_mock .get .return_value = fakes .FakeResource (
254+ None ,
255+ copy .deepcopy (volume_fakes .SNAPSHOT ),
256+ loaded = True )
257+ # Get the command object to test
258+ self .cmd = snapshot .ShowSnapshot (self .app , None )
259+
260+ def test_snapshot_show (self ):
261+ arglist = [
262+ volume_fakes .snapshot_id
263+ ]
264+ verifylist = [
265+ ("snapshot" , volume_fakes .snapshot_id )
266+ ]
267+ parsed_args = self .check_parser (self .cmd , arglist , verifylist )
268+
269+ columns , data = self .cmd .take_action (parsed_args )
270+ self .snapshots_mock .get .assert_called_with (volume_fakes .snapshot_id )
271+
272+ self .assertEqual (volume_fakes .SNAPSHOT_columns , columns )
273+ self .assertEqual (volume_fakes .SNAPSHOT_data , data )
274+
275+
276+ class TestSnapshotUnset (TestSnapshot ):
277+
278+ def setUp (self ):
279+ super (TestSnapshotUnset , self ).setUp ()
280+
281+ self .snapshots_mock .get .return_value = fakes .FakeResource (
282+ None ,
283+ copy .deepcopy (volume_fakes .SNAPSHOT ),
284+ loaded = True
285+ )
286+ self .snapshots_mock .delete_metadata .return_value = None
287+ # Get the command object to mock
288+ self .cmd = snapshot .UnsetSnapshot (self .app , None )
289+
290+ def test_snapshot_unset (self ):
291+ arglist = [
292+ volume_fakes .snapshot_id ,
293+ "--property" , "foo"
294+ ]
295+ verifylist = [
296+ ("snapshot" , volume_fakes .snapshot_id ),
297+ ("property" , ["foo" ])
298+ ]
299+
300+ parsed_args = self .check_parser (self .cmd , arglist , verifylist )
301+ self .cmd .take_action (parsed_args )
302+
303+ self .snapshots_mock .delete_metadata .assert_called_with (
304+ volume_fakes .snapshot_id , ["foo" ]
305+ )
0 commit comments