status-go/protocol/zaputil/encoder_test.go

82 lines
2.1 KiB
Go
Raw Normal View History

package zaputil
import (
"io/ioutil"
"testing"
"time"
"github.com/stretchr/testify/require"
"go.uber.org/zap"
"go.uber.org/zap/zapcore"
)
func TestJSONHexEncoder(t *testing.T) {
encoder := NewJSONHexEncoder(zap.NewDevelopmentEncoderConfig())
encoder.AddBinary("test-key", []byte{0x01, 0x02, 0x03})
buf, err := encoder.EncodeEntry(zapcore.Entry{
LoggerName: "",
Time: time.Now(),
Level: zapcore.DebugLevel,
Message: "",
}, nil)
require.NoError(t, err)
require.Contains(t, buf.String(), `"test-key":"0x010203"`)
}
func TestLoggerWithJSONHexEncoder(t *testing.T) {
err := RegisterJSONHexEncoder()
require.NoError(t, err)
tmpFile, err := ioutil.TempFile("", "")
require.NoError(t, err)
cfg := zap.NewDevelopmentConfig()
cfg.OutputPaths = []string{tmpFile.Name()}
cfg.Encoding = "json-hex"
l, err := cfg.Build()
require.NoError(t, err)
l.With(zap.Binary("some-field", []byte{0x01, 0x02, 0x03})).Debug("test message")
err = l.Sync()
require.NoError(t, err)
data, err := ioutil.ReadFile(tmpFile.Name())
require.NoError(t, err)
require.Contains(t, string(data), "0x010203")
}
func TestConsoleHexEncoder(t *testing.T) {
encoder := NewConsoleHexEncoder(zap.NewDevelopmentEncoderConfig())
encoder.AddBinary("test-key", []byte{0x01, 0x02, 0x03})
buf, err := encoder.EncodeEntry(zapcore.Entry{
LoggerName: "",
Time: time.Now(),
Level: zapcore.DebugLevel,
Message: "",
}, nil)
require.NoError(t, err)
require.Contains(t, buf.String(), `{"test-key": "0x010203"}`)
}
func TestLoggerWithConsoleHexEncoder(t *testing.T) {
err := RegisterConsoleHexEncoder()
require.NoError(t, err)
tmpFile, err := ioutil.TempFile("", "")
require.NoError(t, err)
cfg := zap.NewDevelopmentConfig()
cfg.OutputPaths = []string{tmpFile.Name()}
cfg.Encoding = "console-hex"
l, err := cfg.Build()
require.NoError(t, err)
l.With(zap.Binary("some-field", []byte{0x01, 0x02, 0x03})).Debug("test message")
err = l.Sync()
require.NoError(t, err)
data, err := ioutil.ReadFile(tmpFile.Name())
require.NoError(t, err)
require.Contains(t, string(data), "0x010203")
}