mirror of https://github.com/jech/galene.git
97 lines
1.9 KiB
Go
97 lines
1.9 KiB
Go
package ice
|
|
|
|
import (
|
|
"crypto/hmac"
|
|
"crypto/sha1"
|
|
"encoding/base64"
|
|
"os"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/pion/webrtc/v3"
|
|
|
|
"github.com/jech/galene/turnserver"
|
|
)
|
|
|
|
func TestPassword(t *testing.T) {
|
|
s := Server{
|
|
URLs: []string{"turn:turn.example.org"},
|
|
Username: "jch",
|
|
Credential: "secret",
|
|
}
|
|
|
|
ss := webrtc.ICEServer{
|
|
URLs: []string{"turn:turn.example.org"},
|
|
Username: "jch",
|
|
Credential: "secret",
|
|
}
|
|
|
|
sss, err := getServer(s)
|
|
|
|
if err != nil || !reflect.DeepEqual(sss, ss) {
|
|
t.Errorf("Got %v, expected %v", sss, ss)
|
|
}
|
|
}
|
|
|
|
func TestHMAC(t *testing.T) {
|
|
s := Server{
|
|
URLs: []string{"turn:turn.example.org"},
|
|
Username: "jch",
|
|
Credential: "secret",
|
|
CredentialType: "hmac-sha1",
|
|
}
|
|
|
|
ss := webrtc.ICEServer{
|
|
URLs: []string{"turn:turn.example.org"},
|
|
}
|
|
|
|
sss, err := getServer(s)
|
|
|
|
if !strings.HasSuffix(sss.Username, ":"+s.Username) {
|
|
t.Errorf("username is %v", ss.Username)
|
|
}
|
|
ss.Username = sss.Username
|
|
|
|
mac := hmac.New(sha1.New, []byte(s.Credential.(string)))
|
|
mac.Write([]byte(sss.Username))
|
|
buf := strings.Builder{}
|
|
e := base64.NewEncoder(base64.StdEncoding, &buf)
|
|
e.Write(mac.Sum(nil))
|
|
e.Close()
|
|
ss.Credential = buf.String()
|
|
|
|
if err != nil || !reflect.DeepEqual(sss, ss) {
|
|
t.Errorf("Got %v, expected %v", sss, ss)
|
|
}
|
|
}
|
|
|
|
func TestICEConfiguration(t *testing.T) {
|
|
ICEFilename = "/tmp/no/such/file"
|
|
turnserver.Address = ""
|
|
|
|
conf := ICEConfiguration()
|
|
if conf == nil {
|
|
t.Errorf("conf is nil")
|
|
}
|
|
conf2 := ICEConfiguration()
|
|
if conf2 != conf {
|
|
t.Errorf("conf2 != conf")
|
|
}
|
|
|
|
if len(conf.ICEServers) != 0 {
|
|
t.Errorf("len(ICEServers) = %v", len(conf.ICEServers))
|
|
}
|
|
}
|
|
|
|
func TestRelayTest(t *testing.T) {
|
|
ICEFilename = "/tmp/no/such/file"
|
|
turnserver.Address = ""
|
|
|
|
_, err := RelayTest(200 * time.Millisecond)
|
|
if err == nil || !os.IsTimeout(err) {
|
|
t.Errorf("Relay test returned %v", err)
|
|
}
|
|
}
|