diff --git a/expect.go b/expect.go index f62e316..aa16fd5 100644 --- a/expect.go +++ b/expect.go @@ -28,6 +28,10 @@ type baseMock interface { MatchExpectationsInOrder(b bool) ExpectDo(args ...interface{}) *ExpectedCmd + ExpectIntSliceDo(args ...interface{}) *ExpectedIntSlice + ExpectIntDo(args ...interface{}) *ExpectedInt + ExpectStringSliceDo(args ...interface{}) *ExpectedStringSlice + ExpectStringDo(args ...interface{}) *ExpectedString ExpectCommand() *ExpectedCommandsInfo ExpectClientGetName() *ExpectedString ExpectEcho(message interface{}) *ExpectedString diff --git a/mock.go b/mock.go index d2162b8..7d7154d 100644 --- a/mock.go +++ b/mock.go @@ -447,7 +447,34 @@ func (m *mock) ExpectDo(args ...interface{}) *ExpectedCmd { default: panic("ExpectDo: unsupported client type") } + m.pushExpect(e) + return e +} + +func (m *mock) ExpectIntSliceDo(args ...interface{}) *ExpectedIntSlice { + e := &ExpectedIntSlice{} + e.cmd = redis.NewIntSliceCmd(m.ctx, args...) + m.pushExpect(e) + return e +} +func (m *mock) ExpectIntDo(args ...interface{}) *ExpectedInt { + e := &ExpectedInt{} + e.cmd = redis.NewIntCmd(m.ctx, args...) + m.pushExpect(e) + return e +} + +func (m *mock) ExpectStringDo(args ...interface{}) *ExpectedString { + e := &ExpectedString{} + e.cmd = redis.NewStringCmd(m.ctx, args...) + m.pushExpect(e) + return e +} + +func (m *mock) ExpectStringSliceDo(args ...interface{}) *ExpectedStringSlice { + e := &ExpectedStringSlice{} + e.cmd = redis.NewStringSliceCmd(m.ctx, args...) m.pushExpect(e) return e }