get_the_plan auto-creates a plan if it doesn't exist

Change-Id: I9bf9d800aeb73ede836bfbd477d38cf927f2085b
This commit is contained in:
Tzu-Mainn Chen 2014-08-05 14:51:57 +02:00
parent 7ed35dee84
commit ab767a61b0
2 changed files with 14 additions and 0 deletions

View File

@ -131,6 +131,9 @@ class OvercloudPlan(base.APIDictWrapper):
plan_list = cls.list(request)
for plan in plan_list:
return plan
# if plan doesn't exist, create it
plan = cls.create(request, 'overcloud', 'overcloud')
return plan
@classmethod
def delete(cls, request, plan_id):

View File

@ -47,6 +47,17 @@ class TuskarAPITests(test.APITestCase):
self.assertIsInstance(ret_val, api.tuskar.OvercloudPlan)
def test_plan_get_the_plan(self):
plan = self.tuskarclient_plans.first()
with patch('tuskar_ui.test.test_driver.tuskar_driver.Plan.list',
return_value=[]):
with patch('tuskar_ui.test.test_driver.tuskar_driver.Plan.create',
return_value=plan):
ret_val = api.tuskar.OvercloudPlan.get_the_plan(self.request)
self.assertIsInstance(ret_val, api.tuskar.OvercloudPlan)
def test_plan_delete(self):
plan = self.tuskarclient_plans.first()
api.tuskar.OvercloudPlan.delete(self.request, plan['id'])