rdb/rdb_test.go

package rdb_test

import (
	"os"
	"testing"

	"git.bit5.ru/backend/colog"
	"git.bit5.ru/backend/rdb"
	"github.com/stretchr/testify/assert"
)

func getPool() *rdb.Pool {
	pool := rdb.OpenPool(rdb.Settings{Host: "localhost", Port: 6379, Db: 10, ClientName: "test"}, getLogger())
	return pool
}

func getLogger() *colog.CoLog {
	tlog := colog.NewCoLog(os.Stderr, "", 0)
	ft := &colog.StdFormatter{Flag: 0}
	ft.ColorSupported(true)
	tlog.SetFormatter(ft)
	return tlog
}

func TestBadConnection(t *testing.T) {
	pool := rdb.OpenPool(rdb.Settings{Host: "dummy", Port: 80, ClientName: "test"}, getLogger())
	conn := pool.Get()
	assert.NotNil(t, conn)
	_, err := conn.Do("PING")
	assert.NotNil(t, err)
}

func TestGetConn(t *testing.T) {
	pool := getPool()
	defer pool.Close()

	conn := pool.Get()
	defer conn.Close()
	_, err := conn.Do("PING")
	assert.Nil(t, err)
}

func TestCloseConn(t *testing.T) {
	pool := getPool()
	defer pool.Close()

	conn := pool.Get()
	assert.EqualValues(t, 0, pool.RP.IdleCount())

	_, err := conn.Do("PING")
	assert.Nil(t, err)

	conn.Close()
	_, err = conn.Do("PING")
	assert.NotNil(t, err)
	assert.EqualValues(t, 1, pool.RP.IdleCount())
}

func TestDoubleCloseConnIsOk(t *testing.T) {
	pool := getPool()
	defer pool.Close()

	conn := pool.Get()
	_, err := conn.Do("PING")
	assert.Nil(t, err)

	assert.EqualValues(t, 0, pool.RP.IdleCount())
	conn.Close()
	assert.EqualValues(t, 1, pool.RP.IdleCount())
	conn.Close()
	assert.EqualValues(t, 1, pool.RP.IdleCount())
}