Add waitForStop

jenting · jenting · commit 7f8d3e33f4a8 · 2022-09-13T08:48:21.000+08:00
Signed-off-by: JenTing Hsiao &lt;hsiaoairplane@gmail.com&gt;
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 {

Original file line number	Diff line number	Diff line change
`@@ -206,6 +206,10 @@ func LaunchWorkspaceDirectly(ctx context.Context, api *ComponentAPI, opts ...Lau`
`206`	`206`	`}`
`207`	`207`	`break`
`208`	`208`	`}`
	`209`	`+`
	`210`	`+ if !waitForStop {`
	`211`	`+ return nil, nil`
	`212`	`+ }`
`209`	`213`	`for {`
`210`	`214`	`lastStatus, err = WaitForWorkspaceStop(tctx, api, req.Id)`
`211`	`215`	`if st, ok := status.FromError(err); ok && st.Code() == codes.Unavailable {`