From cb43f341387b0d845d29af98b1328a6029ae18ba Mon Sep 17 00:00:00 2001
From: Andrew Stewart <andrew@stwrt.ca>
Date: Fri, 5 Sep 2014 10:34:03 -0700
Subject: [PATCH] Fix broken tests

---
 test/specs/connection.spec.js | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/specs/connection.spec.js b/test/specs/connection.spec.js
index 40ba06b..85444aa 100644
--- a/test/specs/connection.spec.js
+++ b/test/specs/connection.spec.js
@@ -74,14 +74,14 @@ describe("Connection", function() {
   describe("#disconnect", function() {
     beforeEach(function() {
       stub(Logger, 'info').returns(true);
-      stub(connection.adaptor, 'disconnect').returns(true);
       stub(connection, 'removeAllListeners');
 
+      connection.adaptor.disconnect = stub().returns(true);
+
       connection.disconnect();
     });
 
     afterEach(function() {
-      connection.adaptor.disconnect.restore();
       connection.removeAllListeners.restore();
       Logger.info.restore();
     });