wehub-resource-sync
71 lines
2.0 KiB
Go
71 lines
2.0 KiB
Go
package logging
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestEnforceLogDirSizeLimitDeletesOldest(t *testing.T) {
|
|
dir := t.TempDir()
|
|
|
|
writeLogFile(t, filepath.Join(dir, "old.log"), 60, time.Unix(1, 0))
|
|
writeLogFile(t, filepath.Join(dir, "mid.log"), 60, time.Unix(2, 0))
|
|
protected := filepath.Join(dir, "main.log")
|
|
writeLogFile(t, protected, 60, time.Unix(3, 0))
|
|
|
|
deleted, err := enforceLogDirSizeLimit(dir, 120, protected)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if deleted != 1 {
|
|
t.Fatalf("expected 1 deleted file, got %d", deleted)
|
|
}
|
|
|
|
if _, err := os.Stat(filepath.Join(dir, "old.log")); !os.IsNotExist(err) {
|
|
t.Fatalf("expected old.log to be removed, stat error: %v", err)
|
|
}
|
|
if _, err := os.Stat(filepath.Join(dir, "mid.log")); err != nil {
|
|
t.Fatalf("expected mid.log to remain, stat error: %v", err)
|
|
}
|
|
if _, err := os.Stat(protected); err != nil {
|
|
t.Fatalf("expected protected main.log to remain, stat error: %v", err)
|
|
}
|
|
}
|
|
|
|
func TestEnforceLogDirSizeLimitSkipsProtected(t *testing.T) {
|
|
dir := t.TempDir()
|
|
|
|
protected := filepath.Join(dir, "main.log")
|
|
writeLogFile(t, protected, 200, time.Unix(1, 0))
|
|
writeLogFile(t, filepath.Join(dir, "other.log"), 50, time.Unix(2, 0))
|
|
|
|
deleted, err := enforceLogDirSizeLimit(dir, 100, protected)
|
|
if err != nil {
|
|
t.Fatalf("unexpected error: %v", err)
|
|
}
|
|
if deleted != 1 {
|
|
t.Fatalf("expected 1 deleted file, got %d", deleted)
|
|
}
|
|
|
|
if _, err := os.Stat(protected); err != nil {
|
|
t.Fatalf("expected protected main.log to remain, stat error: %v", err)
|
|
}
|
|
if _, err := os.Stat(filepath.Join(dir, "other.log")); !os.IsNotExist(err) {
|
|
t.Fatalf("expected other.log to be removed, stat error: %v", err)
|
|
}
|
|
}
|
|
|
|
func writeLogFile(t *testing.T, path string, size int, modTime time.Time) {
|
|
t.Helper()
|
|
|
|
data := make([]byte, size)
|
|
if err := os.WriteFile(path, data, 0o644); err != nil {
|
|
t.Fatalf("write file: %v", err)
|
|
}
|
|
if err := os.Chtimes(path, modTime, modTime); err != nil {
|
|
t.Fatalf("set times: %v", err)
|
|
}
|
|
}
|