postgres backend (#48)

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

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 TestSDiffParse(t *testing.T) {
@@ -34,11 +34,11 @@ func TestSDiffParse(t *testing.T) {
for _, test := range tests {
t.Run(test.cmd, func(t *testing.T) {
cmd, err := redis.Parse(ParseSDiff, test.cmd)
testx.AssertEqual(t, err, test.err)
be.Equal(t, err, test.err)
if err == nil {
testx.AssertEqual(t, cmd.keys, test.want.keys)
be.Equal(t, cmd.keys, test.want.keys)
} else {
testx.AssertEqual(t, cmd, test.want)
be.Equal(t, cmd, test.want)
}
})
}
@@ -46,105 +46,97 @@ func TestSDiffParse(t *testing.T) {
func TestSDiffExec(t *testing.T) {
t.Run("non-empty", 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(ParseSDiff, "sdiff key1 key2 key3")
conn := redis.NewFakeConn()
res, err := cmd.Run(conn, red)
testx.AssertNoErr(t, err)
testx.AssertEqual(t, len(res.([]core.Value)), 2)
testx.AssertEqual(t, conn.Out(), "2,fiv,one")
be.Err(t, err, nil)
be.Equal(t, len(res.([]core.Value)), 2)
be.Equal(t, conn.Out(), "2,fiv,one", "2,one,fiv")
})
t.Run("no keys", func(t *testing.T) {
db, red := getDB(t)
defer db.Close()
red := getRedka(t)
cmd := redis.MustParse(ParseSDiff, "sdiff key1")
conn := redis.NewFakeConn()
res, err := cmd.Run(conn, red)
testx.AssertNoErr(t, err)
testx.AssertEqual(t, len(res.([]core.Value)), 0)
testx.AssertEqual(t, conn.Out(), "0")
be.Err(t, err, nil)
be.Equal(t, len(res.([]core.Value)), 0)
be.Equal(t, conn.Out(), "0")
})
t.Run("single key", func(t *testing.T) {
db, red := getDB(t)
defer db.Close()
_, _ = db.Set().Add("key1", "one", "two", "thr")
red := getRedka(t)
_, _ = red.Set().Add("key1", "one", "two", "thr")
cmd := redis.MustParse(ParseSDiff, "sdiff key1")
conn := redis.NewFakeConn()
res, err := cmd.Run(conn, red)
testx.AssertNoErr(t, err)
testx.AssertEqual(t, len(res.([]core.Value)), 3)
testx.AssertEqual(t, conn.Out(), "3,one,thr,two")
be.Err(t, err, nil)
be.Equal(t, len(res.([]core.Value)), 3)
be.Equal(t, conn.Out(), "3,one,thr,two", "3,one,two,thr")
})
t.Run("empty", func(t *testing.T) {
db, red := getDB(t)
defer db.Close()
_, _ = db.Set().Add("key1", "one", "two")
_, _ = db.Set().Add("key2", "one", "fou")
_, _ = db.Set().Add("key3", "two", "fiv")
red := getRedka(t)
_, _ = red.Set().Add("key1", "one", "two")
_, _ = red.Set().Add("key2", "one", "fou")
_, _ = red.Set().Add("key3", "two", "fiv")
cmd := redis.MustParse(ParseSDiff, "sdiff key1 key2 key3")
conn := redis.NewFakeConn()
res, err := cmd.Run(conn, red)
testx.AssertNoErr(t, err)
testx.AssertEqual(t, len(res.([]core.Value)), 0)
testx.AssertEqual(t, conn.Out(), "0")
be.Err(t, err, nil)
be.Equal(t, len(res.([]core.Value)), 0)
be.Equal(t, conn.Out(), "0")
})
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")
red := getRedka(t)
_, _ = red.Set().Add("key2", "two")
_, _ = red.Set().Add("key3", "thr")
cmd := redis.MustParse(ParseSDiff, "sdiff key1 key2 key3")
conn := redis.NewFakeConn()
res, err := cmd.Run(conn, red)
testx.AssertNoErr(t, err)
testx.AssertEqual(t, len(res.([]core.Value)), 0)
testx.AssertEqual(t, conn.Out(), "0")
be.Err(t, err, nil)
be.Equal(t, len(res.([]core.Value)), 0)
be.Equal(t, conn.Out(), "0")
})
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(ParseSDiff, "sdiff key1 key2 key3")
conn := redis.NewFakeConn()
res, err := cmd.Run(conn, red)
testx.AssertNoErr(t, err)
testx.AssertEqual(t, len(res.([]core.Value)), 1)
testx.AssertEqual(t, conn.Out(), "1,one")
be.Err(t, err, nil)
be.Equal(t, len(res.([]core.Value)), 1)
be.Equal(t, conn.Out(), "1,one")
})
t.Run("all not found", func(t *testing.T) {
db, red := getDB(t)
defer db.Close()
red := getRedka(t)
cmd := redis.MustParse(ParseSDiff, "sdiff key1 key2 key3")
conn := redis.NewFakeConn()
res, err := cmd.Run(conn, red)
testx.AssertNoErr(t, err)
testx.AssertEqual(t, len(res.([]core.Value)), 0)
testx.AssertEqual(t, conn.Out(), "0")
be.Err(t, err, nil)
be.Equal(t, len(res.([]core.Value)), 0)
be.Equal(t, conn.Out(), "0")
})
t.Run("key type mismatch", func(t *testing.T) {
db, red := getDB(t)
defer db.Close()
_, _ = db.Set().Add("key1", "one")
_ = db.Str().Set("key2", "two")
_, _ = db.Set().Add("key3", "thr")
red := getRedka(t)
_, _ = red.Set().Add("key1", "one")
_ = red.Str().Set("key2", "two")
_, _ = red.Set().Add("key3", "thr")
cmd := redis.MustParse(ParseSDiff, "sdiff key1 key2 key3")
conn := redis.NewFakeConn()
res, err := cmd.Run(conn, red)
testx.AssertNoErr(t, err)
testx.AssertEqual(t, len(res.([]core.Value)), 1)
testx.AssertEqual(t, conn.Out(), "1,one")
be.Err(t, err, nil)
be.Equal(t, len(res.([]core.Value)), 1)
be.Equal(t, conn.Out(), "1,one")
})
}