@@ -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