allocrunnersim: stop batch tasks

hashicorp-forge · Apr 10, 2024 · eb0c912 · eb0c912
1 parent 5277d30
commit eb0c912
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/allocrunnersim/allocrunnersim.go b/allocrunnersim/allocrunnersim.go
@@ -174,6 +174,11 @@ func (ar *simulatedAllocRunner) Run() {
 	ar.allocStateLock.Unlock()
 
 	ar.updateAllocAndSendUpdate(taskStates)
+
+	go func() {
+		time.Sleep(10 * time.Second)
+		ar.stopAll()
+	}()
 }
 
 // updateAllocAndSendUpdate is a small helper that builds a new allocation