diff options
Diffstat (limited to 'src/payload-processor/processor/processor_test.go')
| -rw-r--r-- | src/payload-processor/processor/processor_test.go | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/payload-processor/processor/processor_test.go b/src/payload-processor/processor/processor_test.go index 5e954c4..5fa34a8 100644 --- a/src/payload-processor/processor/processor_test.go +++ b/src/payload-processor/processor/processor_test.go @@ -18,9 +18,11 @@ func TestProcessMessage(t *testing.T) { message := []byte("test") - mockdb.EXPECT().TestInsert(gomock.Any(), string(message)).Return(nil) + ctx := context.Background() - err := processor.handleMessage(context.Background(), kafka.Message{Value: message}) + mockdb.EXPECT().TestInsert(ctx, string(message)).Return(nil) + + err := processor.handleMessage(ctx, kafka.Message{Value: message}) require.NoError(t, err) } |