1 files changed, 53 insertions, 0 deletions
diff --git a/core/src/test/java/fi/iki/elonen/integration/ShutdownTest.java b/core/src/test/java/fi/iki/elonen/integration/ShutdownTest.java
new file mode 100644
index 0000000..0fcb275
--- /dev/null
+++ b/core/src/test/java/fi/iki/elonen/integration/ShutdownTest.java
@@ -0,0 +1,53 @@
+package fi.iki.elonen.integration;
+
+import static org.junit.Assert.*;
+import fi.iki.elonen.NanoHTTPD;
+
+import org.junit.Test;
+
+import java.io.IOException;
+import java.io.InputStream;
+import java.net.HttpURLConnection;
+import java.net.MalformedURLException;
+import java.net.URL;
+
+public class ShutdownTest {
+
+    @Test
+    public void connectionsAreClosedWhenServerStops() throws IOException {
+        TestServer server = new TestServer();
+        server.start();
+        makeRequest();
+        server.stop();
+        try {
+            makeRequest();
+            fail("Connection should be closed!");
+        } catch (IOException e) {
+            // Expected exception
+        }
+    }
+
+    private void makeRequest() throws MalformedURLException, IOException {
+        HttpURLConnection connection = (HttpURLConnection) new URL("http://localhost:8092/").openConnection();
+        // Keep-alive seems to be on by default, but just in case that changes.
+        connection.addRequestProperty("Connection", "keep-alive");
+        InputStream in = connection.getInputStream();
+        while (in.available() > 0) {
+            in.read();
+        }
+        in.close();
+    }
+
+    private class TestServer extends NanoHTTPD {
+
+        public TestServer() {
+            super(8092);
+        }
+
+        @Override
+        public Response serve(IHTTPSession session) {
+            return new Response("Whatever");
+        }
+    }
+
+}