56 lines
1.7 KiB
Go
56 lines
1.7 KiB
Go
package rpc
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/iamxvbaba/td/clock"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// fakeTURN is a minimal turnsrv.Service for exercising phoneCallConnections.
|
|
type fakeTURN struct {
|
|
ip string
|
|
port int
|
|
enabled bool
|
|
credUser string
|
|
credPass string
|
|
}
|
|
|
|
func (f *fakeTURN) Enabled() bool { return f.enabled }
|
|
func (f *fakeTURN) Credentials(string) (string, string, error) {
|
|
return f.credUser, f.credPass, nil
|
|
}
|
|
func (f *fakeTURN) IP() string { return f.ip }
|
|
func (f *fakeTURN) Port() int { return f.port }
|
|
func (f *fakeTURN) Close() error { return nil }
|
|
|
|
func newTURNRouter(t *testing.T, turn *fakeTURN) *Router {
|
|
t.Helper()
|
|
return New(Config{}, Deps{TURN: turn}, zap.NewNop(), clock.System)
|
|
}
|
|
|
|
func TestPhoneCallConnectionsDisabledTURNReturnsNil(t *testing.T) {
|
|
r := newTURNRouter(t, &fakeTURN{enabled: false})
|
|
if conns := r.phoneCallConnections(1); conns != nil {
|
|
t.Fatalf("disabled TURN: want nil, got %+v", conns)
|
|
}
|
|
}
|
|
|
|
func TestPhoneCallConnectionsSplitStunAndTurn(t *testing.T) {
|
|
r := newTURNRouter(t, &fakeTURN{
|
|
enabled: true, ip: "89.28.58.29", port: 12400,
|
|
credUser: "u", credPass: "p",
|
|
})
|
|
conns := r.phoneCallConnections(1)
|
|
if len(conns) != 2 {
|
|
t.Fatalf("want 2 conns (stun+turn), got %d: %+v", len(conns), conns)
|
|
}
|
|
// STUN and TURN must be SEPARATE entries (DrKLO's JNI ignores the stun flag
|
|
// on a combined entry, dropping STUN).
|
|
if !conns[0].Stun || conns[0].Turn || conns[0].ID != 1 || conns[0].IP != "89.28.58.29" {
|
|
t.Fatalf("conn[0] should be STUN id1, got %+v", conns[0])
|
|
}
|
|
if !conns[1].Turn || conns[1].Stun || conns[1].ID != 2 || conns[1].Username != "u" || conns[1].Password != "p" {
|
|
t.Fatalf("conn[1] should be TURN id2 with creds, got %+v", conns[1])
|
|
}
|
|
}
|