/*
* Copyright 2013 The Netty Project
*
* The Netty Project licenses this file to you under the Apache License, version
* 2.0 (the "License"); you may not use this file except in compliance with the
* License. You may obtain a copy of the License at:
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations under
* the License.
*/
package org.jboss.aerogear.io.netty.handler.codec.sockjs.handler;
import static org.hamcrest.CoreMatchers.equalTo;
import static org.hamcrest.CoreMatchers.is;
import static org.hamcrest.CoreMatchers.notNullValue;
import static org.hamcrest.CoreMatchers.nullValue;
import static org.hamcrest.MatcherAssert.assertThat;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import io.netty.handler.codec.http.DefaultFullHttpRequest;
import io.netty.handler.codec.http.FullHttpResponse;
import io.netty.handler.codec.http.HttpHeaders;
import io.netty.handler.codec.http.HttpMethod;
import io.netty.handler.codec.http.HttpRequest;
import io.netty.handler.codec.http.HttpVersion;
import org.jboss.aerogear.io.netty.handler.codec.sockjs.SockJsConfig;
import io.netty.util.CharsetUtil;
import org.junit.Test;
public class InfoTest {
@Test
public void webSocketSupported() throws Exception {
final SockJsConfig config = SockJsConfig.withPrefix("/simplepush").build();
final FullHttpResponse response = Info.response(config, createHttpRequest("/simplepush"));
assertThat(infoAsJson(response).get("websocket").asBoolean(), is(true));
response.release();
}
@Test
public void webSocketNotSupported() throws Exception {
final SockJsConfig config = SockJsConfig.withPrefix("/simplepush").disableWebSocket().build();
final FullHttpResponse response = Info.response(config, createHttpRequest("/simplepush"));
assertThat(infoAsJson(response).get("websocket").asBoolean(), is(false));
response.release();
}
@Test
public void cookiesNeeded() throws Exception {
final SockJsConfig config = SockJsConfig.withPrefix("/simplepush").disableWebSocket().cookiesNeeded().build();
final FullHttpResponse response = Info.response(config, createHttpRequest("/simplepush"));
assertThat(infoAsJson(response).get("cookie_needed").asBoolean(), is(true));
response.release();
}
@Test
public void cookiesNotNeeded() throws Exception {
final SockJsConfig config = SockJsConfig.withPrefix("/simplepush").disableWebSocket().build();
final FullHttpResponse response = Info.response(config, createHttpRequest("/simplepush"));
assertThat(infoAsJson(response).get("cookie_needed").asBoolean(), is(false));
response.release();
}
@Test
public void origins() throws Exception {
final SockJsConfig config = SockJsConfig.withPrefix("/simplepush").disableWebSocket().build();
final FullHttpResponse response = Info.response(config, createHttpRequest("/simplepush"));
assertThat(infoAsJson(response).get("origins").get(0).asText(), is("*:*"));
response.release();
}
@Test
public void entropy() throws Exception {
final SockJsConfig config = SockJsConfig.withPrefix("/simplepush").disableWebSocket().build();
final FullHttpResponse response = Info.response(config, createHttpRequest("/simplepush"));
assertThat(infoAsJson(response).get("entropy").asLong(), is(notNullValue()));
response.release();
}
@Test
public void contentTypeHeader() throws Exception {
assertThat(headersFromInfo().get(HttpHeaders.Names.CONTENT_TYPE), equalTo("application/json; charset=UTF-8"));
}
@Test
public void expiresHeader() throws Exception {
assertThat(headersFromInfo().get(HttpHeaders.Names.EXPIRES), is(nullValue()));
}
@Test
public void lastModifiedHeader() throws Exception {
assertThat(headersFromInfo().get(HttpHeaders.Names.LAST_MODIFIED), is(nullValue()));
}
@Test
public void setCookieHeader() throws Exception {
assertThat(headersFromInfo().get(HttpHeaders.Names.SET_COOKIE), is(nullValue()));
}
@Test
public void contentLenghtHeader() throws Exception {
assertThat(headersFromInfo().get(HttpHeaders.Names.CONTENT_LENGTH), is(notNullValue()));
}
@Test
public void cacheControlHeader() throws Exception {
final String cacheControl = headersFromInfo().get(HttpHeaders.Names.CACHE_CONTROL);
assertThat(cacheControl.contains("no-store"), is(true));
assertThat(cacheControl.contains("no-cache"), is(true));
assertThat(cacheControl.contains("must-revalidate"), is(true));
assertThat(cacheControl.contains("max-age"), is(true));
}
private static HttpHeaders headersFromInfo() throws Exception {
final SockJsConfig config = SockJsConfig.withPrefix("/simplepush").disableWebSocket().build();
final FullHttpResponse response = Info.response(config, createHttpRequest("/simplepush"));
final HttpHeaders headers = response.headers();
response.release();
return headers;
}
private static JsonNode infoAsJson(final FullHttpResponse response) throws Exception {
final ObjectMapper om = new ObjectMapper();
return om.readTree(response.content().toString(CharsetUtil.UTF_8));
}
private static HttpRequest createHttpRequest(final String prefix) {
return new DefaultFullHttpRequest(HttpVersion.HTTP_1_1, HttpMethod.GET, prefix + "/info");
}
}