单元测试中如何模拟 cadence RegisterDelayedCallback 执行相同的工作流程？

是否可以使用执行相同节奏工作流程的 registerdelayedcallback 运行单元测试？

我有以下代码运行工作流两次，第一次执行保存回调令牌，第二次执行检索保存的令牌以异步完成活动。

工作流.go

package workflow

import (
    "context"
    "encoding/base64"
    "fmt"

    "go.uber.org/cadence/activity"
    "go.uber.org/cadence/workflow"
)

type workflowimpl struct {
    worker.worker
    client.client
}

func (w workflowimpl) tactivity(ctx context.context, action string) error {
    fmt.println("tactivity started", action)
    if action != "approved" {
        activityinfo := activity.getinfo(ctx)
        callbacktoken := base64.stdencoding.encodetostring(activityinfo.tasktoken)
        fmt.println("save callbacktoken", callbacktoken)
        // saves callbacktoken.

        return activity.errresultpending
    }

    fmt.println("approved")
    // do some approved things.
    // get saved callback token.
    // call w.completeactivity() with the saved callback token.
    return nil
}

func (w workflowimpl) tworkflow(ctx workflow.context, action string) (result string, err error) {
    fmt.println("tworkflow started", action)

    waitchannel := workflow.newchannel(ctx)
    workflow.go(ctx, func(ctx workflow.context) {
        if err := workflow.executeactivity(ctx, w.tactivity, action).get(ctx, nil); err != nil {
            // do nothing, keep workflow open.
            return
        }

        waitchannel.send(ctx, "ok")
    })

    var signal string
    waitchannel.receive(ctx, &signal)

    return signal, nil
}

workflow_test.go

package workflow_test

import (
    "time"
    "go.uber.org/cadence/worker"
)

func (s *UnitTestSuite) Test_TWorkflow() {
    env := s.NewTestWorkflowEnvironment()

    worker := workflow.WorkflowImpl{
         Worker: ...
         Client: ...
    }

    s.worker = &worker

    env.RegisterActivity(s.worker.TActivity)

    // Delay second TWorkflow.
    env.RegisterDelayedCallback(func() {
        env.ExecuteWorkflow(s.worker.TWorkflow, "Approved")
    }, time.Second*2)

    env.ExecuteWorkflow(s.worker.TWorkflow, "Noop")
    s.True(env.IsWorkflowCompleted())
    s.NoError(env.GetWorkflowError())
}

上面的代码并不完整，它没有保存回调令牌并调用completeactivity。为了测试序列的目的，我只希望看到工作流启动的日志和活动启动两次，但我没有看到这一点。第一个工作流程启动后，如果没有任何活动的日志，测试就会挂起直到超时。

缺少什么或者是否可以像这样执行相同的工作流程两次？

正确答案

env.registerdelayedcallback(func() {
    env.executeworkflow(s.worker.tworkflow, "approved")
}, time.second*2)

这里出现了僵局。 env 在回调运行时被锁定（请参阅 源代码）。并且回调想要在同一个 env 上执行工作流，这需要在 env 上获取相同的锁（参见 源代码）。

让我们尝试通过在新的 goroutine 中运行回调来打破僵局：

env.registerdelayedcallback(func() {
    go env.executeworkflow(s.worker.tworkflow, "approved")
}, time.second*2)

现在我们感到恐慌：

panic: Current TestWorkflowEnvironment is used to execute s.worker.TWorkflow. Please create a new TestWorkflowEnvironment for s.worker.TWorkflow.

目前，testworkflowenvironment 无法运行 2 个非父子工作流程。请参阅跟踪任务的问题以使 testworkflowenvironment 支持测试多个工作流程 .

正如恐慌消息所建议的那样，您必须创建一个新的 testworkflowenvironment 来执行另一个工作流程（但我不确定它是否适合您的用例）。