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.assertEquals;
     37 import static org.junit.Assert.assertTrue;
     38 
     39 import java.util.ArrayList;
     40 import java.util.Calendar;
     41 import java.util.List;
     42 
     43 import org.apache.http.client.CookieStore;
     44 import org.apache.http.client.ResponseHandler;
     45 import org.apache.http.client.methods.HttpGet;
     46 import org.apache.http.impl.client.BasicResponseHandler;
     47 import org.apache.http.impl.cookie.BasicClientCookie;
     48 import org.junit.Test;
     49 
     50 import fi.iki.elonen.NanoHTTPD;
     51 
     52 /**
     53  * @author Paul S. Hawke (paul.hawke (at) gmail.com) On: 9/2/13 at 10:10 PM
     54  */
     55 public class CookieIntegrationTest extends IntegrationTestBase<CookieIntegrationTest.CookieTestServer> {
     56 
     57     public static class CookieTestServer extends NanoHTTPD {
     58 
     59         List<Cookie> cookiesReceived = new ArrayList<Cookie>();
     60 
     61         List<Cookie> cookiesToSend = new ArrayList<Cookie>();
     62 
     63         public CookieTestServer() {
     64             super(8192);
     65         }
     66 
     67         @Override
     68         public Response serve(IHTTPSession session) {
     69             CookieHandler cookies = session.getCookies();
     70             for (String cookieName : cookies) {
     71                 this.cookiesReceived.add(new Cookie(cookieName, cookies.read(cookieName)));
     72             }
     73             for (Cookie c : this.cookiesToSend) {
     74                 cookies.set(c);
     75             }
     76             return newFixedLengthResponse("Cookies!");
     77         }
     78     }
     79 
     80     @Override
     81     public CookieTestServer createTestServer() {
     82         return new CookieTestServer();
     83     }
     84 
     85     @Test
     86     public void testCookieSentBackToClient() throws Exception {
     87         this.testServer.cookiesToSend.add(new NanoHTTPD.Cookie("name", "value", 30));
     88         HttpGet httpget = new HttpGet("http://localhost:8192/");
     89         ResponseHandler<String> responseHandler = new BasicResponseHandler();
     90         this.httpclient.execute(httpget, responseHandler);
     91 
     92         CookieStore cookies = this.httpclient.getCookieStore();
     93         assertEquals(1, cookies.getCookies().size());
     94         assertEquals("name", cookies.getCookies().get(0).getName());
     95         assertEquals("value", cookies.getCookies().get(0).getValue());
     96     }
     97 
     98     @Test
     99     public void testNoCookies() throws Exception {
    100         HttpGet httpget = new HttpGet("http://localhost:8192/");
    101         ResponseHandler<String> responseHandler = new BasicResponseHandler();
    102         this.httpclient.execute(httpget, responseHandler);
    103 
    104         CookieStore cookies = this.httpclient.getCookieStore();
    105         assertEquals(0, cookies.getCookies().size());
    106     }
    107 
    108     @Test
    109     public void testServerReceivesCookiesSentFromClient() throws Exception {
    110         BasicClientCookie clientCookie = new BasicClientCookie("name", "value");
    111         Calendar calendar = Calendar.getInstance();
    112         calendar.add(Calendar.DAY_OF_YEAR, 100);
    113         clientCookie.setExpiryDate(calendar.getTime());
    114         clientCookie.setDomain("localhost");
    115         this.httpclient.getCookieStore().addCookie(clientCookie);
    116         HttpGet httpget = new HttpGet("http://localhost:8192/");
    117         ResponseHandler<String> responseHandler = new BasicResponseHandler();
    118         this.httpclient.execute(httpget, responseHandler);
    119 
    120         assertEquals(1, this.testServer.cookiesReceived.size());
    121         assertTrue(this.testServer.cookiesReceived.get(0).getHTTPHeader().contains("name=value"));
    122     }
    123 }
    124