Enforce nexus request timeout in workflow test suite

internal/internal_nexus_task_handler.go

@@ -336,6 +336,7 @@ func (h *nexusTaskHandler) goContextForTask(nctx *NexusOperationContext, header
 		if err != nil {
 			return nil, nil, nexusHandlerError(nexus.HandlerErrorTypeBadRequest, "cannot parse request timeout")
 		}
+
 		ctx, cancel := context.WithTimeout(ctx, timeout)
 		return ctx, cancel, nil
 	}

internal/internal_workflow_testsuite.go

@@ -2358,6 +2358,17 @@ func (env *testWorkflowEnvironmentImpl) ExecuteNexusOperation(
 ) int64 {
 	seq := env.nextID()
 	taskHandler := env.newTestNexusTaskHandler()
+	// Use lower case header values to simulate how the Nexus SDK (used internally by the "real" server) would transmit
+	// these headers over the wire.
+	nexusHeader := make(map[string]string, len(params.nexusHeader))
+	for k, v := range params.nexusHeader {
+		nexusHeader[strings.ToLower(k)] = v
+	}
+	params.nexusHeader = nexusHeader
+	// The real server allows requests to take up to 10 seconds, mimic that behavior here.
+	// Note that if a user sets the Request-Timeout header, it gets overridden.
+	params.nexusHeader[strings.ToLower(nexus.HeaderRequestTimeout)] = "10s"
+
 	handle := &testNexusOperationHandle{
 		env:         env,
 		seq:         seq,

test/nexus_test.go

@@ -735,6 +735,10 @@ func TestReplay(t *testing.T) {
 
 func TestWorkflowTestSuite_NexusSyncOperation(t *testing.T) {
 	op := nexus.NewSyncOperation("op", func(ctx context.Context, outcome string, opts nexus.StartOperationOptions) (string, error) {
+		dealine, ok := ctx.Deadline()
+		require.True(t, ok)
+		timeout := time.Until(dealine)
+		require.GreaterOrEqual(t, 10*time.Second, timeout)
 		require.NotPanicsf(t, func() {
 			temporalnexus.GetMetricsHandler(ctx)
 			temporalnexus.GetLogger(ctx)