Minor fixes on getfoldstatus and imports

temporalio · Apr 9, 2024 · 3f78c09 · 3f78c09
1 parent b345676
commit 3f78c09
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/temporalcli/commands.workflow_trace.go b/temporalcli/commands.workflow_trace.go
@@ -27,7 +27,7 @@ var workflowTraceFoldFlags = map[string]enums.WorkflowExecutionStatus{
 
 func (c *TemporalWorkflowTraceCommand) getFoldStatuses() ([]enums.WorkflowExecutionStatus, error) {
 	// defaults
-	if len(foldFlags) == 0 {
+	if len(c.Fold) == 0 {
 		return []enums.WorkflowExecutionStatus{
 			enums.WORKFLOW_EXECUTION_STATUS_COMPLETED,
 			enums.WORKFLOW_EXECUTION_STATUS_CANCELED,
@@ -37,8 +37,8 @@ func (c *TemporalWorkflowTraceCommand) getFoldStatuses() ([]enums.WorkflowExecut
 
 	// parse flags
 	var values []enums.WorkflowExecutionStatus
-	for _, flag := range foldFlags {
-		status, ok := foldFlag[flag]
+	for _, flag := range c.Fold {
+		status, ok := workflowTraceFoldFlags[flag]
 		if !ok {
 			return nil, fmt.Errorf("fold status %q not recognized", flag)
 		}
@@ -62,7 +62,7 @@ func (c *TemporalWorkflowTraceCommand) run(cctx *CommandContext, _ []string) err
 		Concurrency: c.Concurrency,
 		NoFold:      c.NoFold,
 	}
-	opts.FoldStatuses, err = getFoldStatuses(c.Fold)
+	opts.FoldStatuses, err = c.getFoldStatuses()
 	if err != nil {
 		return err
 	}

diff --git a/temporalcli/commands.workflow_trace_test.go b/temporalcli/commands.workflow_trace_test.go
@@ -6,10 +6,8 @@ import (
 	"time"
 
 	"go.temporal.io/api/enums/v1"
-
-	"go.temporal.io/sdk/workflow"
-
 	"go.temporal.io/sdk/client"
+	"go.temporal.io/sdk/workflow"
 )
 
 func (s *SharedServerSuite) TestWorkflow_Trace_Summary() {