fix tests

Signed-off-by: WashingtonKK <[email protected]>

manager/service_test.go

@@ -124,6 +124,10 @@ func TestRun(t *testing.T) {
 }
 
 func TestStop(t *testing.T) {
+	vmf := new(mocks.Provider)
+	vmMock := new(mocks.VM)
+	vmf.On("Execute", mock.Anything, mock.Anything, mock.Anything).Return(vmMock)
+
 	tests := []struct {
 		name           string
 		computationID  string
@@ -156,8 +160,12 @@ func TestStop(t *testing.T) {
 
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
+			logger := slog.Default()
+			eventsChan := make(chan *manager.ClientStreamMessage, 10)
 			ms := &managerService{
-				vms: make(map[string]vm.VM),
+				logger:     logger,
+				vms:        make(map[string]vm.VM),
+				eventsChan: eventsChan,
 			}
 			vmMock := new(mocks.VM)
 
@@ -180,6 +188,11 @@ func TestStop(t *testing.T) {
 				assert.NoError(t, err)
 				assert.Len(t, ms.vms, 0)
 			}
+
+			// Clear the events channel
+			for len(eventsChan) > 0 {
+				<-eventsChan
+			}
 		})
 	}
 }