Add waitForStop

Signed-off-by: JenTing Hsiao <hsiaoairplane@gmail.com>
gitpod-io · Sep 13, 2022 · 8dce48a · 8dce48a
1 parent 094f593
commit 8dce48a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/pkg/integration/workspace.go b/test/pkg/integration/workspace.go
@@ -206,6 +206,10 @@ func LaunchWorkspaceDirectly(ctx context.Context, api *ComponentAPI, opts ...Lau
 			}
 			break
 		}
+
+		if !waitForStop {
+			return nil, nil
+		}
 		for {
 			lastStatus, err = WaitForWorkspaceStop(tctx, api, req.Id)
 			if st, ok := status.FromError(err); ok && st.Code() == codes.Unavailable {