add Do() commands to mock

Author: ideasculptor

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.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
 }