diff --git a/redis/mocks/mocks.go b/redis/mocks/mocks.go index b611294..fb23138 100644 --- a/redis/mocks/mocks.go +++ b/redis/mocks/mocks.go @@ -77,6 +77,11 @@ type MockPipeline struct { LLenCall func(key string) HIncrByCall func(key string, field string, value int64) HLenCall func(key string) + SetCall func(key string, value interface{}, expiration time.Duration) + IncrCall func(key string) + DecrCall func(key string) + SAddCall func(key string, members ...interface{}) + SRemCall func(key string, members ...interface{}) ExecCall func() ([]redis.Result, error) } @@ -100,6 +105,26 @@ func (m *MockPipeline) HLen(key string) { m.HLenCall(key) } +func (m *MockPipeline) Set(key string, value interface{}, expiration time.Duration) { + m.SetCall(key, value, expiration) +} + +func (m *MockPipeline) Incr(key string) { + m.IncrCall(key) +} + +func (m *MockPipeline) Decr(key string) { + m.DecrCall(key) +} + +func (m *MockPipeline) SAdd(key string, members ...interface{}) { + m.SAddCall(key, members...) +} + +func (m *MockPipeline) SRem(key string, members ...interface{}) { + m.SRemCall(key, members...) +} + func (m *MockPipeline) Exec() ([]redis.Result, error) { return m.ExecCall() }