apps/redka
1
0
Fork 0
mirror of https://github.com/nalgeon/redka.git synced 2025-11-01 03:42:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

postgres backend (#48)

Browse Source
This commit is contained in:
Anton Zhiyanov
2025-07-13 13:05:18 +05:00
committed by GitHub
parent 7c532df931
commit 2178c81fca
179 changed files with 8084 additions and 7408 deletions
Download Patch File Download Diff File Expand all files Collapse all files

158
internal/command/set/sdiffstore_test.go
View File

@@ -3,9 +3,9 @@ package set
import (
"testing"
"github.com/nalgeon/be"
"github.com/nalgeon/redka/internal/core"
"github.com/nalgeon/redka/internal/redis"
"github.com/nalgeon/redka/internal/testx"
)
func TestSDiffStoreParse(t *testing.T) {
@@ -39,12 +39,12 @@ func TestSDiffStoreParse(t *testing.T) {
for _, test := range tests {
t.Run(test.cmd, func(t *testing.T) {
cmd, err := redis.Parse(ParseSDiffStore, test.cmd)
testx.AssertEqual(t, err, test.err)
be.Equal(t, err, test.err)
if err == nil {
testx.AssertEqual(t, cmd.dest, test.want.dest)
testx.AssertEqual(t, cmd.keys, test.want.keys)
be.Equal(t, cmd.dest, test.want.dest)
be.Equal(t, cmd.keys, test.want.keys)
} else {
testx.AssertEqual(t, cmd, test.want)
be.Equal(t, cmd, test.want)
}
})
}
@@ -52,139 +52,131 @@ func TestSDiffStoreParse(t *testing.T) {
func TestSDiffStoreExec(t *testing.T) {
t.Run("store", func(t *testing.T) {
db, red := getDB(t)
defer db.Close()
_, _ = db.Set().Add("key1", "one", "two", "thr", "fiv")
_, _ = db.Set().Add("key2", "two", "fou", "six")
_, _ = db.Set().Add("key3", "thr", "six")
red := getRedka(t)
_, _ = red.Set().Add("key1", "one", "two", "thr", "fiv")
_, _ = red.Set().Add("key2", "two", "fou", "six")
_, _ = red.Set().Add("key3", "thr", "six")
cmd := redis.MustParse(ParseSDiffStore, "sdiffstore dest key1 key2 key3")
conn := redis.NewFakeConn()
res, err := cmd.Run(conn, red)
testx.AssertNoErr(t, err)
testx.AssertEqual(t, res, 2)
testx.AssertEqual(t, conn.Out(), "2")
be.Err(t, err, nil)
be.Equal(t, res, 2)
be.Equal(t, conn.Out(), "2")
items, _ := db.Set().Items("dest")
items, _ := red.Set().Items("dest")
sortValues(items)
testx.AssertEqual(t, items, []core.Value{core.Value("fiv"), core.Value("one")})
be.Equal(t, items, []core.Value{core.Value("fiv"), core.Value("one")})
})
t.Run("rewrite dest", func(t *testing.T) {
db, red := getDB(t)
defer db.Close()
_, _ = db.Set().Add("key1", "one")
_, _ = db.Set().Add("key2", "two")
_, _ = db.Set().Add("dest", "old")
red := getRedka(t)
_, _ = red.Set().Add("key1", "one")
_, _ = red.Set().Add("key2", "two")
_, _ = red.Set().Add("dest", "old")
cmd := redis.MustParse(ParseSDiffStore, "sdiffstore dest key1 key2")
conn := redis.NewFakeConn()
res, err := cmd.Run(conn, red)
testx.AssertNoErr(t, err)
testx.AssertEqual(t, res, 1)
testx.AssertEqual(t, conn.Out(), "1")
be.Err(t, err, nil)
be.Equal(t, res, 1)
be.Equal(t, conn.Out(), "1")
items, _ := db.Set().Items("dest")
testx.AssertEqual(t, items, []core.Value{core.Value("one")})
items, _ := red.Set().Items("dest")
be.Equal(t, items, []core.Value{core.Value("one")})
})
t.Run("single key", func(t *testing.T) {
db, red := getDB(t)
defer db.Close()
_, _ = db.Set().Add("key1", "one", "two")
_, _ = db.Set().Add("dest", "old")
red := getRedka(t)
_, _ = red.Set().Add("key1", "one", "two")
_, _ = red.Set().Add("dest", "old")
cmd := redis.MustParse(ParseSDiffStore, "sdiffstore dest key1")
conn := redis.NewFakeConn()
res, err := cmd.Run(conn, red)
testx.AssertNoErr(t, err)
testx.AssertEqual(t, res, 2)
testx.AssertEqual(t, conn.Out(), "2")
be.Err(t, err, nil)
be.Equal(t, res, 2)
be.Equal(t, conn.Out(), "2")
items, _ := db.Set().Items("dest")
items, _ := red.Set().Items("dest")
sortValues(items)
testx.AssertEqual(t, items, []core.Value{core.Value("one"), core.Value("two")})
be.Equal(t, items, []core.Value{core.Value("one"), core.Value("two")})
})
t.Run("empty", func(t *testing.T) {
db, red := getDB(t)
defer db.Close()
_, _ = db.Set().Add("key1", "one")
_, _ = db.Set().Add("key2", "one")
_, _ = db.Set().Add("dest", "old")
red := getRedka(t)
_, _ = red.Set().Add("key1", "one")
_, _ = red.Set().Add("key2", "one")
_, _ = red.Set().Add("dest", "old")
cmd := redis.MustParse(ParseSDiffStore, "sdiffstore dest key1 key2")
conn := redis.NewFakeConn()
res, err := cmd.Run(conn, red)
testx.AssertNoErr(t, err)
testx.AssertEqual(t, res, 0)
testx.AssertEqual(t, conn.Out(), "0")
be.Err(t, err, nil)
be.Equal(t, res, 0)
be.Equal(t, conn.Out(), "0")
items, _ := db.Set().Items("dest")
testx.AssertEqual(t, items, []core.Value(nil))
items, _ := red.Set().Items("dest")
be.Equal(t, items, []core.Value(nil))
})
t.Run("first not found", func(t *testing.T) {
db, red := getDB(t)
defer db.Close()
_, _ = db.Set().Add("key2", "two")
_, _ = db.Set().Add("key3", "thr")
_, _ = db.Set().Add("dest", "old")
red := getRedka(t)
_, _ = red.Set().Add("key2", "two")
_, _ = red.Set().Add("key3", "thr")
_, _ = red.Set().Add("dest", "old")
cmd := redis.MustParse(ParseSDiffStore, "sdiffstore dest key1 key2 key3")
conn := redis.NewFakeConn()
res, err := cmd.Run(conn, red)
testx.AssertNoErr(t, err)
testx.AssertEqual(t, res, 0)
testx.AssertEqual(t, conn.Out(), "0")
be.Err(t, err, nil)
be.Equal(t, res, 0)
be.Equal(t, conn.Out(), "0")
items, _ := db.Set().Items("dest")
testx.AssertEqual(t, items, []core.Value(nil))
items, _ := red.Set().Items("dest")
be.Equal(t, items, []core.Value(nil))
})
t.Run("rest not found", func(t *testing.T) {
db, red := getDB(t)
defer db.Close()
_, _ = db.Set().Add("key1", "one")
_, _ = db.Set().Add("key2", "two")
red := getRedka(t)
_, _ = red.Set().Add("key1", "one")
_, _ = red.Set().Add("key2", "two")
cmd := redis.MustParse(ParseSDiffStore, "sdiffstore dest key1 key2 key3")
conn := redis.NewFakeConn()
res, err := cmd.Run(conn, red)
testx.AssertNoErr(t, err)
testx.AssertEqual(t, res, 1)
testx.AssertEqual(t, conn.Out(), "1")
be.Err(t, err, nil)
be.Equal(t, res, 1)
be.Equal(t, conn.Out(), "1")
items, _ := db.Set().Items("dest")
testx.AssertEqual(t, items, []core.Value{core.Value("one")})
items, _ := red.Set().Items("dest")
be.Equal(t, items, []core.Value{core.Value("one")})
})
t.Run("source key type mismatch", func(t *testing.T) {
db, red := getDB(t)
defer db.Close()
_, _ = db.Set().Add("key1", "one")
_, _ = db.Set().Add("key2", "two")
_ = db.Str().Set("key3", "thr")
red := getRedka(t)
_, _ = red.Set().Add("key1", "one")
_, _ = red.Set().Add("key2", "two")
_ = red.Str().Set("key3", "thr")
cmd := redis.MustParse(ParseSDiffStore, "sdiffstore dest key1 key2 key3")
conn := redis.NewFakeConn()
res, err := cmd.Run(conn, red)
testx.AssertNoErr(t, err)
testx.AssertEqual(t, res, 1)
testx.AssertEqual(t, conn.Out(), "1")
be.Err(t, err, nil)
be.Equal(t, res, 1)
be.Equal(t, conn.Out(), "1")
items, _ := db.Set().Items("dest")
testx.AssertEqual(t, items, []core.Value{core.Value("one")})
items, _ := red.Set().Items("dest")
be.Equal(t, items, []core.Value{core.Value("one")})
})
t.Run("dest key type mismatch", func(t *testing.T) {
db, red := getDB(t)
defer db.Close()
_, _ = db.Set().Add("key1", "one")
_, _ = db.Set().Add("key2", "two")
_ = db.Str().Set("dest", "old")
red := getRedka(t)
_, _ = red.Set().Add("key1", "one")
_, _ = red.Set().Add("key2", "two")
_ = red.Str().Set("dest", "old")
cmd := redis.MustParse(ParseSDiffStore, "sdiffstore dest key1 key2")
conn := redis.NewFakeConn()
res, err := cmd.Run(conn, red)
testx.AssertErr(t, err, core.ErrKeyType)
testx.AssertEqual(t, res, nil)
testx.AssertEqual(t, conn.Out(), core.ErrKeyType.Error()+" (sdiffstore)")
be.Err(t, err, core.ErrKeyType)
be.Equal(t, res, nil)
be.Equal(t, conn.Out(), core.ErrKeyType.Error()+" (sdiffstore)")
sval, _ := db.Str().Get("dest")
testx.AssertEqual(t, sval, core.Value("old"))
sval, _ := red.Str().Get("dest")
be.Equal(t, sval, core.Value("old"))
})
}