Updating responses from workflows building block to be more generic

Signed-off-by: Ryan Lettieri <ryanLettieri@microsoft.com>
This commit is contained in:
Ryan Lettieri 2022-10-21 23:58:48 -06:00
parent d8d8d3d5bd
commit d7ef6d41bc
3 changed files with 9 additions and 8 deletions

View File

@ -70,16 +70,16 @@ func ConformanceTests(t *testing.T, props map[string]string, workflowItem workfl
assert.NoError(t, err)
resp, err := workflowItem.Get(context.Background(), wf)
assert.NoError(t, err)
assert.Equal(t, "Running", resp.Status)
assert.Equal(t, "Running", resp.Metadata["status"])
time.Sleep(5 * time.Second)
resp, err = workflowItem.Get(context.Background(), wf)
assert.NoError(t, err)
assert.Equal(t, resp.Status, "Running")
assert.Equal(t, resp.Metadata["status"], "Running")
err = workflowItem.Terminate(context.Background(), wf)
assert.NoError(t, err)
resp, err = workflowItem.Get(context.Background(), wf)
assert.NoError(t, err)
assert.Equal(t, "Terminated", resp.Status)
assert.Equal(t, "Terminated", resp.Metadata["status"])
})
testLogger.Info("Start test done.")
}

View File

@ -2,7 +2,6 @@ package workflows
type StateResponse struct {
WFInfo WorkflowReference
StartTime string `json:"start_time"`
TaskQueue string `json:"task_queue"`
Status string `json:"status"`
StartTime string `json:"start_time"`
Metadata map[string]string `json:"metadata"`
}

View File

@ -115,8 +115,10 @@ func (c *TemporalWF) Get(ctx context.Context, req *workflows.WorkflowReference)
outputStruct := workflows.StateResponse{
WFInfo: workflows.WorkflowReference{InstanceID: req.InstanceID},
StartTime: resp.WorkflowExecutionInfo.StartTime.Format(time.RFC3339),
TaskQueue: resp.WorkflowExecutionInfo.GetTaskQueue(),
Status: lookupStatus(resp.WorkflowExecutionInfo.Status),
Metadata: map[string]string{
"task_queue": resp.WorkflowExecutionInfo.GetTaskQueue(),
"status": lookupStatus(resp.WorkflowExecutionInfo.Status),
},
}
return &outputStruct, nil