You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
898 B
Go
49 lines
898 B
Go
package logx
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log"
|
|
"strings"
|
|
"sync/atomic"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
const testlog = "Stay hungry, stay foolish."
|
|
|
|
func TestCollectSysLog(t *testing.T) {
|
|
CollectSysLog()
|
|
content := getContent(captureOutput(func() {
|
|
log.Print(testlog)
|
|
}))
|
|
assert.True(t, strings.Contains(content, testlog))
|
|
}
|
|
|
|
func TestRedirector(t *testing.T) {
|
|
var r redirector
|
|
content := getContent(captureOutput(func() {
|
|
r.Write([]byte(testlog))
|
|
}))
|
|
assert.Equal(t, testlog, content)
|
|
}
|
|
|
|
func captureOutput(f func()) string {
|
|
atomic.StoreUint32(&initialized, 1)
|
|
writer := new(mockWriter)
|
|
infoLog = writer
|
|
|
|
prevLevel := atomic.LoadUint32(&logLevel)
|
|
SetLevel(InfoLevel)
|
|
f()
|
|
SetLevel(prevLevel)
|
|
|
|
return writer.builder.String()
|
|
}
|
|
|
|
func getContent(jsonStr string) string {
|
|
var entry logEntry
|
|
json.Unmarshal([]byte(jsonStr), &entry)
|
|
return entry.Content
|
|
}
|