Files
mautrix-telegram/pkg/gotd/crypto/encrypted_message_data_test.go
T
2025-06-27 20:03:37 -07:00

38 lines
895 B
Go

package crypto
import (
"testing"
"go.mau.fi/mautrix-telegram/pkg/gotd/bin"
"github.com/stretchr/testify/require"
)
func TestEncryptedMessageData_Encode(t *testing.T) {
const padding = 16
paddingRequired := func(l int) int {
return padding + (padding - (l % padding))
}
d := EncryptedMessageData{
Salt: 1034,
SeqNo: 1,
MessageID: 3401235566,
SessionID: 2345512351,
MessageDataLen: 5,
MessageDataWithPadding: []byte{1, 2, 3, 100, 112},
}
b := new(bin.Buffer)
if err := d.Encode(b); err != nil {
t.Fatal(err)
}
b.Buf = append(b.Buf, make([]byte, paddingRequired(b.Len()))...)
decoded := EncryptedMessageData{}
if err := decoded.Decode(b); err != nil {
t.Fatal(err)
}
decoded.MessageDataWithPadding = decoded.MessageDataWithPadding[:decoded.MessageDataLen]
require.Equal(t, d, decoded)
}