Revert "Use a more natural callback signature."

This reverts commit 7433da4.
prebid · Nov 6, 2024 · 9be4acd · 9be4acd
1 parent 7433da4
commit 9be4acd
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/endpoints/openrtb2/auction_test.go b/endpoints/openrtb2/auction_test.go
@@ -5993,10 +5993,11 @@ func TestRequestContextPropagates(t *testing.T) {
 	reqBody := validRequest(t, "site.json")
 	mockEntrypoint := mocks.NewEntrypoint(t)
 	mockEntrypoint.On("HandleEntrypointHook", mock.Anything, mock.Anything, mock.Anything).
-		Return(func(ctx context.Context, _ hookstage.ModuleInvocationContext, _ hookstage.EntrypointPayload) (hookstage.HookResult[hookstage.EntrypointPayload], error) {
+		Run(func(args mock.Arguments) {
+			ctx := args.Get(0).(context.Context)
 			assert.Equal(t, "testing", ctx.Value(testKey))
-			return hookstage.HookResult[hookstage.EntrypointPayload]{}, nil
-		})
+		}).
+		Return(hookstage.HookResult[hookstage.EntrypointPayload]{}, nil)
 	deps := &endpointDeps{
 		fakeUUIDGenerator{},
 		&mockExchange{},