diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_reports_collection.py | 10 | ||||
-rw-r--r-- | tests/test_store_state.py | 29 |
2 files changed, 39 insertions, 0 deletions
diff --git a/tests/test_reports_collection.py b/tests/test_reports_collection.py new file mode 100644 index 0000000..264b6b4 --- /dev/null +++ b/tests/test_reports_collection.py @@ -0,0 +1,10 @@ + +from onaptests.steps.reports_collection import ReportsCollection + + +def test_reports_collection(): + rc = ReportsCollection() + assert rc.report == {} + + rc.put({"a": "b"}) + assert rc.report == {"a": "b"} 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"} |