mirror of
https://github.com/pion/ice.git
synced 2025-10-05 15:37:07 +08:00
39 lines
792 B
Go
39 lines
792 B
Go
package ice
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/pion/stun"
|
|
)
|
|
|
|
func TestPriority_GetFrom(t *testing.T) { //nolint:dupl
|
|
m := new(stun.Message)
|
|
var p PriorityAttr
|
|
if err := p.GetFrom(m); !errors.Is(err, stun.ErrAttributeNotFound) {
|
|
t.Error("unexpected error")
|
|
}
|
|
if err := m.Build(stun.BindingRequest, &p); err != nil {
|
|
t.Error(err)
|
|
}
|
|
m1 := new(stun.Message)
|
|
if _, err := m1.Write(m.Raw); err != nil {
|
|
t.Error(err)
|
|
}
|
|
var p1 PriorityAttr
|
|
if err := p1.GetFrom(m1); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if p1 != p {
|
|
t.Error("not equal")
|
|
}
|
|
t.Run("IncorrectSize", func(t *testing.T) {
|
|
m3 := new(stun.Message)
|
|
m3.Add(stun.AttrPriority, make([]byte, 100))
|
|
var p2 PriorityAttr
|
|
if err := p2.GetFrom(m3); !stun.IsAttrSizeInvalid(err) {
|
|
t.Error("should error")
|
|
}
|
|
})
|
|
}
|