Golang日志轮转测试方法分享

package main

import (
    "log"
    "os"

    "gopkg.in/natefinch/lumberjack.v2"
)

var Logger *log.Logger

func InitLogger(logFile string) {
    Logger = log.New(&lumberjack.Logger{
        Filename:   logFile,
        MaxSize:    1,  // 兆字节
        MaxBackups: 3,
        MaxAge:     7, // 天
        Compress:   true,
    }, "", log.LstdFlags)
}

package main

import (
    "io/ioutil"
    "os"
    "path/filepath"
    "strings"
    "testing"
)

func TestLogRotation(t *testing.T) {
    tempDir, err := ioutil.TempDir("", "logtest")
    if err != nil {
        t.Fatal(err)
    }
    defer os.RemoveAll(tempDir)

    logFile := filepath.Join(tempDir, "app.log")

    InitLogger(logFile)

    // 写入一些日志内容
    Logger.Println("这是第一条测试日志")
    Logger.Println("这是第二条测试日志")

    // 检查日志文件是否存在并有内容
    content, err := os.ReadFile(logFile)
    if err != nil {
        t.Fatalf("无法读取日志文件: %v", err)
    }

    if len(content) == 0 {
        t.Error("日志文件为空")
    }

    if !strings.Contains(string(content), "这是第一条测试日志") {
        t.Error("日志内容未正确写入")
    }

    // 验证 lumberjack 配置（通过类型断言检查）
    multiWriter := Logger.Writer()
    lw, ok := multiWriter.(*lumberjack.Logger)
    if !ok {
        t.Fatal("日志 writer 不是 lumberjack.Logger 类型")
    }

    if lw.MaxSize != 1 {
        t.Errorf("期望 MaxSize=1，实际为 %d", lw.MaxSize)
    }

    if lw.MaxBackups != 3 {
        t.Errorf("期望 MaxBackups=3，实际为 %d", lw.MaxBackups)
    }
}

if !lw.Compress {
    t.Error("期望启用压缩")
}

go mod init your-project-name
go get gopkg.in/natefinch/lumberjack.v2
go test -v