How to write unit test for CompleteActivityByID?

[3tbraden]

In the testsuite there's a CompleteActivity method from the test env
https://github.com/uber-go/cadence-client/blob/1b4849be48832fedd78dc7d29510eded822a8e7c/internal/workflow_testsuite.go#L604

which I can use to test behaviour about pending and asynchronously completing activities, e.g.

(s *testsuite.WorkflowTestSuite) TestCompleteActivity() {
  env := s.NewTestWorkflowEnvironment()

  token := []byte{}
  env.RegisterActivity(s.worker.RegisterTokenThenReturnPending)

  env.OnActivity(s.worker.RegisterTokenThenReturnPending, mock.Anything, mock.MatchedBy(t []byte) bool {
    token = t
    return true
  }).Return(nil).Once()

  env.RegisterDelayedCallback(func() {
    err := env.CompleteActivity(token, result, nil)
    s.NoError(err)
  })

  env.ExecuteWorkflow(s.worker.WorkflowThatCallsRegisterTokenThenReturnPending)

  s.True(env.IsWorkflowCompleted())
}

However recently I had to switch to CompleteActivityByID due to the fact that the activity token will get updated when the activity is rescheduled(because of scheduling timeout), and hence it would fail to find the activity when calling CompleteActivity with an outdated token later. While CompleteActivityByID relies only on workflowID and activityID and has no problem finding a rescheduled activity.

But there's no CompleteActivityByID in the testsuite environment. Is it possible to write a unit test similar to the above?