increase timeout before mining the chunk (#906)

pkg/api/pss_test.go

@@ -35,6 +35,7 @@ var (
 	payload     = []byte("testdata")
 	topic       = pss.NewTopic("testtopic")
 	timeout     = 10 * time.Second
+	longTimeout = 30 * time.Second
 )
 
 // creates a single websocket handler for an arbitrary topic, and receives a message
@@ -48,7 +49,9 @@ func TestPssWebsocketSingleHandler(t *testing.T) {
 		done       = make(chan struct{})
 	)
 
-	err := cl.SetReadDeadline(time.Now().Add(timeout))
+	// the long timeout is needed so that we dont time out while still mining the message with Wrap()
+	// otherwise the test (and other tests below) flakes
+	err := cl.SetReadDeadline(time.Now().Add(longTimeout))
 	if err != nil {
 		t.Fatal(err)
 	}
@@ -80,7 +83,7 @@ func TestPssWebsocketSingleHandlerDeregister(t *testing.T) {
 		done       = make(chan struct{})
 	)
 
-	err := cl.SetReadDeadline(time.Now().Add(timeout))
+	err := cl.SetReadDeadline(time.Now().Add(longTimeout))
 
 	if err != nil {
 		t.Fatal(err)
@@ -122,7 +125,7 @@ func TestPssWebsocketMultiHandler(t *testing.T) {
 		t.Fatalf("dial: %v. url %v", err, u.String())
 	}
 
-	err = cl.SetReadDeadline(time.Now().Add(timeout))
+	err = cl.SetReadDeadline(time.Now().Add(longTimeout))
 	if err != nil {
 		t.Fatal(err)
 	}