Add teardown methods to Vows contexts.

yui · Apr 3, 2012 · a3fea60 · a3fea60
1 parent 26129b1
commit a3fea60
Show file tree

Hide file tree

Showing 5 changed files with 21 additions and 2 deletions.
diff --git a/test/functional.js b/test/functional.js
@@ -216,6 +216,9 @@ function attachServerContext(testContext, explicitRoute) {
                 vow.callback(null, server);
             });
         },
+        teardown: function (server) {
+            server.close();
+        },
         "is connected": function (server) {
             assert.isNumber(server.address().port);
         },
@@ -260,6 +263,10 @@ function attachServerContext(testContext, explicitRoute) {
             "used by the Hub Client": {
                 // TODO: Handle without trailing slash.
                 topic: hub.clientTopic(route + "/"),
+                teardown: function (topic) {
+                    // TODO Add end method, bug #54.
+                    // topic.client.end();
+                },
                 "a browser for testing": {
                     topic: hub.phantomTopic(),
                     "visits Yeti": testContext
@@ -270,7 +277,6 @@ function attachServerContext(testContext, explicitRoute) {
 }
 
 function attachServerBatch(definition) {
-
     var batch = {},
         routeWords = ["foo", "bar", "baz", "quux"];
 

diff --git a/test/lib/blizzard-http.js b/test/lib/blizzard-http.js
@@ -28,7 +28,10 @@ exports.sessionContext = function (subContext) {
                     server: serverBlizzardSession
                 });
             });
-
+        },
+        teardown: function (topic) {
+            topic.client.end();
+            topic.server.end();
         }
     };
 

diff --git a/test/lib/blizzard.js b/test/lib/blizzard.js
@@ -50,6 +50,10 @@ exports.sessionContext = function (subContext) {
             });
 
             clientSocket.once("error", vow.callback);
+        },
+        teardown: function (topic) {
+            topic.client.end();
+            topic.server.end();
         }
     };
 

diff --git a/test/lib/hub.js b/test/lib/hub.js
@@ -50,6 +50,9 @@ var clientContext = exports.clientContext = function (subContext) {
                 });
                 hub.once("error", vow.callback);
             },
+            teardown: function (hub) {
+                hub.close();
+            },
             "is ok": function (hub) {
                 assert.ok(hub);
                 assert.isNumber(hub.server.address().port);

diff --git a/test/listen.js b/test/listen.js
@@ -52,6 +52,9 @@ vows.describe("Yeti Listen").addBatch({
                 vow.callback(null, server);
             });
         },
+        teardown: function (server) {
+            server.close();
+        },
         "is connected": function (topic) {
             assert.ok(topic.address().port);
         },