From 2349840dd2be8284d02b7d4513691e2da5566b32 Mon Sep 17 00:00:00 2001
From: Lehman Garrison <lgarrison@flatironinstitute.org>
Date: Wed, 20 Nov 2024 19:27:45 -0500
Subject: [PATCH] Test --resume-from

---
 tests/test_slurm/run.sh | 8 ++++++++
 tests/test_ssh/run.sh   | 8 ++++++++
 2 files changed, 16 insertions(+)

diff --git a/tests/test_slurm/run.sh b/tests/test_slurm/run.sh
index edde8b0..b4ee5dd 100755
--- a/tests/test_slurm/run.sh
+++ b/tests/test_slurm/run.sh
@@ -12,6 +12,14 @@ salloc -n 2 disBatch Tasks
 [[ -f A.txt && -f B.txt && -f C.txt ]]
 success=$?
 
+# Add a task and check that we can resume
+rm A.txt B.txt C.txt
+echo "touch D.txt" >> Tasks
+salloc -n 2 disBatch Tasks -r Tasks*_status.txt
+
+[[ -f D.txt && ! -f A.txt && ! -f B.txt && ! -f C.txt ]]
+success=$((success + $?))
+
 # Test empty task file
 salloc -n 2 disBatch /dev/null
 success=$((success + $?))
diff --git a/tests/test_ssh/run.sh b/tests/test_ssh/run.sh
index e767055..691a127 100755
--- a/tests/test_ssh/run.sh
+++ b/tests/test_ssh/run.sh
@@ -12,6 +12,14 @@ disBatch -s localhost:2 Tasks
 [[ -f A.txt && -f B.txt && -f C.txt ]]
 success=$?
 
+# Add a task and check that we can resume
+rm A.txt B.txt C.txt
+echo "touch D.txt" >> Tasks
+disBatch -s localhost:2 Tasks -r Tasks*_status.txt
+
+[[ -f D.txt && ! -f A.txt && ! -f B.txt && ! -f C.txt ]]
+success=$((success + $?))
+
 # Test empty task file
 disBatch -s localhost:2 /dev/null
 success=$((success + $?))