test fix

hapifhir · Apr 16, 2024 · e7ed124 · e7ed124
1 parent 3914445
commit e7ed124
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/...hir-storage-batch2/src/main/java/ca/uhn/fhir/batch2/maintenance/JobInstanceProcessor.java b/...hir-storage-batch2/src/main/java/ca/uhn/fhir/batch2/maintenance/JobInstanceProcessor.java
@@ -43,6 +43,7 @@
 
 import java.util.Iterator;
 import java.util.List;
+import java.util.Optional;
 import java.util.Set;
 
 public class JobInstanceProcessor {
@@ -101,10 +102,14 @@ public void process() {
 		triggerGatedExecutions(theInstance, jobDefinition);
 
 		if (theInstance.hasGatedStep() && theInstance.isRunning()) {
-			JobInstance updatedInstance =
-					myJobPersistence.fetchInstance(theInstance.getInstanceId()).orElseThrow();
+			Optional<JobInstance> updatedInstance = myJobPersistence.fetchInstance(theInstance.getInstanceId());
+
+			if (updatedInstance.isEmpty()){
+				return;
+			}
+
 			JobWorkCursor<?, ?, ?> jobWorkCursor = JobWorkCursor.fromJobDefinitionAndRequestedStepId(
-					jobDefinition, updatedInstance.getCurrentGatedStepId());
+					jobDefinition, updatedInstance.get().getCurrentGatedStepId());
 			if (jobWorkCursor.isReductionStep()) {
 				// Reduction step work chunks should never be sent to the queue but to its specific service instead.
 				triggerReductionStep(theInstance, jobWorkCursor);