diff --git a/internal/internal_task_handlers_test.go b/internal/internal_task_handlers_test.go index 919b1e40d..d614c5113 100644 --- a/internal/internal_task_handlers_test.go +++ b/internal/internal_task_handlers_test.go @@ -946,12 +946,15 @@ func (t *TaskHandlersTestSuite) testSideEffectDeferHelper(cacheSize int) { workflowFunc := func(ctx Context) error { defer func() { + fmt.Println("deferred function called") if !IsReplaying(ctx) { // This is an side effect op value = "" } + fmt.Println("Channel closed") close(doneCh) }() + fmt.Println("Sleeping") _ = Sleep(ctx, 1*time.Second) return nil } @@ -976,10 +979,12 @@ func (t *TaskHandlersTestSuite) testSideEffectDeferHelper(cacheSize int) { wftask := workflowTask{task: task} wfctx := t.mustWorkflowContextImpl(&wftask, taskHandler) _, err := taskHandler.ProcessWorkflowTask(&wftask, wfctx, nil) + t.logger.Info("Unlocking") wfctx.Unlock(err) t.Nil(err) // Make sure the workflow coroutine has exited. + t.logger.Info("Waiting for channel to close") <-doneCh // The side effect op should not be executed. t.Equal(expectedValue, value)