test fixes & extension

Open-EO · Sep 11, 2024 · 0c2831a · 0c2831a
1 parent 525de5e
commit 0c2831a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/extra/test_job_management.py b/tests/extra/test_job_management.py
@@ -628,13 +628,14 @@ def test_partial_read_write(self, tmp_path, orig: pandas.DataFrame):
         db.persist(orig)
         assert path.exists()
 
-        loaded = db.get_by_status(include=["not_started"], max=2)
+        loaded = db.get_by_status(statuses=["not_started"], max=2)
+        assert db.count_by_status(statuses=["not_started"])["not_started"] >1
 
         assert len(loaded) == 2
         loaded.loc[0,"status"] = "running"
         loaded.loc[1, "status"] = "error"
-
         db.persist(loaded)
+        assert db.count_by_status(statuses=["error"])["error"] == 1
 
         all = db.read()
         assert len(all) == len(orig)