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/reply/asserts"
	"testing"
)

func TestExec(t *testing.T) {
	testCluster2 := MakeTestCluster([]string{"127.0.0.1:6379"})
	conn := &connection.FakeConn{}
	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.FakeConn{}
	ret := testCluster.Exec(conn, toArgs("GET", "a"))
	asserts.AssertErrReply(t, ret, "NOAUTH Authentication required")
	ret = testCluster.Exec(conn, toArgs("AUTH", passwd))
	asserts.AssertStatusReply(t, ret, "OK")
	ret = testCluster.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.FakeConn{}
	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.FakeConn{}, toArgs("SET", key, value))
	for _, v := range rets {
		asserts.AssertNotError(t, v)
	}
}