Home | History | Annotate | Download | only in integration 
      1 package fi.iki.elonen.integration;
      2 
      3 /*
      4  * #%L
      5  * NanoHttpd-Core
      6  * %%
      7  * Copyright (C) 2012 - 2015 nanohttpd
      8  * %%
      9  * Redistribution and use in source and binary forms, with or without modification,
     10  * are permitted provided that the following conditions are met:
     11  *
     12  * 1. Redistributions of source code must retain the above copyright notice, this
     13  *    list of conditions and the following disclaimer.
     14  *
     15  * 2. Redistributions in binary form must reproduce the above copyright notice,
     16  *    this list of conditions and the following disclaimer in the documentation
     17  *    and/or other materials provided with the distribution.
     18  *
     19  * 3. Neither the name of the nanohttpd nor the names of its contributors
     20  *    may be used to endorse or promote products derived from this software without
     21  *    specific prior written permission.
     22  *
     23  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
     24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
     25  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
     26  * IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
     27  * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
     28  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
     29  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
     30  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
     31  * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
     32  * OF THE POSSIBILITY OF SUCH DAMAGE.
     33  * #L%
     34  */
     35 
     36 import static org.junit.Assert.fail;
     37 
     38 import java.io.IOException;
     39 import java.io.InputStream;
     40 import java.net.HttpURLConnection;
     41 import java.net.MalformedURLException;
     42 import java.net.URL;
     43 
     44 import org.junit.Test;
     45 
     46 import fi.iki.elonen.NanoHTTPD;
     47 
     48 public class ShutdownTest {
     49 
     50     private class TestServer extends NanoHTTPD {
     51 
     52         public TestServer() {
     53             super(8092);
     54         }
     55 
     56         @Override
     57         public Response serve(IHTTPSession session) {
     58             return newFixedLengthResponse("Whatever");
     59         }
     60     }
     61 
     62     @Test
     63     public void connectionsAreClosedWhenServerStops() throws IOException {
     64         TestServer server = new TestServer();
     65         server.start();
     66         makeRequest();
     67         server.stop();
     68         try {
     69             makeRequest();
     70             fail("Connection should be closed!");
     71         } catch (IOException e) {
     72             // Expected exception
     73         }
     74     }
     75 
     76     private void makeRequest() throws MalformedURLException, IOException {
     77         HttpURLConnection connection = (HttpURLConnection) new URL("http://localhost:8092/").openConnection();
     78         // Keep-alive seems to be on by default, but just in case that changes.
     79         connection.addRequestProperty("Connection", "keep-alive");
     80         InputStream in = connection.getInputStream();
     81         while (in.available() > 0) {
     82             in.read();
     83         }
     84         in.close();
     85     }
     86 
     87 }
     88