NOISSUE - Fix race condition in tests (ultravioletrs#248)

Signed-off-by: Sammy Oina <[email protected]>
WashingtonKK · Sep 20, 2024 · a7caa59 · a7caa59
1 parent eab3a06
commit a7caa59
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/agent/service.go b/agent/service.go
@@ -90,8 +90,6 @@ func New(ctx context.Context, logger *slog.Logger, eventSvc events.Service, cmp
 		quoteProvider: quoteProvider,
 	}
 
-	go svc.sm.Start(ctx)
-	svc.sm.SendEvent(start)
 	svc.sm.StateFunctions[Idle] = svc.publishEvent(IdleState.String(), json.RawMessage{})
 	svc.sm.StateFunctions[ReceivingManifest] = svc.publishEvent(InProgress.String(), json.RawMessage{})
 	svc.sm.StateFunctions[ReceivingAlgorithm] = svc.publishEvent(InProgress.String(), json.RawMessage{})
@@ -101,6 +99,9 @@ func New(ctx context.Context, logger *slog.Logger, eventSvc events.Service, cmp
 	svc.sm.StateFunctions[Running] = svc.runComputation
 	svc.sm.StateFunctions[Failed] = svc.publishEvent(Failed.String(), json.RawMessage{})
 
+	go svc.sm.Start(ctx)
+	svc.sm.SendEvent(start)
+
 	svc.computation = cmp
 
 	svc.sm.SendEvent(manifestReceived)