From 50671eaeb76bee15aabd4ad79f9b1149b9739cee Mon Sep 17 00:00:00 2001
From: Lucky samadhiya <lucky.samadhiya@nectechnologies.in>
Date: Wed, 21 Sep 2016 12:25:39 +0530
Subject: [PATCH] Add unit test case for deletion of execution in case of
 (error and cancelled)

Change-Id: I07d5e01883adc7d1bb3db73b7b5ccfc17772c867
---
 .../unit/api/v2/test_action_executions.py     | 62 +++++++++++++++++++
 1 file changed, 62 insertions(+)

diff --git a/mistral/tests/unit/api/v2/test_action_executions.py b/mistral/tests/unit/api/v2/test_action_executions.py
index 9f82fe47..b7581e83 100644
--- a/mistral/tests/unit/api/v2/test_action_executions.py
+++ b/mistral/tests/unit/api/v2/test_action_executions.py
@@ -66,6 +66,34 @@ AD_HOC_ACTION_EX_DB = models.ActionExecution(
     updated_at=datetime.datetime(1970, 1, 1)
 )
 
+AD_HOC_ACTION_EX_ERROR = models.ActionExecution(
+    id='123',
+    state=states.ERROR,
+    state_info=states.ERROR,
+    tags=['foo', 'fee'],
+    name='std.echo',
+    description='something',
+    accepted=True,
+    input={},
+    output={},
+    created_at=datetime.datetime(1970, 1, 1),
+    updated_at=datetime.datetime(1970, 1, 1)
+)
+
+AD_HOC_ACTION_EX_CANCELLED = models.ActionExecution(
+    id='123',
+    state=states.CANCELLED,
+    state_info=states.CANCELLED,
+    tags=['foo', 'fee'],
+    name='std.echo',
+    description='something',
+    accepted=True,
+    input={},
+    output={},
+    created_at=datetime.datetime(1970, 1, 1),
+    updated_at=datetime.datetime(1970, 1, 1)
+)
+
 ACTION_EX_DB_NOT_COMPLETE = models.ActionExecution(
     id='123',
     state=states.RUNNING,
@@ -144,6 +172,14 @@ MOCK_ACTION = mock.MagicMock(return_value=ACTION_EX_DB)
 MOCK_ACTION_NOT_COMPLETE = mock.MagicMock(
     return_value=ACTION_EX_DB_NOT_COMPLETE
 )
+
+MOCK_ACTION_COMPLETE_ERROR = mock.MagicMock(
+    return_value=AD_HOC_ACTION_EX_ERROR
+)
+MOCK_ACTION_COMPLETE_CANCELLED = mock.MagicMock(
+    return_value=AD_HOC_ACTION_EX_CANCELLED
+)
+
 MOCK_AD_HOC_ACTION = mock.MagicMock(return_value=AD_HOC_ACTION_EX_DB)
 MOCK_ACTIONS = mock.MagicMock(return_value=[ACTION_EX_DB])
 MOCK_EMPTY = mock.MagicMock(return_value=[])
@@ -440,3 +476,29 @@ class TestActionExecutionsController(base.APITest):
             "Only completed action execution can be deleted",
             resp.body.decode()
         )
+
+    @mock.patch.object(
+        db_api,
+        'get_action_execution',
+        MOCK_ACTION_COMPLETE_ERROR
+    )
+    @mock.patch.object(db_api, 'delete_action_execution', MOCK_DELETE)
+    def test_delete_action_exeuction_complete_error(self):
+        cfg.CONF.set_default('allow_action_execution_deletion', True, 'api')
+
+        resp = self.app.delete('/v2/action_executions/123', expect_errors=True)
+
+        self.assertEqual(204, resp.status_int)
+
+    @mock.patch.object(
+        db_api,
+        'get_action_execution',
+        MOCK_ACTION_COMPLETE_CANCELLED
+    )
+    @mock.patch.object(db_api, 'delete_action_execution', MOCK_DELETE)
+    def test_delete_action_exeuction_complete_cancelled(self):
+        cfg.CONF.set_default('allow_action_execution_deletion', True, 'api')
+
+        resp = self.app.delete('/v2/action_executions/123', expect_errors=True)
+
+        self.assertEqual(204, resp.status_int)