Added StartWorkflowWithContext to executor

Author: jmigueprieto

sdk/workflow/executor/executor.go

@@ -131,8 +131,16 @@ func (e *WorkflowExecutor) MonitorExecution(workflowId string) (workflowMonitor
 // StartWorkflow Start workflows
 // Returns the id of the newly created workflow
 func (e *WorkflowExecutor) StartWorkflow(startWorkflowRequest *model.StartWorkflowRequest) (workflowId string, err error) {
+	return e.StartWorkflowWithContext(context.Background(), startWorkflowRequest)
+}
+
+func (e *WorkflowExecutor) StartWorkflowWithContext(ctx context.Context, startWorkflowRequest *model.StartWorkflowRequest) (workflowId string, err error) {
+	if err := ctx.Err(); err != nil {
+		return "", err
+	}
+
 	id, _, err := e.workflowClient.StartWorkflowWithRequest(
-		context.Background(),
+		ctx,
 		*startWorkflowRequest,
 	)
 	if err != nil {

test/integration_tests/executor_test.go

@@ -1,14 +1,14 @@
 package integration_tests
 
 import (
+	"context"
 	"fmt"
-	"testing"
-
 	"github.com/conductor-sdk/conductor-go/sdk/client"
 	"github.com/conductor-sdk/conductor-go/sdk/model"
 	"github.com/conductor-sdk/conductor-go/sdk/workflow"
 	"github.com/conductor-sdk/conductor-go/test/testdata"
 	"github.com/stretchr/testify/assert"
+	"testing"
 )
 
 const (
@@ -90,3 +90,14 @@ func TestUpdate(t *testing.T) {
 		assert.Fail(t, "err is not of type GenericSwaggerError")
 	}
 }
+
+func TestStartWorkflowWithContext(t *testing.T) {
+	executor := testdata.WorkflowExecutor
+
+	ctx, cancel := context.WithCancel(context.Background())
+	cancel()
+
+	_, err := executor.StartWorkflowWithContext(ctx, &model.StartWorkflowRequest{})
+	assert.Error(t, err, "StartWorkflowWithContext is expected to return an error")
+	assert.Equal(t, context.Canceled, err, "Expected context canceled error")
+}