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.
62 lines
1.0 KiB
Go
62 lines
1.0 KiB
Go
package nlog
|
|
|
|
import (
|
|
"encoding/json"
|
|
"log"
|
|
"strings"
|
|
"sync/atomic"
|
|
"testing"
|
|
|
|
"git.noahlan.cn/noahlan/ntool/ntest/assert"
|
|
)
|
|
|
|
const testlog = "Stay hungry, stay foolish."
|
|
|
|
var testobj = map[string]any{"foo": "bar"}
|
|
|
|
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 {
|
|
w := new(mockWriter)
|
|
old := writer.Swap(w)
|
|
defer writer.Store(old)
|
|
|
|
prevLevel := atomic.LoadUint32(&logLevel)
|
|
SetLevel(InfoLevel)
|
|
f()
|
|
SetLevel(prevLevel)
|
|
|
|
return w.String()
|
|
}
|
|
|
|
func getContent(jsonStr string) string {
|
|
var entry map[string]any
|
|
json.Unmarshal([]byte(jsonStr), &entry)
|
|
|
|
val, ok := entry[contentKey]
|
|
if !ok {
|
|
return ""
|
|
}
|
|
|
|
str, ok := val.(string)
|
|
if !ok {
|
|
return ""
|
|
}
|
|
|
|
return str
|
|
}
|