Commit 0a8f75d6 authored by Juliusz Chroboczek's avatar Juliusz Chroboczek

Make RelayTest take the timeout as parameter.

parent e8bc42d8
...@@ -130,7 +130,7 @@ func main() { ...@@ -130,7 +130,7 @@ func main() {
func relayTest() { func relayTest() {
now := time.Now() now := time.Now()
d, err := ice.RelayTest() d, err := ice.RelayTest(20 * time.Second)
if err != nil { if err != nil {
log.Printf("Relay test failed: %v", err) log.Printf("Relay test failed: %v", err)
log.Printf("Perhaps you didn't configure a TURN server?") log.Printf("Perhaps you didn't configure a TURN server?")
......
...@@ -137,7 +137,7 @@ func ICEConfiguration() *webrtc.Configuration { ...@@ -137,7 +137,7 @@ func ICEConfiguration() *webrtc.Configuration {
return &conf.conf return &conf.conf
} }
func RelayTest() (time.Duration, error) { func RelayTest(timeout time.Duration) (time.Duration, error) {
conf := ICEConfiguration() conf := ICEConfiguration()
conf2 := *conf conf2 := *conf
...@@ -223,7 +223,7 @@ func RelayTest() (time.Duration, error) { ...@@ -223,7 +223,7 @@ func RelayTest() (time.Duration, error) {
}) })
}) })
timer := time.NewTimer(20 * time.Second) timer := time.NewTimer(timeout)
defer timer.Stop() defer timer.Stop()
select { select {
case err := <-ch1: case err := <-ch1:
...@@ -235,6 +235,6 @@ func RelayTest() (time.Duration, error) { ...@@ -235,6 +235,6 @@ func RelayTest() (time.Duration, error) {
} }
return time.Now().Sub(tm), nil return time.Now().Sub(tm), nil
case <-timer.C: case <-timer.C:
return 0, errors.New("timeout") return 0, os.ErrDeadlineExceeded
} }
} }
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment