Issue 22 - Added ExpectScanType to mock

client_test.go

@@ -922,6 +922,14 @@ var _ = Describe("RedisMock", func() {
 			})
 		})
 
+		It("ScanType", func() {
+			operationScanCmd(clientMock, func() *ExpectedScan {
+				return clientMock.ExpectScanType(0, "match", 2, "type")
+			}, func() *redis.ScanCmd {
+				return client.ScanType(ctx, 0, "match", 2, "type")
+			})
+		})
+
 		It("SScan", func() {
 			operationScanCmd(clientMock, func() *ExpectedScan {
 				return clientMock.ExpectSScan("key", 1, "match", 2)

expect.go

@@ -2,12 +2,11 @@ package redismock
 
 import (
 	"fmt"
+	"github.com/go-redis/redis/v8"
 	"reflect"
 	"sync"
 	"time"
 	"unsafe"
-
-	"github.com/go-redis/redis/v8"
 )
 
 type baseMock interface {
@@ -89,6 +88,7 @@ type baseMock interface {
 	ExpectBitField(key string, args ...interface{}) *ExpectedIntSlice
 
 	ExpectScan(cursor uint64, match string, count int64) *ExpectedScan
+	ExpectScanType(cursor uint64, match string, count int64, keyType string) *ExpectedScan
 	ExpectSScan(key string, cursor uint64, match string, count int64) *ExpectedScan
 	ExpectHScan(key string, cursor uint64, match string, count int64) *ExpectedScan
 	ExpectZScan(key string, cursor uint64, match string, count int64) *ExpectedScan

mock.go

@@ -4,12 +4,11 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"github.com/go-redis/redis/v8"
 	"reflect"
 	"regexp"
 	"strings"
 	"time"
-
-	"github.com/go-redis/redis/v8"
 )
 
 type mock struct {
@@ -828,6 +827,13 @@ func (m *mock) ExpectScan(cursor uint64, match string, count int64) *ExpectedSca
 	return e
 }
 
+func (m *mock) ExpectScanType(cursor uint64, match string, count int64, keyType string) *ExpectedScan {
+	e := &ExpectedScan{}
+	e.cmd = m.factory.ScanType(m.ctx, cursor, match, count, keyType)
+	m.pushExpect(e)
+	return e
+}
+
 func (m *mock) ExpectSScan(key string, cursor uint64, match string, count int64) *ExpectedScan {
 	e := &ExpectedScan{}
 	e.cmd = m.factory.SScan(m.ctx, key, cursor, match, count)