From db516b2c96dd449daad5ff6c9479d87f1457f410 Mon Sep 17 00:00:00 2001
From: Tang Chen <tangchen@cn.fujitsu.com>
Date: Sat, 28 Nov 2015 16:09:01 +0800
Subject: [PATCH] Use FakeFlavor in TestFlavorUnset.
Change-Id: Ifbd360db39d380efd9632300367b13283ac75f54
Implements: blueprint improve-flavor-unit-test
---
openstackclient/tests/compute/v2/test_flavor.py | 11 ++++-------
1 file changed, 4 insertions(+), 7 deletions(-)
diff --git a/openstackclient/tests/compute/v2/test_flavor.py b/openstackclient/tests/compute/v2/test_flavor.py
index 8ad1f85e20..095b5f2812 100644
--- a/openstackclient/tests/compute/v2/test_flavor.py
+++ b/openstackclient/tests/compute/v2/test_flavor.py
@@ -13,8 +13,6 @@
# under the License.
#
-import copy
-
from openstackclient.compute.v2 import flavor
from openstackclient.tests.compute.v2 import fakes as compute_fakes
@@ -242,14 +240,13 @@ class TestFlavorSet(TestFlavor):
class TestFlavorUnset(TestFlavor):
+ # Return value of self.flavors_mock.find().
+ flavor = compute_fakes.FakeFlavor.create_one_flavor()
+
def setUp(self):
super(TestFlavorUnset, self).setUp()
- self.flavors_mock.find.return_value = compute_fakes.FakeFlavorResource(
- None,
- copy.deepcopy(compute_fakes.FLAVOR),
- loaded=True,
- )
+ self.flavors_mock.find.return_value = self.flavor
self.cmd = flavor.UnsetFlavor(self.app, None)