add expect method for the LMove and BLMove redis command go-redis#17

burgesQ · Oct 21, 2021 · 5124588 · 5124588
1 parent 85ddc25
commit 5124588
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/expect.go b/expect.go
@@ -110,6 +110,7 @@ type baseMock interface {
 	ExpectBLPop(timeout time.Duration, keys ...string) *ExpectedStringSlice
 	ExpectBRPop(timeout time.Duration, keys ...string) *ExpectedStringSlice
 	ExpectBRPopLPush(source, destination string, timeout time.Duration) *ExpectedString
+	ExpectBLMove(source, destination, srcpos, destpos string, timeout time.Duration) *ExpectedString
 	ExpectLIndex(key string, index int64) *ExpectedString
 	ExpectLInsert(key, op string, pivot, value interface{}) *ExpectedInt
 	ExpectLInsertBefore(key string, pivot, value interface{}) *ExpectedInt
@@ -126,6 +127,7 @@ type baseMock interface {
 	ExpectLTrim(key string, start, stop int64) *ExpectedStatus
 	ExpectRPop(key string) *ExpectedString
 	ExpectRPopLPush(source, destination string) *ExpectedString
+	ExpectLMove(source, destination, srcpos, destpos string) *ExpectedString
 	ExpectRPush(key string, values ...interface{}) *ExpectedInt
 	ExpectRPushX(key string, values ...interface{}) *ExpectedInt
 

diff --git a/mock.go b/mock.go
@@ -961,6 +961,13 @@ func (m *mock) ExpectBRPopLPush(source, destination string, timeout time.Duratio
 	return e
 }
 
+func (m *mock) ExpectBLMove(source, destination, srcpos, dstpos string, timeout time.Duration) *ExpectedString {
+	e := &ExpectedString{}
+	e.cmd = m.factory.BLMove(m.ctx, source, destination, srcpos, dstpos, timeout)
+	m.pushExpect(e)
+	return e
+}
+
 func (m *mock) ExpectLIndex(key string, index int64) *ExpectedString {
 	e := &ExpectedString{}
 	e.cmd = m.factory.LIndex(m.ctx, key, index)
@@ -1073,6 +1080,13 @@ func (m *mock) ExpectRPopLPush(source, destination string) *ExpectedString {
 	return e
 }
 
+func (m *mock) ExpectLMove(source, destination, srcpos, destpos string) *ExpectedString {
+	e := &ExpectedString{}
+	e.cmd = m.factory.LMove(m.ctx, source, destination, srcpos, destpos)
+	m.pushExpect(e)
+	return e
+}
+
 func (m *mock) ExpectRPush(key string, values ...interface{}) *ExpectedInt {
 	e := &ExpectedInt{}
 	e.cmd = m.factory.RPush(m.ctx, key, values...)