Add clear status test for file backend #38

pepkit · Aug 4, 2023 · ef2481a · ef2481a
1 parent ca1e18d
commit ef2481a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/test_pipestat.py b/tests/test_pipestat.py
@@ -72,9 +72,15 @@ def test_basics(
             psm = PipestatManager(**args)
             psm.report(sample_name=rec_id, values=val, force_overwrite=True)
             val_name = list(val.keys())[0]
+            psm.set_status(status_identifier="running", sample_name=rec_id)
+            status = psm.get_status(sample_name=rec_id)
+            assert status == "running"
             assert val_name in psm.retrieve(sample_name=rec_id)
             psm.remove(sample_name=rec_id, result_identifier=val_name)
             if backend == "file":
+                psm.clear_status(sample_name=rec_id)
+                status = psm.get_status(sample_name=rec_id)
+                assert status is None
                 with pytest.raises(PipestatDataError):
                     psm.retrieve(sample_name=rec_id)
             if backend == "db":