Make RelayTest take the timeout as parameter.

2021-02-26 13:22:47 +01:00
parent e8bc42d84c
commit 0a8f75d6c9
2 changed files with 4 additions and 4 deletions
@@ -137,7 +137,7 @@ func ICEConfiguration() *webrtc.Configuration {
 	return &conf.conf
 }

-func RelayTest() (time.Duration, error) {
+func RelayTest(timeout time.Duration) (time.Duration, error) {

 	conf := ICEConfiguration()
 	conf2 := *conf
@@ -223,7 +223,7 @@ func RelayTest() (time.Duration, error) {
 		})
 	})

-	timer := time.NewTimer(20 * time.Second)
+	timer := time.NewTimer(timeout)
 	defer timer.Stop()
 	select {
 	case err := <-ch1:
@@ -235,6 +235,6 @@ func RelayTest() (time.Duration, error) {
 		}
 		return time.Now().Sub(tm), nil
 	case <-timer.C:
-		return 0, errors.New("timeout")
+		return 0, os.ErrDeadlineExceeded
 	}
 }