diff --git a/mock.go b/mock.go index c8c28b9..132d999 100644 --- a/mock.go +++ b/mock.go @@ -89,17 +89,18 @@ func (redisClientHook) DialHook(hook redis.DialHook) redis.DialHook { return hook } -func (h redisClientHook) ProcessHook(_ redis.ProcessHook) redis.ProcessHook { +func (h redisClientHook) ProcessHook(hook redis.ProcessHook) redis.ProcessHook { return func(ctx context.Context, cmd redis.Cmder) error { err := h.fn(cmd) if h.returnErr != nil && (err == nil || cmd.Err() == nil) { - err = h.returnErr + return h.returnErr } - return err + + return hook(ctx, cmd) } } -func (h redisClientHook) ProcessPipelineHook(_ redis.ProcessPipelineHook) redis.ProcessPipelineHook { +func (h redisClientHook) ProcessPipelineHook(hook redis.ProcessPipelineHook) redis.ProcessPipelineHook { return func(ctx context.Context, cmds []redis.Cmder) error { for _, cmd := range cmds { err := h.fn(cmd) @@ -110,7 +111,8 @@ func (h redisClientHook) ProcessPipelineHook(_ redis.ProcessPipelineHook) redis. return err } } - return nil + + return hook(ctx, cmds) } }