Merge pull request #46 from libp2p/test-timeout

pick test timeout from env var if available.

test/stream.go

@@ -8,7 +8,9 @@ import (
 	"io"
 	"io/ioutil"
 	mrand "math/rand"
+	"os"
 	"runtime/debug"
+	"strconv"
 	"sync"
 	"testing"
 	"time"
@@ -25,12 +27,20 @@ var VerboseDebugging = false
 
 var randomness []byte
 
+var StressTestTimeout = 1 * time.Minute
+
 func init() {
 	// read 1MB of randomness
 	randomness = make([]byte, 1<<20)
 	if _, err := crand.Read(randomness); err != nil {
 		panic(err)
 	}
+
+	if timeout := os.Getenv("TEST_STRESS_TIMEOUT_MS"); timeout != "" {
+		if v, err := strconv.ParseInt(timeout, 10, 32); err == nil {
+			StressTestTimeout = time.Duration(v) * time.Millisecond
+		}
+	}
 }
 
 type Options struct {
@@ -342,7 +352,7 @@ func SubtestStreamOpenStress(t *testing.T, ta, tb tpt.Transport, maddr ma.Multia
 		}
 	}()
 
-	timeout := time.After(time.Minute * 1)
+	timeout := time.After(StressTestTimeout)
 	done := make(chan struct{})
 
 	go func() {