Fix Go 1.4 build.

eSDhee · Feb 18, 2017 · c2156b5 · c2156b5
1 parent 59a96d9
commit c2156b5
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/pubsub_test.go b/pubsub_test.go
@@ -6,10 +6,10 @@ import (
 	"sync"
 	"time"
 
+	"gopkg.in/redis.v5"
+
 	. "github.com/onsi/ginkgo"
 	. "github.com/onsi/gomega"
-
-	"gopkg.in/redis.v5"
 )
 
 var _ = Describe("PubSub", func() {
@@ -348,10 +348,13 @@ var _ = Describe("PubSub", func() {
 
 			wg.Done()
 			defer wg.Done()
-
-			_, err := pubsub.ReceiveMessage()
-			Expect(err).To(MatchError("redis: client is closed"))
 
+			_, err := pubsub.ReceiveMessage()
+			Expect(err).To(HaveOccurred())
+			Expect(err).To(SatisfyAny(
+				MatchError("redis: client is closed"),
+				MatchError("use of closed network connection"), // Go 1.4
+			))
 		}()
 
 		wg.Wait()