34 lines
844 B
Go
34 lines
844 B
Go
package logger
|
|
|
|
import (
|
|
"sync/atomic"
|
|
"testing"
|
|
|
|
"github.com/perfect-panel/server/pkg/color"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestWithColor(t *testing.T) {
|
|
old := atomic.SwapUint32(&encoding, plainEncodingType)
|
|
defer atomic.StoreUint32(&encoding, old)
|
|
|
|
output := WithColor("hello", color.BgBlue)
|
|
assert.Equal(t, "hello", output)
|
|
|
|
atomic.StoreUint32(&encoding, jsonEncodingType)
|
|
output = WithColor("hello", color.BgBlue)
|
|
assert.Equal(t, "hello", output)
|
|
}
|
|
|
|
func TestWithColorPadding(t *testing.T) {
|
|
old := atomic.SwapUint32(&encoding, plainEncodingType)
|
|
defer atomic.StoreUint32(&encoding, old)
|
|
|
|
output := WithColorPadding("hello", color.BgBlue)
|
|
assert.Equal(t, " hello ", output)
|
|
|
|
atomic.StoreUint32(&encoding, jsonEncodingType)
|
|
output = WithColorPadding("hello", color.BgBlue)
|
|
assert.Equal(t, "hello", output)
|
|
}
|