diff --git a/pubsub_test.go b/pubsub_test.go index ba26be7..22f3130 100644 --- 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()