mirror of
https://github.com/Control-D-Inc/ctrld.git
synced 2026-07-16 13:17:19 +02:00
Codescribe
33 lines
626 B
Go
33 lines
626 B
Go
package cli
|
|
|
|
import (
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
|
|
"go.uber.org/zap"
|
|
"go.uber.org/zap/zapcore"
|
|
|
|
"github.com/Control-D-Inc/ctrld"
|
|
)
|
|
|
|
var logOutput strings.Builder
|
|
|
|
func TestMain(m *testing.M) {
|
|
// Create a custom writer that writes to logOutput
|
|
writer := zapcore.AddSync(&logOutput)
|
|
|
|
// Create zap encoder
|
|
encoderConfig := zap.NewDevelopmentEncoderConfig()
|
|
encoder := zapcore.NewConsoleEncoder(encoderConfig)
|
|
|
|
// Create core that writes to our string builder
|
|
core := zapcore.NewCore(encoder, writer, zap.DebugLevel)
|
|
|
|
// Create logger
|
|
l := zap.New(core)
|
|
|
|
mainLog.Store(&ctrld.Logger{Logger: l})
|
|
os.Exit(m.Run())
|
|
}
|