From 0eecedc0b69d61e457be54a18fe75b6255f77def Mon Sep 17 00:00:00 2001
From: xiexs <xiexs@cn.fujitsu.com>
Date: Mon, 30 Nov 2015 00:18:17 -0500
Subject: [PATCH] Add unit testcases for "openstack flavor delete"

Change-Id: If23a71c678193e5c9c91300fddd17e79f674bf82
---
 .../tests/compute/v2/test_flavor.py           | 46 +++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/openstackclient/tests/compute/v2/test_flavor.py b/openstackclient/tests/compute/v2/test_flavor.py
index 095b5f2812..60356efa8b 100644
--- a/openstackclient/tests/compute/v2/test_flavor.py
+++ b/openstackclient/tests/compute/v2/test_flavor.py
@@ -13,6 +13,7 @@
 #   under the License.
 #
 
+from openstackclient.common import exceptions
 from openstackclient.compute.v2 import flavor
 from openstackclient.tests.compute.v2 import fakes as compute_fakes
 
@@ -27,6 +28,51 @@ class TestFlavor(compute_fakes.TestComputev2):
         self.flavors_mock.reset_mock()
 
 
+class TestFlavorDelete(TestFlavor):
+
+    flavor = compute_fakes.FakeFlavor.create_one_flavor()
+
+    def setUp(self):
+        super(TestFlavorDelete, self).setUp()
+
+        self.flavors_mock.get.return_value = self.flavor
+        self.flavors_mock.delete.return_value = None
+
+        self.cmd = flavor.DeleteFlavor(self.app, None)
+
+    def test_flavor_delete(self):
+        arglist = [
+            self.flavor.id
+        ]
+        verifylist = [
+            ('flavor', self.flavor.id),
+        ]
+
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        self.cmd.take_action(parsed_args)
+
+        self.flavors_mock.delete.assert_called_with(self.flavor.id)
+
+    def test_flavor_delete_with_unexist_flavor(self):
+        self.flavors_mock.get.side_effect = exceptions.NotFound(None)
+        self.flavors_mock.find.side_effect = exceptions.NotFound(None)
+
+        arglist = [
+            'unexist_flavor'
+        ]
+        verifylist = [
+            ('flavor', 'unexist_flavor'),
+        ]
+
+        parsed_args = self.check_parser(self.cmd, arglist, verifylist)
+
+        self.assertRaises(
+            exceptions.CommandError,
+            self.cmd.take_action,
+            parsed_args)
+
+
 class TestFlavorList(TestFlavor):
 
     # Return value of self.flavors_mock.list().