Merge pull request #62 from monkey92t/new_cmd

feat: add ZAddLT,ZAddGT command

Author: monkey92t

commands_test.go

@@ -1675,6 +1675,34 @@ var _ = Describe("Commands", func() {
 			})
 		})
 
+		It("ZAddLT", func() {
+			operationIntCmd(clientMock, func() *ExpectedInt {
+				return clientMock.ExpectZAddLT("zset", redis.Z{
+					Member: "a",
+					Score:  1,
+				})
+			}, func() *redis.IntCmd {
+				return client.ZAddLT(ctx, "zset", redis.Z{
+					Member: "a",
+					Score:  1,
+				})
+			})
+		})
+
+		It("ZAddGT", func() {
+			operationIntCmd(clientMock, func() *ExpectedInt {
+				return clientMock.ExpectZAddGT("zset", redis.Z{
+					Member: "a",
+					Score:  1,
+				})
+			}, func() *redis.IntCmd {
+				return client.ZAddGT(ctx, "zset", redis.Z{
+					Member: "a",
+					Score:  1,
+				})
+			})
+		})
+
 		It("ZAddNX", func() {
 			operationIntCmd(clientMock, func() *ExpectedInt {
 				return clientMock.ExpectZAddNX("zset", redis.Z{

expect.go

@@ -207,6 +207,8 @@ type baseMock interface {
 	ExpectBZMPop(timeout time.Duration, order string, count int64, keys ...string) *ExpectedZSliceWithKey
 
 	ExpectZAdd(key string, members ...redis.Z) *ExpectedInt
+	ExpectZAddLT(key string, members ...redis.Z) *ExpectedInt
+	ExpectZAddGT(key string, members ...redis.Z) *ExpectedInt
 	ExpectZAddNX(key string, members ...redis.Z) *ExpectedInt
 	ExpectZAddXX(key string, members ...redis.Z) *ExpectedInt
 	ExpectZAddArgs(key string, args redis.ZAddArgs) *ExpectedInt

go.mod

@@ -5,7 +5,7 @@ go 1.18
 require (
 	github.com/onsi/ginkgo v1.16.5
 	github.com/onsi/gomega v1.25.0
-	github.com/redis/go-redis/v9 v9.0.3-0.20230223112541-6045d2e4561c
+	github.com/redis/go-redis/v9 v9.0.3-0.20230304070314-9664fd235954
 )
 
 require (

go.sum

@@ -3,7 +3,6 @@ github.com/bsm/gomega v1.26.0 h1:LhQm+AFcgV2M0WyKroMASzAzCAJVpAxQXv4SaI9a69Y=
 github.com/cespare/xxhash/v2 v2.2.0 h1:DC2CZ1Ep5Y4k3ZQ899DldepgrayRUGE6BBZ/cd9Cj44=
 github.com/cespare/xxhash/v2 v2.2.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs=
 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
-github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f h1:lO4WD4F/rVNCu3HqELle0jiPLLBs70cWOduZpkS1E78=
 github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f/go.mod h1:cuUVRXasLTGF7a8hSLbxyZXjz+1KgoB3wDUb6vlszIc=
@@ -38,13 +37,11 @@ github.com/onsi/gomega v1.7.1/go.mod h1:XdKZgCCFLUoM/7CFJVPcG8C1xQ1AJ0vpAezJrB7J
 github.com/onsi/gomega v1.10.1/go.mod h1:iN09h71vgCQne3DLsj+A5owkum+a2tYe+TOCB1ybHNo=
 github.com/onsi/gomega v1.25.0 h1:Vw7br2PCDYijJHSfBOWhov+8cAnUf8MfMaIOV323l6Y=
 github.com/onsi/gomega v1.25.0/go.mod h1:r+zV744Re+DiYCIPRlYOTxn0YkOLcAnW8k1xXdMPGhM=
-github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
 github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
-github.com/redis/go-redis/v9 v9.0.3-0.20230223112541-6045d2e4561c h1:HU8TAwucfDaYERowWS2hTe3GyMx46wOdg5iNnhZzZuY=
-github.com/redis/go-redis/v9 v9.0.3-0.20230223112541-6045d2e4561c/go.mod h1:6kjiKPy72LcR7xJX4bMLuf+2Rd/tpBFk4YKJTZ1jTNA=
+github.com/redis/go-redis/v9 v9.0.3-0.20230304070314-9664fd235954 h1:my7nLlAOlob02/9yr4GR0pStBXgQ3yGcaS/f5MxI24k=
+github.com/redis/go-redis/v9 v9.0.3-0.20230304070314-9664fd235954/go.mod h1:WqMKv5vnQbRuZstUwxQI195wHy+t4PuXDOjzMvcuQHk=
 github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
 github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA=
-github.com/stretchr/testify v1.8.1 h1:w7B6lhMri9wdJUVmEZPGGhZzrYTPvgJArz7wNPgYKsk=
 github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
 golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
 golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=

mock.go

@@ -1653,6 +1653,20 @@ func (m *mock) ExpectZAdd(key string, members ...redis.Z) *ExpectedInt {
 	return e
 }
 
+func (m *mock) ExpectZAddLT(key string, members ...redis.Z) *ExpectedInt {
+	e := &ExpectedInt{}
+	e.cmd = m.factory.ZAddLT(m.ctx, key, members...)
+	m.pushExpect(e)
+	return e
+}
+
+func (m *mock) ExpectZAddGT(key string, members ...redis.Z) *ExpectedInt {
+	e := &ExpectedInt{}
+	e.cmd = m.factory.ZAddGT(m.ctx, key, members...)
+	m.pushExpect(e)
+	return e
+}
+
 func (m *mock) ExpectZAddNX(key string, members ...redis.Z) *ExpectedInt {
 	e := &ExpectedInt{}
 	e.cmd = m.factory.ZAddNX(m.ctx, key, members...)