Merge "Collect steps execution result"

1 files changed, 29 insertions, 0 deletions

diff --git a/tests/test_store_state.py b/tests/test_store_state.py
new file mode 100644
index 0000000..e0c8b6b
--- /dev/null
+++ b/tests/test_store_state.py
@@ -0,0 +1,29 @@
+import pytest
+from onaptests.steps.base import BaseStep
+
+
+class TestStep(BaseStep):
+
+    @BaseStep.store_state
+    def execute(self):
+        return super().execute()
+
+
+class TestFailStep(BaseStep):
+
+    @BaseStep.store_state
+    def execute(self):
+        super().execute()
+        raise Exception
+
+
+def test_store_state():
+    ts = TestStep()
+    ts.execute()
+    assert ts.reports_collection.report == {"TestStep": "PASS"}
+
+    fs = TestFailStep()
+    fs.add_step(TestStep())
+    with pytest.raises(Exception):
+        fs.execute()
+    fs.reports_collection.report == {"TestFailStep": "FAIL", "TestStep": "PASS"}