test: more

Author: snitin315

lib/Server.js

@@ -750,7 +750,7 @@ class Server {
     }
   }
 
-  listen(port, hostname, fn) {
+  listen(port, hostname = this.options.host, fn) {
     if (
       typeof this.options.host !== 'undefined' &&
       hostname !== this.options.host

test/server/Server.test.js

@@ -12,7 +12,6 @@ jest.mock('sockjs/lib/transport');
 
 const baseDevConfig = {
   port,
-  host: 'localhost',
   static: false,
 };
 
@@ -137,6 +136,10 @@ describe('Server', () => {
       compiler = webpack(config);
     });
 
+    afterEach((done) => {
+      server.close(done);
+    });
+
     it('should listen and not throw error', (done) => {
       const options = {
         host: 'localhost',
@@ -148,6 +151,17 @@ describe('Server', () => {
       server.close(done);
     });
 
+    it('should work and listen', (done) => {
+      const options = {
+        host: 'localhost',
+      };
+
+      server = new Server(compiler, options);
+
+      expect(() => server.listen(port)).not.toThrowError();
+      server.close(done);
+    });
+
     it('should throw an error', (done) => {
       const options = {
         host: '192.160.21.34',