Merge "Use FakeFlavor in TestFlavorUnset."

1 files changed, 4 insertions, 7 deletions

diff --git a/openstackclient/tests/compute/v2/test_flavor.py b/openstackclient/tests/compute/v2/test_flavor.py
index 8ad1f85e..095b5f28 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)