forked from mirror/redis
137 lines
3.2 KiB
Go
137 lines
3.2 KiB
Go
package redis_test
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
|
|
"git.internal/re/redis/v8"
|
|
)
|
|
|
|
var _ = Describe("ScanIterator", func() {
|
|
var client *redis.Client
|
|
|
|
seed := func(n int) error {
|
|
pipe := client.Pipeline()
|
|
for i := 1; i <= n; i++ {
|
|
pipe.Set(ctx, fmt.Sprintf("K%02d", i), "x", 0).Err()
|
|
}
|
|
_, err := pipe.Exec(ctx)
|
|
return err
|
|
}
|
|
|
|
extraSeed := func(n int, m int) error {
|
|
pipe := client.Pipeline()
|
|
for i := 1; i <= m; i++ {
|
|
pipe.Set(ctx, fmt.Sprintf("A%02d", i), "x", 0).Err()
|
|
}
|
|
for i := 1; i <= n; i++ {
|
|
pipe.Set(ctx, fmt.Sprintf("K%02d", i), "x", 0).Err()
|
|
}
|
|
_, err := pipe.Exec(ctx)
|
|
return err
|
|
}
|
|
|
|
hashKey := "K_HASHTEST"
|
|
hashSeed := func(n int) error {
|
|
pipe := client.Pipeline()
|
|
for i := 1; i <= n; i++ {
|
|
pipe.HSet(ctx, hashKey, fmt.Sprintf("K%02d", i), "x").Err()
|
|
}
|
|
_, err := pipe.Exec(ctx)
|
|
return err
|
|
}
|
|
|
|
BeforeEach(func() {
|
|
client = redis.NewClient(redisOptions())
|
|
Expect(client.FlushDB(ctx).Err()).NotTo(HaveOccurred())
|
|
})
|
|
|
|
AfterEach(func() {
|
|
Expect(client.Close()).NotTo(HaveOccurred())
|
|
})
|
|
|
|
It("should scan across empty DBs", func() {
|
|
iter := client.Scan(ctx, 0, "", 10).Iterator()
|
|
Expect(iter.Next(ctx)).To(BeFalse())
|
|
Expect(iter.Err()).NotTo(HaveOccurred())
|
|
})
|
|
|
|
It("should scan across one page", func() {
|
|
Expect(seed(7)).NotTo(HaveOccurred())
|
|
|
|
var vals []string
|
|
iter := client.Scan(ctx, 0, "", 0).Iterator()
|
|
for iter.Next(ctx) {
|
|
vals = append(vals, iter.Val())
|
|
}
|
|
Expect(iter.Err()).NotTo(HaveOccurred())
|
|
Expect(vals).To(ConsistOf([]string{"K01", "K02", "K03", "K04", "K05", "K06", "K07"}))
|
|
})
|
|
|
|
It("should scan across multiple pages", func() {
|
|
Expect(seed(71)).NotTo(HaveOccurred())
|
|
|
|
var vals []string
|
|
iter := client.Scan(ctx, 0, "", 10).Iterator()
|
|
for iter.Next(ctx) {
|
|
vals = append(vals, iter.Val())
|
|
}
|
|
Expect(iter.Err()).NotTo(HaveOccurred())
|
|
Expect(vals).To(HaveLen(71))
|
|
Expect(vals).To(ContainElement("K01"))
|
|
Expect(vals).To(ContainElement("K71"))
|
|
})
|
|
|
|
It("should hscan across multiple pages", func() {
|
|
Expect(hashSeed(71)).NotTo(HaveOccurred())
|
|
|
|
var vals []string
|
|
iter := client.HScan(ctx, hashKey, 0, "", 10).Iterator()
|
|
for iter.Next(ctx) {
|
|
vals = append(vals, iter.Val())
|
|
}
|
|
Expect(iter.Err()).NotTo(HaveOccurred())
|
|
Expect(vals).To(HaveLen(71 * 2))
|
|
Expect(vals).To(ContainElement("K01"))
|
|
Expect(vals).To(ContainElement("K71"))
|
|
})
|
|
|
|
It("should scan to page borders", func() {
|
|
Expect(seed(20)).NotTo(HaveOccurred())
|
|
|
|
var vals []string
|
|
iter := client.Scan(ctx, 0, "", 10).Iterator()
|
|
for iter.Next(ctx) {
|
|
vals = append(vals, iter.Val())
|
|
}
|
|
Expect(iter.Err()).NotTo(HaveOccurred())
|
|
Expect(vals).To(HaveLen(20))
|
|
})
|
|
|
|
It("should scan with match", func() {
|
|
Expect(seed(33)).NotTo(HaveOccurred())
|
|
|
|
var vals []string
|
|
iter := client.Scan(ctx, 0, "K*2*", 10).Iterator()
|
|
for iter.Next(ctx) {
|
|
vals = append(vals, iter.Val())
|
|
}
|
|
Expect(iter.Err()).NotTo(HaveOccurred())
|
|
Expect(vals).To(HaveLen(13))
|
|
})
|
|
|
|
It("should scan with match across empty pages", func() {
|
|
Expect(extraSeed(2, 10)).NotTo(HaveOccurred())
|
|
|
|
var vals []string
|
|
iter := client.Scan(ctx, 0, "K*", 1).Iterator()
|
|
for iter.Next(ctx) {
|
|
vals = append(vals, iter.Val())
|
|
}
|
|
Expect(iter.Err()).NotTo(HaveOccurred())
|
|
Expect(vals).To(HaveLen(2))
|
|
})
|
|
})
|