logger_test.go

package lark

import (
	"context"
	"fmt"
	"testing"

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

func TestSetLogger(t *testing.T) {
	bot := newTestBot()
	newLogger := initDefaultLogger()
	bot.SetLogger(newLogger)
	assert.Equal(t, newLogger, bot.logger)
	assert.Equal(t, newLogger, bot.Logger())
}

func TestLogLevel(t *testing.T) {
	var logLevel LogLevel = LogLevelDebug
	assert.Equal(t, "DEBUG", logLevel.String())
	logLevel = LogLevelError
	assert.Equal(t, "ERROR", logLevel.String())
	logLevel = LogLevelTrace
	assert.Equal(t, "TRACE", logLevel.String())
	logLevel = LogLevelWarn
	assert.Equal(t, "WARN", logLevel.String())
	logLevel = LogLevelInfo
	assert.Equal(t, "INFO", logLevel.String())
	logLevel = 1000
	assert.Equal(t, "", logLevel.String())
}

func TestWithContext(t *testing.T) {
	bot := newTestBot()
	assert.Equal(t, "context.Background", fmt.Sprintf("%s", bot.ctx))
	bot.WithContext(context.TODO())
	assert.Equal(t, "context.TODO", fmt.Sprintf("%s", bot.ctx))
}