test: Add a handling if the workspace already has gone when try to delete it

Author: utam0k

test/pkg/integration/workspace.go

@@ -301,24 +301,27 @@ func stopWsF(t *testing.T, instanceID string, api *ComponentAPI) stopWorkspaceFu
 			break
 		}
 
-		if !waitForStop {
-			wm, err := api.WorkspaceManager()
-			if err != nil {
-				return nil, err
-			}
+		wm, err := api.WorkspaceManager()
+		if err != nil {
+			return nil, err
+		}
 
-			dr, err := wm.DescribeWorkspace(sctx, &wsmanapi.DescribeWorkspaceRequest{
-				Id: instanceID,
-			})
-			if err != nil {
-				return nil, err
-			}
+		dr, err := wm.DescribeWorkspace(sctx, &wsmanapi.DescribeWorkspaceRequest{
+			Id: instanceID,
+		})
+		s, ok := status.FromError(err)
+		if ok && s.Code() == codes.NotFound {
+			t.Log("the workspace is already gone")
+			return &wsmanapi.WorkspaceStatus{
+				Id:    instanceID,
+				Phase: wsmanapi.WorkspacePhase_STOPPED,
+			}, nil
+		} else if err != nil {
+			return nil, err
+		}
 
-			s, ok := status.FromError(err)
-			if ok && s.Code() == codes.NotFound {
-				return nil, err
-			}
-			return dr.Status, err
+		if !waitForStop {
+			return dr.Status, nil
 		}
 
 		var lastStatus *wsmanapi.WorkspaceStatus
@@ -335,7 +338,7 @@ func stopWsF(t *testing.T, instanceID string, api *ComponentAPI) stopWorkspaceFu
 			}
 			break
 		}
-		return lastStatus, err
+		return lastStatus, nil
 	}
 }