44 lines
1.1 KiB
Go
44 lines
1.1 KiB
Go
package app
|
|
|
|
import (
|
|
"errors"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"git.pcux.cn/xuans/Smail/services/runtimelog"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
func TestFatalErrorIsLoggedBeforeRuntimeClose(t *testing.T) {
|
|
for _, panics := range []bool{false, true} {
|
|
root := t.TempDir()
|
|
previousHooks := log.StandardLogger().ReplaceHooks(make(log.LevelHooks))
|
|
writer, err := runtimelog.Start(root)
|
|
if err != nil {
|
|
log.StandardLogger().ReplaceHooks(previousHooks)
|
|
t.Fatal(err)
|
|
}
|
|
err = func() (runErr error) {
|
|
defer finishRuntimeLogs(writer, &runErr)
|
|
if panics {
|
|
panic("fatal-probe")
|
|
}
|
|
return errors.New("fatal-probe")
|
|
}()
|
|
log.StandardLogger().ReplaceHooks(previousHooks)
|
|
if err == nil || !strings.Contains(err.Error(), "fatal-probe") {
|
|
t.Fatalf("fatal error lost: %v", err)
|
|
}
|
|
files, err := filepath.Glob(filepath.Join(root, "log", "runtime-*.log"))
|
|
if err != nil || len(files) != 1 {
|
|
t.Fatalf("missing runtime log: %v", err)
|
|
}
|
|
data, err := os.ReadFile(files[0])
|
|
if err != nil || !strings.Contains(string(data), "fatal-probe") {
|
|
t.Fatal("fatal error was not written before close")
|
|
}
|
|
}
|
|
}
|