tests/org.eclipse.rap.rwt.cluster.testfixture.test/src/org/eclipse/rap/rwt/cluster/testfixture/client/RWTClient_Test.java

/*******************************************************************************
 * Copyright (c) 2011, 2012 EclipseSource and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    EclipseSource - initial API and implementation
 ******************************************************************************/
package org.eclipse.rap.rwt.cluster.testfixture.client;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertSame;

import java.io.IOException;
import java.net.URL;
import java.net.URLConnection;
import java.util.HashMap;
import java.util.Map;

import org.eclipse.rap.rwt.cluster.testfixture.test.TestHttpUrlConnection;
import org.eclipse.rap.rwt.cluster.testfixture.test.TestServletEngine;
import org.junit.Before;
import org.junit.Test;


public class RWTClient_Test {

  private TestServletEngine servletEngine;
  private TestConnectionProvider connectionProvider;

  @Before
  public void setUp() throws Exception {
    servletEngine = new TestServletEngine();
    connectionProvider = new TestConnectionProvider();
  }

  @Test
  public void testConstructor() {
    RWTClient client = new RWTClient( servletEngine );

    assertSame( servletEngine, client.getServletEngine() );
  }

  @Test
  public void testSendPostRequest() throws IOException {
    String responseContent = "responseContent";
    connectionProvider.setConnection( new TestHttpUrlConnection( responseContent ) );
    RWTClient client = new RWTClient( servletEngine, connectionProvider );

    Response response = client.sendPostRequest( new JsonMessage() );

    assertEquals( responseContent, response.getContentText() );
  }

  @Test
  public void testSendResourceRequest() throws IOException {
    connectionProvider.setConnection( new TestHttpUrlConnection( "responseContent" ) );
    RWTClient client = new RWTClient( servletEngine, connectionProvider );

    client.sendResourceRequest( "foo/bar.gif" );

    String connectionUrl = connectionProvider.getConnectionUrl().toExternalForm();
    assertEquals( "http://localhost:-1/foo/bar.gif", connectionUrl );
  }

  @Test
  public void testSendUICallBackRequest() throws IOException {
    connectionProvider.setConnection( new TestHttpUrlConnection( "responseContent" ) );
    RWTClient client = new RWTClient( servletEngine, connectionProvider );

    client.sendUICallBackRequest( 2000 );

    String expectedUrl = "http://localhost:-1/rwt?"
                         + "servicehandler=org.eclipse.rap.serverpush";
    String connectionUrl = connectionProvider.getConnectionUrl().toExternalForm();
    assertEquals( expectedUrl, connectionUrl );
  }

  @Test
  public void testRequestWithParameters() throws IOException {
    connectionProvider.setConnection( new TestHttpUrlConnection( "" ) );
    RWTClient client = new RWTClient( servletEngine, connectionProvider );
    Map<String,String> parameters = new HashMap<String,String>();
    parameters.put( "foo", "bar" );
    client.sendGetRequest( parameters );

    String connectionUrl = connectionProvider.getConnectionUrl().toExternalForm();
    assertEquals( "http://localhost:-1/rwt?foo=bar", connectionUrl );
  }

  @Test
  public void testSessionId_isReadFromCookieHeader() throws IOException {
    TestHttpUrlConnection connection = new TestHttpUrlConnection( "" );
    connection.setCookieToResponse( "JSESSIONID=node0xyz.node0;Path=/" );
    connectionProvider.setConnection( connection );
    RWTClient client = new RWTClient( servletEngine, connectionProvider );

    client.sendPostRequest( new JsonMessage() );

    assertEquals( "node0xyz.node0", client.getSessionId() );
  }

  @Test
  public void testSessionId_isEmptyWhenNoHeaderWasSent() throws IOException {
    connectionProvider.setConnection( new TestHttpUrlConnection( "" ) );
    RWTClient client = new RWTClient( servletEngine, connectionProvider );

    client.sendPostRequest( new JsonMessage() );

    assertEquals( "", client.getSessionId() );
  }

  @Test
  public void testSessionId_isNotOverridden() throws IOException {
    TestHttpUrlConnection connection1 = new TestHttpUrlConnection( "" );
    connection1.setCookieToResponse( "JSESSIONID=xyz;Path=/" );
    connectionProvider.setConnection( connection1 );
    RWTClient client = new RWTClient( servletEngine, connectionProvider );
    client.sendPostRequest( new JsonMessage() );

    TestHttpUrlConnection connection2 = new TestHttpUrlConnection( "" );
    connectionProvider.setConnection( connection2 );
    client.sendPostRequest( new JsonMessage() );

    assertEquals( "xyz", client.getSessionId() );
  }

  @Test
  public void testSessionId_remainsAfterChangedServletEngine() throws IOException {
    TestHttpUrlConnection connection = new TestHttpUrlConnection( "" );
    connection.setCookieToResponse( "JSESSIONID=xyz;Path=/" );
    connectionProvider.setConnection( connection );
    RWTClient client = new RWTClient( servletEngine, connectionProvider );
    client.sendPostRequest( new JsonMessage() );
    String sessionIdBeforeChange = client.getSessionId();
    TestServletEngine otherServletEngine = new TestServletEngine();

    client.changeServletEngine( otherServletEngine );

    assertNotNull( sessionIdBeforeChange );
    assertEquals( sessionIdBeforeChange, client.getSessionId() );
  }

  @Test
  public void testSessionId_isSentWithRequest() throws IOException {
    RWTClient client = new RWTClient( servletEngine, connectionProvider );
    TestHttpUrlConnection connection1 = new TestHttpUrlConnection( "" );
    connection1.setCookieToResponse( "JSESSIONID=xyz;Path=/" );
    connectionProvider.setConnection( connection1 );
    client.sendPostRequest( new JsonMessage() );

    TestHttpUrlConnection connection2 = new TestHttpUrlConnection( "" );
    connectionProvider.setConnection( connection2 );
    client.sendPostRequest( new JsonMessage() );

    assertEquals( "JSESSIONID=xyz", connection2.getCookieFromRequest() );
  }

  @Test
  public void testChangeServletEngine() {
    RWTClient client = new RWTClient( servletEngine );
    TestServletEngine otherServletEngine = new TestServletEngine();
    client.changeServletEngine( otherServletEngine );

    assertSame( otherServletEngine, client.getServletEngine() );
  }

  @Test
  public void testParseRequestCounter() {
    int counter = RWTClient.parseRequestCounter( "{\"requestCounter\": 23}" );
    assertEquals( 23, counter );
  }

  private static class TestConnectionProvider implements IConnectionProvider {
    private URLConnection connection;
    private URL connectionUrl;

    public URLConnection createConnection( URL url ) throws IOException {
      connectionUrl = url;
      return connection;
    }

    void setConnection( URLConnection connection ) {
      this.connection = connection;
    }

    URL getConnectionUrl() {
      return connectionUrl;
    }
  }
}