From 7c675a02302532f248aecf3ddc96ae5e41b562b1 Mon Sep 17 00:00:00 2001
From: Harry Vangberg <harry@vangberg.name>
Date: Tue, 23 Jun 2015 15:14:20 +0200
Subject: [PATCH] Add ability to stop MockServer again

This is useful when using MockServer in different test files
and you are running the full suite.
---
 lib/mock_server.rb       | 21 +++++++++++++++++++++
 test/mock_server_test.rb | 17 +++++++++++++++++
 2 files changed, 38 insertions(+)

diff --git a/lib/mock_server.rb b/lib/mock_server.rb
index 99c4089..763dfc5 100644
--- a/lib/mock_server.rb
+++ b/lib/mock_server.rb
@@ -23,6 +23,14 @@ def start
     self
   end
 
+  def stop
+    Rack::Handler::WEBrick.shutdown
+
+    wait_for_shutdown("0.0.0.0", @port)
+
+    self
+  end
+
   module Methods
     def mock_server(*args, &block)
       app = Class.new(Sinatra::Base)
@@ -45,6 +53,7 @@ def listening?(host, port)
       socket.close unless socket.nil?
       true
     rescue Errno::ECONNREFUSED,
+      Errno::ECONNRESET,
       Errno::EBADF,           # Windows
       Errno::EADDRNOTAVAIL    # Windows
       false
@@ -62,4 +71,16 @@ def wait_for_service(host, port, timeout = 5)
 
     true
   end
+
+  def wait_for_shutdown(host, port, timeout = 5)
+    start_time = Time.now
+
+    until !listening?(host, port)
+      if timeout && (Time.now > (start_time + timeout))
+        raise SocketError.new("Socket did not close within #{timeout} seconds")
+      end
+    end
+
+    true
+  end
 end
diff --git a/test/mock_server_test.rb b/test/mock_server_test.rb
index b6097f5..4589442 100644
--- a/test/mock_server_test.rb
+++ b/test/mock_server_test.rb
@@ -50,3 +50,20 @@ def test_server
     assert_equal "Goodbye", open("http://localhost:4002").read
   end
 end
+
+class MockServerStopTest < Test::Unit::TestCase
+  def setup
+    @server = MockServer.new(HelloWorldSinatra, 4003)
+    @server.start
+  end
+
+  def test_stop_server
+    assert_equal "Hello", open("http://localhost:4003").read
+
+    @server.stop
+
+    assert_raises(Errno::ECONNREFUSED) {
+      open("http://localhost:4003").read
+    }
+  end
+end