mirror of
https://github.com/netbirdio/netbird.git
synced 2025-03-01 08:21:16 +01:00
44 lines
966 B
Go
44 lines
966 B
Go
|
package messages
|
||
|
|
||
|
import (
|
||
|
"testing"
|
||
|
)
|
||
|
|
||
|
func TestMarshalHelloMsg(t *testing.T) {
|
||
|
peerID := []byte("abdFAaBcawquEiCMzAabYosuUaGLtSNhKxz+")
|
||
|
bHello, err := MarshalHelloMsg(peerID)
|
||
|
if err != nil {
|
||
|
t.Fatalf("error: %v", err)
|
||
|
}
|
||
|
|
||
|
receivedPeerID, err := UnmarshalHelloMsg(bHello)
|
||
|
if err != nil {
|
||
|
t.Fatalf("error: %v", err)
|
||
|
}
|
||
|
if string(receivedPeerID) != string(peerID) {
|
||
|
t.Errorf("expected %s, got %s", peerID, receivedPeerID)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
func TestMarshalTransportMsg(t *testing.T) {
|
||
|
peerID := []byte("abdFAaBcawquEiCMzAabYosuUaGLtSNhKxz+")
|
||
|
payload := []byte("payload")
|
||
|
msg, err := MarshalTransportMsg(peerID, payload)
|
||
|
if err != nil {
|
||
|
t.Fatalf("error: %v", err)
|
||
|
}
|
||
|
|
||
|
id, respPayload, err := UnmarshalTransportMsg(msg)
|
||
|
if err != nil {
|
||
|
t.Fatalf("error: %v", err)
|
||
|
}
|
||
|
|
||
|
if string(id) != string(peerID) {
|
||
|
t.Errorf("expected %s, got %s", peerID, id)
|
||
|
}
|
||
|
|
||
|
if string(respPayload) != string(payload) {
|
||
|
t.Errorf("expected %s, got %s", payload, respPayload)
|
||
|
}
|
||
|
}
|