diff --git a/pubsub/message/execution/context_test.go b/pubsub/message/execution/context_test.go index dcb341d..38ae346 100644 --- a/pubsub/message/execution/context_test.go +++ b/pubsub/message/execution/context_test.go @@ -45,14 +45,15 @@ func TestMessageExecutionCtx_Send(t *testing.T) { receivedMessage := message.NewReceivedMessage("123", &someTestType{}, message.Headers{}, time.Now(), "bus") outcomingMsg := message.NewOutcomingMessage(&someTestType{}) + ctx := context.Background() testRouter. EXPECT(). Route(outcomingMsg.Payload()). Return(nil). - Times(1) + Times(1 - execCtx := factory.CreateCtx(context.Background(), receivedMessage) + execCtx := factory.CreateCtx(ctx, receivedMessage) err := execCtx.Send(outcomingMsg) assert.NoError(t, err)