X-Git-Url: https://git.r.bdr.sh/rbdr/mobius/blobdiff_plain/d4c152a4dba0eec7c8ecd13732900909f51b1c97..f8e4cd540b87de3e308ec18a2b040b284a741522:/hotline/server_blackbox_test.go?ds=sidebyside diff --git a/hotline/server_blackbox_test.go b/hotline/server_blackbox_test.go index 17ad7f4..b45e279 100644 --- a/hotline/server_blackbox_test.go +++ b/hotline/server_blackbox_test.go @@ -1,43 +1,70 @@ package hotline import ( + "encoding/hex" "github.com/stretchr/testify/assert" - "go.uber.org/zap" - "go.uber.org/zap/zapcore" + "log/slog" "os" "testing" ) -func NewTestLogger() *zap.SugaredLogger { - encoderCfg := zap.NewProductionEncoderConfig() - encoderCfg.TimeKey = "timestamp" - encoderCfg.EncodeTime = zapcore.ISO8601TimeEncoder - - core := zapcore.NewCore( - zapcore.NewConsoleEncoder(encoderCfg), - zapcore.Lock(os.Stdout), - zap.DebugLevel, - ) - - cores := []zapcore.Core{core} - l := zap.New(zapcore.NewTee(cores...)) - defer func() { _ = l.Sync() }() - return l.Sugar() +func NewTestLogger() *slog.Logger { + return slog.New(slog.NewTextHandler(os.Stdout, nil)) } -// tranAssertEqual compares equality of transactions slices after stripping out the random ID +// assertTransferBytesEqual takes a string with a hexdump in the same format that `hexdump -C` produces and compares with +// a hexdump for the bytes in got, after stripping the create/modify timestamps. +// I don't love this, but as git does not preserve file create/modify timestamps, we either need to fully mock the +// filesystem interactions or work around in this way. +// TODO: figure out a better solution +func assertTransferBytesEqual(t *testing.T, wantHexDump string, got []byte) bool { + if wantHexDump == "" { + return true + } + + var clean []byte + clean = append(clean, got[:92]...) // keep the first 92 bytes + clean = append(clean, make([]byte, 16)...) // replace the next 16 bytes for create/modify timestamps + clean = append(clean, got[108:]...) // keep the rest + + return assert.Equal(t, wantHexDump, hex.Dump(clean)) +} + +// tranAssertEqual compares equality of transactions slices after stripping out the random transaction ID func tranAssertEqual(t *testing.T, tran1, tran2 []Transaction) bool { var newT1 []Transaction var newT2 []Transaction + for _, trans := range tran1 { trans.ID = []byte{0, 0, 0, 0} + var fs []Field + for _, field := range trans.Fields { + if field.ID == [2]byte{0x00, 0x6b} { // FieldRefNum + continue + } + if field.ID == [2]byte{0x00, 0x72} { // FieldChatID + continue + } + fs = append(fs, field) + } + trans.Fields = fs newT1 = append(newT1, trans) } for _, trans := range tran2 { trans.ID = []byte{0, 0, 0, 0} + var fs []Field + for _, field := range trans.Fields { + if field.ID == [2]byte{0x00, 0x6b} { // FieldRefNum + continue + } + if field.ID == [2]byte{0x00, 0x72} { // FieldChatID + continue + } + fs = append(fs, field) + } + trans.Fields = fs newT2 = append(newT2, trans) - } return assert.Equal(t, newT1, newT2)