client_store_options_test.go

package pg

import (
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestWithClientStoreInitTableDisabled(t *testing.T) {
	store, err := NewClientStore(nil, WithClientStoreInitTableDisabled())
	require.NoError(t, err)
	assert.True(t, store.initTableDisabled)
}

func TestWithClientStoreTableName(t *testing.T) {
	randomName := time.Now().String()

	store, err := NewClientStore(nil, WithClientStoreTableName(randomName), WithClientStoreInitTableDisabled())
	require.NoError(t, err)
	assert.Equal(t, randomName, store.tableName)
}

func TestWithClientStoreLogger(t *testing.T) {
	l := new(memoryLogger)

	store, err := NewClientStore(nil, WithClientStoreLogger(l), WithClientStoreInitTableDisabled())
	require.NoError(t, err)

	store.logger.Printf("log1", 1, "2", "333")
	store.logger.Printf("log2", 12, "22")

	require.Equal(t, 2, len(l.formats))
	require.Equal(t, 2, len(l.args))

	assert.Equal(t, "log1", l.formats[0])
	assert.Equal(t, "log2", l.formats[1])

	require.Equal(t, 3, len(l.args[0]))
	require.Equal(t, 2, len(l.args[1]))

	assert.Equal(t, 1, l.args[0][0])
	assert.Equal(t, "2", l.args[0][1])
	assert.Equal(t, "333", l.args[0][2])

	assert.Equal(t, 12, l.args[1][0])
	assert.Equal(t, "22", l.args[1][1])
}