shanshanzhong
c582087c0f
feat: 添加版本和构建时间变量 fix: 修正短信队列类型注释错误 style: 清理未使用的代码和测试文件 docs: 更新安装文档中的下载链接 chore: 迁移数据库脚本添加日志和订阅配置
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)
|
|
}
|