redis/go-redis
1
0
Fork 0
mirror of https://github.com/redis/go-redis.git synced 2025-07-29 17:41:15 +03:00
Code Activity

pubsub: add PSubscribe test. Updates #233.

Browse Source
This commit is contained in:
Vladimir Mihailenco
2015-12-31 11:27:28 +02:00
parent d226168576
commit 1739cd9380
2 changed files with 18 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
pubsub_test.go
View File

@ -255,11 +255,7 @@ var _ = Describe("PubSub", func() {
wg.Wait()
})
It("should reconnect on ReceiveMessage error", func() {
pubsub, err := client.Subscribe("mychannel")
Expect(err).NotTo(HaveOccurred())
defer pubsub.Close()
expectReceiveMessage := func(pubsub *redis.PubSub) {
cn1, _, err := pubsub.Pool().Get()
Expect(err).NotTo(HaveOccurred())
cn1.SetNetConn(&badConn{
@ -284,6 +280,22 @@ var _ = Describe("PubSub", func() {
Expect(msg.Payload).To(Equal("hello"))
wg.Wait()
}
It("Subscribe should reconnect on ReceiveMessage error", func() {
pubsub, err := client.Subscribe("mychannel")
Expect(err).NotTo(HaveOccurred())
defer pubsub.Close()
expectReceiveMessage(pubsub)
})
It("PSubscribe should reconnect on ReceiveMessage error", func() {
pubsub, err := client.PSubscribe("mychannel")
Expect(err).NotTo(HaveOccurred())
defer pubsub.Close()
expectReceiveMessage(pubsub)
})
It("should return on Close", func() {