Enable sessions on existing servers (fixes #30)

rjz · Jan 24, 2018 · 3ae4930 · 3ae4930
1 parent 1070373
commit 3ae4930
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/index.js b/index.js
@@ -19,6 +19,9 @@ function Session (app, options) {
 
   if (typeof app === 'function') {
     app = http.createServer(app);
+  }
+
+  if (app instanceof http.Server) {
     url = supertest.Test.prototype.serverAddress(app, '/');
   }
 

diff --git a/test/main_spec.js b/test/main_spec.js
@@ -84,6 +84,19 @@ describe('supertest-session', function () {
         .end(done);
     });
   });
+
+  describe('(#30) requesting URL of instantiated app', function () {
+    it('behaves correctly', function (done) {
+      var server = http.createServer(app);
+      var sess = session(server);
+      sess
+        .get('/')
+        .expect(200)
+        .expect('GET,,1')
+        .end(done);
+    });
+  });
+
 });
 
 describe('Session with a .before hook', function () {