1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
package proto
import (
"net"
"testing"
"github.com/pion/stun"
"github.com/stretchr/testify/assert"
)
func TestRelayedAddress(t *testing.T) {
// Simple tests because already tested in stun.
a := RelayedAddress{
IP: net.IPv4(111, 11, 1, 2),
Port: 333,
}
t.Run("String", func(t *testing.T) {
if a.String() != "111.11.1.2:333" {
t.Error("invalid string")
}
})
m := new(stun.Message)
if err := a.AddTo(m); err != nil {
t.Fatal(err)
}
m.WriteHeader()
decoded := new(stun.Message)
_, err := decoded.Write(m.Raw)
assert.NoError(t, err)
var aGot RelayedAddress
assert.NoError(t, aGot.GetFrom(decoded))
}
|
