Migrate integration tests to new API

internal/internal_workflow_client.go

@@ -69,9 +69,9 @@ var (
 )
 
 var (
-	errUnsupportedOperation     = fmt.Errorf("unsupported operation")
-	errInvalidServerResponse    = fmt.Errorf("invalid server response")
-	errInvalidWorkflowOperation = fmt.Errorf("invalid WithStartOperation")
+	errUnsupportedOperation              = fmt.Errorf("unsupported operation")
+	errInvalidServerResponse             = fmt.Errorf("invalid server response")
+	errInvalidWithStartWorkflowOperation = fmt.Errorf("invalid WithStartWorkflowOperation")
 )
 
 const (
@@ -1189,6 +1189,13 @@ func (wc *WorkflowClient) UpdateWithStartWorkflow(
 		return nil, err
 	}
 
+	if updateOptions.RunID != "" {
+		return nil, errors.New("invalid UpdateWorkflowOptions: RunID cannot be set for UpdateWithStartWorkflow because the workflow might not be running")
+	}
+	if updateOptions.FirstExecutionRunID != "" {
+		return nil, errors.New("invalid UpdateWorkflowOptions: FirstExecutionRunID cannot be set for UpdateWithStartWorkflow because the workflow might not be running")
+	}
+
 	updateInput, err := createUpdateWorkflowInput(updateOptions)
 	if err != nil {
 		return nil, err
@@ -1721,19 +1728,12 @@ func (w *workflowClientInterceptor) UpdateWithStartWorkflow(
 	updateInput *ClientUpdateWorkflowInput,
 	startOperation *WithStartWorkflowOperation,
 ) (WorkflowUpdateHandle, error) {
-	if updateInput.RunID != "" {
-		return nil, errors.New("RunID and StartWorkflowInput cannot both be set")
-	}
-	if updateInput.FirstExecutionRunID != "" {
-		return nil, errors.New("FirstExecutionRunID and StartWorkflowInput cannot both be set")
-	}
-
 	// Create start request
 	if err := startOperation.markExecuted(); err != nil {
-		return nil, fmt.Errorf("%w: %w", errInvalidWorkflowOperation, err)
+		return nil, fmt.Errorf("%w: %w", errInvalidWithStartWorkflowOperation, err)
 	}
 	if startOperation.err != nil {
-		return nil, fmt.Errorf("%w: %w", errInvalidWorkflowOperation, startOperation.err)
+		return nil, fmt.Errorf("%w: %w", errInvalidWithStartWorkflowOperation, startOperation.err)
 	}
 	startReq, err := w.createStartWorkflowRequest(ctx, startOperation.input)
 	if err != nil {
@@ -1743,7 +1743,7 @@ func (w *workflowClientInterceptor) UpdateWithStartWorkflow(
 	// Create update request
 	updateReq, err := w.createUpdateWorkflowRequest(ctx, updateInput)
 	if err != nil {
-		return nil, fmt.Errorf("%w: %w", errInvalidWorkflowOperation, err)
+		return nil, fmt.Errorf("%w: %w", errInvalidWithStartWorkflowOperation, err)
 	}
 	if updateReq.WorkflowExecution.WorkflowId == "" {
 		updateReq.WorkflowExecution.WorkflowId = startReq.WorkflowId
@@ -1782,7 +1782,7 @@ func (w *workflowClientInterceptor) UpdateWithStartWorkflow(
 	}
 	handle, err := w.updateHandleFromResponse(ctx, updateReq.WaitPolicy.LifecycleStage, updateResp)
 	if err != nil {
-		return nil, fmt.Errorf("%w: %w", errInvalidWorkflowOperation, err)
+		return nil, fmt.Errorf("%w: %w", errInvalidWithStartWorkflowOperation, err)
 	}
 	return handle, nil
 }
@@ -1857,7 +1857,7 @@ func (w *workflowClientInterceptor) updateWithStartWorkflow(
 					}
 				case *workflowservice.ExecuteMultiOperationRequest_Operation_UpdateWorkflow:
 					if !errors.As(opErr, &abortedErr) {
-						startErr = fmt.Errorf("%w: %w", errInvalidWorkflowOperation, opErr)
+						startErr = fmt.Errorf("%w: %w", errInvalidWithStartWorkflowOperation, opErr)
 					}
 				default:
 					// this would only happen if a case statement for a newly added operation is missing above