add Do() commands to mock

Author: ideasculptor

mock.go

@@ -408,6 +408,34 @@ func (m *mock) ExpectWatch(keys ...string) *ExpectedError {
 	return e
 }
 
+func (m *mock) ExpectIntSliceDo(args ...interface{}) *ExpectedIntSlice {
+	e := &ExpectedIntSlice{}
+	e.cmd = redis.NewCmd(m.ctx, args)
+	m.pushExpect(e)
+	return e
+}
+
+func (m *mock) ExpectIntDo(args ...interface{}) *ExpectedInt {
+	e := &ExpectedInt{}
+	e.cmd = redis.NewCmd(m.ctx, args)
+	m.pushExpect(e)
+	return e
+}
+
+func (m *mock) ExpectStringDo(args ...interface{}) *ExpectedString {
+	e := &ExpectedString{}
+	e.cmd = redis.NewCmd(m.ctx, args)
+	m.pushExpect(e)
+	return e
+}
+
+func (m *mock) ExpectStringSliceDo(args ...interface{}) *ExpectedStringSlice {
+	e := &ExpectedStringSlice{}
+	e.cmd = redis.NewCmd(m.ctx, args)
+	m.pushExpect(e)
+	return e
+}
+
 func (m *mock) ExpectCommand() *ExpectedCommandsInfo {
 	e := &ExpectedCommandsInfo{}
 	e.cmd = m.factory.Command(m.ctx)