From 296ee518a7ceec442cc5bb7222a6fd402f8a10e2 Mon Sep 17 00:00:00 2001
From: Tim Callow <timjcallow@gmail.com>
Date: Tue, 7 Jan 2025 11:40:23 +0100
Subject: [PATCH] Fix remaining issues

---
 datalad_slurm/tests/test_register.py | 18 +++++++++---------
 1 file changed, 9 insertions(+), 9 deletions(-)

diff --git a/datalad_slurm/tests/test_register.py b/datalad_slurm/tests/test_register.py
index 759d100..5f3253e 100644
--- a/datalad_slurm/tests/test_register.py
+++ b/datalad_slurm/tests/test_register.py
@@ -1,5 +1,5 @@
 from datalad.tests.utils_pytest import assert_result_count
-
+import datalad.support.exceptions as dl_exceptions
 
 def test_register():
     import datalad.api as da
@@ -9,13 +9,13 @@ def test_register():
     assert_result_count(
         da.schedule(cmd="echo test", dry_run="basic"),
         1,
-        action='demo')
+        status="ok")
     assert_result_count(
         da.finish(),
-        1,
-        action='demo')
-    assert_result_count(
-        da.reschedule(since="HEAD~1", report=True),
-        1,
-        action='demo')
-
+        0,
+        status="ok")
+    try:
+        da.reschedule(since="HEAD~1", report=True)
+        assert False
+    except dl_exceptions.IncompleteResultsError:
+        assert True