From 200ced827af8cebdf3e8926285147d37cdf0b420 Mon Sep 17 00:00:00 2001
From: Matt Ranney <mjr@ranney.com>
Date: Sun, 19 Sep 2010 23:47:39 -0700
Subject: [PATCH] Add test for pub/sub.

---
 test.js | 33 +++++++++++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/test.js b/test.js
index 53de3d5726..3faa9ef7b0 100644
--- a/test.js
+++ b/test.js
@@ -1,6 +1,7 @@
 /*global require console setTimeout process */
 var redis = require("./index"),
     client = redis.createClient(),
+    client2 = redis.createClient(),
     assert = require("assert"),
     sys = require('sys'),
     tests = {}, iterations = 10000;
@@ -107,6 +108,37 @@ tests.HINCRBY = function () {
     client.HINCRBY("hash incr", "value 2", 1, last(name, require_number(1, name)));
 };
 
+tests.SUBSCRIBE = function () {
+    var client1 = client, msg_count = 0, name = "SUBSCRIBE";
+    
+    client1.on("subscribe", function (channel, count) {
+        if (channel === "chan1") {
+            client2.publish("chan1", "message 1", require_number(1, name));
+            client2.publish("chan2", "message 2", require_number(1, name));
+            client2.publish("chan1", "message 3", require_number(1, name));
+        }
+    });
+
+    client1.on("unsubscribe", function (channel, count) {
+        if (count === 0) {
+            // make sure this connection can go into and out of pub/sub mode
+            client1.incr("did a thing", last(name, require_number(2, name)));
+        }
+    });
+
+    client1.on("message", function (channel, message) {
+        msg_count += 1;
+        assert.strictEqual("message " + msg_count, message.toString());
+        if (msg_count === 3) {
+            client1.unsubscribe("chan1", "chan2");
+        }
+    });
+
+    // make sure this connection can go into and out of pub/sub mode
+    client1.set("did a thing", 1, require_string("OK", name));
+    client1.subscribe("chan1", "chan2");
+};
+
 tests.EXISTS = function () {
     var name = "EXISTS";
     client.del("foo", "foo2", require_number_any(name));
@@ -328,6 +360,7 @@ function run_next_test() {
     } else {
         console.log('\n  completed \x1b[32m%d\x1b[0m tests in \x1b[33m%d\x1b[0m ms\n', test_count, new Date - all_start);
         client.end();
+        client2.end();
     }
 }