49 lines
1.1 KiB
Go
49 lines
1.1 KiB
Go
package tgquick
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"testing"
|
|
)
|
|
|
|
func TestPopRI_RoundTrip(t *testing.T) {
|
|
payload := []byte{1, 2, 3, 4, 5, 6, 7, 8}
|
|
enc, err := randomizedIntermediateEncode(payload)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var acc []byte
|
|
acc = append(acc, enc...)
|
|
got, ok := popRI(&acc)
|
|
if !ok || len(acc) != 0 {
|
|
t.Fatalf("popRI: ok=%v rest=%d", ok, len(acc))
|
|
}
|
|
if string(got) != string(payload) {
|
|
t.Fatalf("payload mismatch: %v vs %v", got, payload)
|
|
}
|
|
}
|
|
|
|
func TestIsResPQ(t *testing.T) {
|
|
nonce := make([]byte, 16)
|
|
// resPQ: constructor + nonce(16) + server_nonce(16) + pq bytes + Vector<long>
|
|
msgData := make([]byte, 4+16+16+4+4)
|
|
binary.LittleEndian.PutUint32(msgData, tlResPQ)
|
|
copy(msgData[4:20], nonce)
|
|
copy(msgData[20:36], nonce)
|
|
// pq empty, fingerprints empty
|
|
if !isResPQ(wrapUnencrypted(msgData)) {
|
|
t.Fatal("expected resPQ")
|
|
}
|
|
}
|
|
|
|
func TestIsUnencryptedDCEnvelope(t *testing.T) {
|
|
msgData := make([]byte, 8)
|
|
binary.LittleEndian.PutUint32(msgData, 0x11223344)
|
|
w := wrapUnencrypted(msgData)
|
|
if !isUnencryptedDCEnvelope(w) {
|
|
t.Fatal("expected envelope for arbitrary constructor")
|
|
}
|
|
if isResPQ(w) {
|
|
t.Fatal("not resPQ")
|
|
}
|
|
}
|