mirror of
https://github.com/HDT3213/godis.git
synced 2025-10-05 16:57:06 +08:00
58 lines
1.7 KiB
Go
58 lines
1.7 KiB
Go
package cluster
|
|
|
|
import (
|
|
"github.com/hdt3213/godis/config"
|
|
"github.com/hdt3213/godis/lib/utils"
|
|
"github.com/hdt3213/godis/redis/connection"
|
|
"github.com/hdt3213/godis/redis/protocol/asserts"
|
|
"testing"
|
|
)
|
|
|
|
func TestExec(t *testing.T) {
|
|
testCluster2 := MakeTestCluster([]string{"127.0.0.1:6379"})
|
|
conn := connection.NewFakeConn()
|
|
for i := 0; i < 1000; i++ {
|
|
key := RandString(4)
|
|
value := RandString(4)
|
|
testCluster2.Exec(conn, toArgs("SET", key, value))
|
|
ret := testCluster2.Exec(conn, toArgs("GET", key))
|
|
asserts.AssertBulkReply(t, ret, value)
|
|
}
|
|
}
|
|
|
|
func TestAuth(t *testing.T) {
|
|
passwd := utils.RandString(10)
|
|
config.Properties.RequirePass = passwd
|
|
defer func() {
|
|
config.Properties.RequirePass = ""
|
|
}()
|
|
conn := connection.NewFakeConn()
|
|
ret := testNodeA.Exec(conn, toArgs("GET", "a"))
|
|
asserts.AssertErrReply(t, ret, "NOAUTH Authentication required")
|
|
ret = testNodeA.Exec(conn, toArgs("AUTH", passwd))
|
|
asserts.AssertStatusReply(t, ret, "OK")
|
|
ret = testNodeA.Exec(conn, toArgs("GET", "a"))
|
|
asserts.AssertNotError(t, ret)
|
|
}
|
|
|
|
func TestRelay(t *testing.T) {
|
|
testCluster2 := MakeTestCluster([]string{"127.0.0.1:6379"})
|
|
key := RandString(4)
|
|
value := RandString(4)
|
|
conn := connection.NewFakeConn()
|
|
ret := testCluster2.relay("127.0.0.1:6379", conn, toArgs("SET", key, value))
|
|
asserts.AssertNotError(t, ret)
|
|
ret = testCluster2.relay("127.0.0.1:6379", conn, toArgs("GET", key))
|
|
asserts.AssertBulkReply(t, ret, value)
|
|
}
|
|
|
|
func TestBroadcast(t *testing.T) {
|
|
testCluster2 := MakeTestCluster([]string{"127.0.0.1:6379"})
|
|
key := RandString(4)
|
|
value := RandString(4)
|
|
rets := testCluster2.broadcast(connection.NewFakeConn(), toArgs("SET", key, value))
|
|
for _, v := range rets {
|
|
asserts.AssertNotError(t, v)
|
|
}
|
|
}
|