Merge remote-tracking branch 'origin/master' into reset-all-buckets

dafortune · Sep 20, 2018 · c50b25e · c50b25e
2 parents 8585ef1 + 71997f6
commit c50b25e
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/lib/client.js b/lib/client.js
@@ -113,6 +113,13 @@ class LimitdRedis extends EventEmitter {
   resetAll(cb) {
     this.db.resetAll(cb);
   }
+
+  close(callback) {
+    this.db.close((err) => {
+      this.db.removeAllListeners();
+      callback(err);
+    });
+  }
 }
 
 module.exports = LimitdRedis;

diff --git a/test/client.tests.js b/test/client.tests.js
@@ -168,4 +168,15 @@ describe('LimitdRedis', () => {
       client.resetAll(done);
     });
   });
+
+  describe('#close', () => {
+    it('should call db.close', (done) => {
+      client.db.close = (cb) => cb();
+      client.close((err) => {
+        assert.equal(client.db.listenerCount('error'), 0);
+        assert.equal(client.db.listenerCount('ready'), 0);
+        done(err);
+      });
+    });
+  });
 });