`log.Fatal()`을 사용하는 Go 함수를 효과적으로 테스트하려면 어떻게 해야 합니까?

package main

import (
    "log"
    "os"
)

type TestLogger struct {
    *log.Logger
}

func (l *TestLogger) Fatal(v ...interface{}) {
    l.Output(2, fmt.Sprint(v...))
}

func (l *TestLogger) Fatalf(format string, v ...interface{}) {
    l.Output(2, fmt.Sprintf(format, v...))
}

import (
    "bytes"
    "github.com/YourPackage/logging"
)

func TestFunctionWithLogFatal(t *testing.T) {
    buffer := new(bytes.Buffer)
    testLogger := logging.NewTestLogger(buffer)
    log.SetOutput(testLogger)

    // Call the function under test here

    output := buffer.String()
    // Assert that the output contains the expected log message
}