mirror of https://github.com/ollama/ollama
71 lines
2.0 KiB
Go
71 lines
2.0 KiB
Go
//go:build windows || darwin
|
|
|
|
package logrotate
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"testing"
|
|
)
|
|
|
|
func TestRotate(t *testing.T) {
|
|
logDir := t.TempDir()
|
|
logFile := filepath.Join(logDir, "testlog.log")
|
|
|
|
// No log exists
|
|
Rotate(logFile)
|
|
|
|
if err := os.WriteFile(logFile, []byte("1"), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := os.Stat(logFile); os.IsNotExist(err) {
|
|
t.Fatal("expected log file to exist")
|
|
}
|
|
|
|
// First rotation
|
|
Rotate(logFile)
|
|
if _, err := os.Stat(filepath.Join(logDir, "testlog-1.log")); os.IsNotExist(err) {
|
|
t.Fatal("expected rotated log file to exist")
|
|
}
|
|
if _, err := os.Stat(filepath.Join(logDir, "testlog-2.log")); !os.IsNotExist(err) {
|
|
t.Fatal("expected no second rotated log file")
|
|
}
|
|
if _, err := os.Stat(logFile); !os.IsNotExist(err) {
|
|
t.Fatal("expected original log file to be moved")
|
|
}
|
|
|
|
// Should be a no-op without a new log
|
|
Rotate(logFile)
|
|
if _, err := os.Stat(filepath.Join(logDir, "testlog-1.log")); os.IsNotExist(err) {
|
|
t.Fatal("expected rotated log file to still exist")
|
|
}
|
|
if _, err := os.Stat(filepath.Join(logDir, "testlog-2.log")); !os.IsNotExist(err) {
|
|
t.Fatal("expected no second rotated log file")
|
|
}
|
|
if _, err := os.Stat(logFile); !os.IsNotExist(err) {
|
|
t.Fatal("expected no original log file")
|
|
}
|
|
|
|
for i := 2; i <= MaxLogFiles+1; i++ {
|
|
if err := os.WriteFile(logFile, []byte(strconv.Itoa(i)), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := os.Stat(logFile); os.IsNotExist(err) {
|
|
t.Fatal("expected log file to exist")
|
|
}
|
|
Rotate(logFile)
|
|
if _, err := os.Stat(logFile); !os.IsNotExist(err) {
|
|
t.Fatal("expected log file to be moved")
|
|
}
|
|
for j := 1; j < i; j++ {
|
|
if _, err := os.Stat(filepath.Join(logDir, "testlog-"+strconv.Itoa(j)+".log")); os.IsNotExist(err) {
|
|
t.Fatalf("expected rotated log file %d to exist", j)
|
|
}
|
|
}
|
|
if _, err := os.Stat(filepath.Join(logDir, "testlog-"+strconv.Itoa(i+1)+".log")); !os.IsNotExist(err) {
|
|
t.Fatalf("expected no rotated log file %d", i+1)
|
|
}
|
|
}
|
|
}
|