无法修改初始化逻辑时，如何捕获Zap Logger输出做单元测试？

方案一：用Observer Core包装原有Logger（推荐，完整捕获所有字段） #

之前用Observer方案没抓到InitialFields，大概率是直接替换了Core，没保留原Logger的上下文（包括初始字段、命名空间）。这个方案通过包装原有Core，能完整捕获所有日志信息。

实现步骤&代码 #

import (
    "testing"

    "go.uber.org/zap"
    "go.uber.org/zap/zapcore"
    "go.uber.org/zap/zaptest/observer"
    "github.com/stretchr/testify/assert"
)

func TestLoggerOutput(t *testing.T) {
    // 拿到已初始化的Logger（这部分代码不能改）
    logger, err := NewCustomLogger(context.Background(), zapcore.DebugLevel, "unit tests")
    if err != nil {
        assert.Fail(t, "初始化Logger失败")
        return
    }

    // 创建Observer核心和日志观测器
    observedCore, observedLogs := observer.New(zapcore.DebugLevel)
    
    // 包装原Logger的核心：保留原Encoder，替换为ObserverCore
    // 关键：WrapCore会保留原Logger的所有上下文（初始字段、命名空间等）
    wrappedLogger := logger.WithOptions(zap.WrapCore(func(originalCore zapcore.Core) zapcore.Core {
        return observer.NewObserverCore(originalCore.GetEncoder(), observedCore.LevelEnabler())
    }))

    // 输出测试日志
    wrappedLogger.Info("测试日志", zap.String("动态字段", "动态值"))

    // 断言日志内容
    entries := observedLogs.All()
    assert.Len(t, entries, 1)
    entry := entries[0]

    // 验证基础字段
    assert.Equal(t, zapcore.InfoLevel, entry.Level)
    assert.Equal(t, "测试日志", entry.Message)
    assert.Equal(t, "unit tests", entry.LoggerName) // 验证命名空间

    // 验证InitialFields
    assert.Equal(t, "INITIAL_FIELD_VALUE", entry.ContextMap()["INITIAL_FIELD"])
    // 验证动态添加的字段
    assert.Equal(t, "动态值", entry.ContextMap()["动态字段"])
}

为什么这个方案能解决问题？ #

zap.WrapCore不会重建Logger，而是在原有Logger的上下文基础上替换输出核心，原有的InitialFields、命名空间、钩子等都会被保留，ObserverCore会完整接收所有日志字段。

方案二：正确捕获标准输出（适合验证原始JSON格式） #

如果你需要验证日志的原始JSON输出格式，这个方案更合适。之前捕获失败可能是因为没提前替换stdout，或者没手动刷新输出缓冲。

场景1：可以提前替换stdout（在Logger初始化前） #

import (
    "bytes"
    "encoding/json"
    "os"
    "testing"

    "go.uber.org/zap"
    "go.uber.org/zap/zapcore"
    "github.com/stretchr/testify/assert"
)

func TestLoggerStdoutCapture(t *testing.T) {
    // 先把标准输出替换成缓冲区，必须在初始化Logger前做
    originalStdout := os.Stdout
    defer func() { os.Stdout = originalStdout }()

    var buf bytes.Buffer
    os.Stdout = &buf

    // 初始化Logger（此时Logger会输出到缓冲区）
    logger, err := NewCustomLogger(context.Background(), zapcore.DebugLevel, "unit tests")
    if err != nil {
        assert.Fail(t, "初始化Logger失败")
        return
    }

    // 输出测试日志
    logger.Info("测试日志", zap.String("动态字段", "动态值"))

    // 手动刷新缓冲（Zap生产配置默认用同步写入，需要触发刷新）
    if syncWriter, ok := zapcore.Lock(&buf).(zapcore.WriteSyncer); ok {
        syncWriter.Sync()
    }

    // 解析JSON并断言
    var logEntry map[string]interface{}
    err = json.Unmarshal(buf.Bytes(), &logEntry)
    assert.NoError(t, err)

    assert.Equal(t, "info", logEntry["level"])
    assert.Equal(t, "测试日志", logEntry["msg"])
    assert.Equal(t, "unit tests", logEntry["logger"])
    assert.Equal(t, "INITIAL_FIELD_VALUE", logEntry["INITIAL_FIELD"])
    assert.Equal(t, "动态值", logEntry["dynamic_field"])
}

场景2：Logger已初始化，无法提前替换stdout #

可以用反射修改Logger核心的输出目标：

import (
    "bytes"
    "reflect"
    "testing"

    "go.uber.org/zap"
    "go.uber.org/zap/zapcore"
    "github.com/stretchr/testify/assert"
)

// 替换Logger的输出目标
func replaceLoggerWriter(logger *zap.Logger, writer zapcore.WriteSyncer) {
    core := logger.Core()
    // 反射获取core的writer字段（针对Zap默认的ioCore）
    coreVal := reflect.ValueOf(core).Elem()
    writerField := coreVal.FieldByName("ws")
    if writerField.IsValid() && writerField.CanSet() {
        writerField.Set(reflect.ValueOf(zapcore.Lock(writer)))
    }
}

func TestExistingLoggerStdoutCapture(t *testing.T) {
    // 拿到已初始化的Logger
    logger, err := NewCustomLogger(context.Background(), zapcore.DebugLevel, "unit tests")
    if err != nil {
        assert.Fail(t, "初始化Logger失败")
        return
    }

    // 创建缓冲区并替换Logger的输出
    var buf bytes.Buffer
    replaceLoggerWriter(logger, &buf)

    // 输出测试日志
    logger.Info("测试日志", zap.String("动态字段", "动态值"))

    // 刷新缓冲
    if syncWriter, ok := zapcore.Lock(&buf).(zapcore.WriteSyncer); ok {
        syncWriter.Sync()
    }

    // 解析并断言（同场景1的解析逻辑）
    // ...
}

注意事项 #

反射方案依赖Zap内部Core的结构，不同版本可能有变化，优先用场景1的方案。

内容的提问来源于stack exchange，提问作者Anatoly