From dfbf41c55fb759184e60663ee667a1e2ad6e19fa Mon Sep 17 00:00:00 2001
From: Tang Chen <tangchen@cn.fujitsu.com>
Date: Sat, 28 Nov 2015 16:05:14 +0800
Subject: [PATCH] Use FakeFlavor in TestFlavorSet.

Change-Id: I335298b07afb3f969c76748527dda06cb5393fa8
Implements: blueprint improve-flavor-unit-test
---
 openstackclient/tests/compute/v2/test_flavor.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/openstackclient/tests/compute/v2/test_flavor.py b/openstackclient/tests/compute/v2/test_flavor.py
index bafa06c304..8ad1f85e20 100644
--- a/openstackclient/tests/compute/v2/test_flavor.py
+++ b/openstackclient/tests/compute/v2/test_flavor.py
@@ -210,14 +210,13 @@ class TestFlavorList(TestFlavor):
 
 class TestFlavorSet(TestFlavor):
 
+    # Return value of self.flavors_mock.find().
+    flavor = compute_fakes.FakeFlavor.create_one_flavor()
+
     def setUp(self):
         super(TestFlavorSet, 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.SetFlavor(self.app, None)