1 files changed, 51 insertions, 0 deletions
diff --git a/tests/pubsub.py b/tests/pubsub.py
new file mode 100644
index 0000000..1746ac4
--- /dev/null
+++ b/tests/pubsub.py
@@ -0,0 +1,51 @@
+import redis
+import unittest
+
+class PubSubTestCase(unittest.TestCase):
+    def setUp(self):
+        self.connection_pool = redis.ConnectionPool()
+        self.client = redis.Redis(connection_pool=self.connection_pool)
+        self.pubsub = self.client.pubsub()
+
+    def tearDown(self):
+        self.connection_pool.disconnect()
+
+    def test_channel_subscribe(self):
+        self.assertEquals(
+            self.pubsub.subscribe('foo'),
+            ['subscribe', 'foo', 1]
+            )
+        self.assertEquals(self.client.publish('foo', 'hello foo'), 1)
+        self.assertEquals(
+            self.pubsub.listen().next(),
+            {
+                'type': 'message',
+                'pattern': None,
+                'channel': 'foo',
+                'data': 'hello foo'
+            }
+            )
+        self.assertEquals(
+            self.pubsub.unsubscribe('foo'),
+            ['unsubscribe', 'foo', 0]
+            )
+
+    def test_pattern_subscribe(self):
+        self.assertEquals(
+            self.pubsub.psubscribe('fo*'),
+            ['psubscribe', 'fo*', 1]
+            )
+        self.assertEquals(self.client.publish('foo', 'hello foo'), 1)
+        self.assertEquals(
+            self.pubsub.listen().next(),
+            {
+                'type': 'pmessage',
+                'pattern': 'fo*',
+                'channel': 'foo',
+                'data': 'hello foo'
+            }
+            )
+        self.assertEquals(
+            self.pubsub.punsubscribe('fo*'),
+            ['punsubscribe', 'fo*', 0]
+            )