Use FakeFlavor in TestFlavorUnset.

Change-Id: Ifbd360db39d380efd9632300367b13283ac75f54
Implements: blueprint improve-flavor-unit-test
This commit is contained in:
Tang Chen 2015-11-28 16:09:01 +08:00
parent dfbf41c55f
commit db516b2c96

@ -13,8 +13,6 @@
# under the License. # under the License.
# #
import copy
from openstackclient.compute.v2 import flavor from openstackclient.compute.v2 import flavor
from openstackclient.tests.compute.v2 import fakes as compute_fakes from openstackclient.tests.compute.v2 import fakes as compute_fakes
@ -242,14 +240,13 @@ class TestFlavorSet(TestFlavor):
class TestFlavorUnset(TestFlavor): class TestFlavorUnset(TestFlavor):
# Return value of self.flavors_mock.find().
flavor = compute_fakes.FakeFlavor.create_one_flavor()
def setUp(self): def setUp(self):
super(TestFlavorUnset, self).setUp() super(TestFlavorUnset, self).setUp()
self.flavors_mock.find.return_value = compute_fakes.FakeFlavorResource( self.flavors_mock.find.return_value = self.flavor
None,
copy.deepcopy(compute_fakes.FLAVOR),
loaded=True,
)
self.cmd = flavor.UnsetFlavor(self.app, None) self.cmd = flavor.UnsetFlavor(self.app, None)