/** * Copyright © 2010-2011 Nokia * * Licensed 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 com.github.restdriver.serverdriver.acceptance; import static com.github.restdriver.serverdriver.Matchers.*; import static com.github.restdriver.serverdriver.RestServerDriver.*; import static org.hamcrest.MatcherAssert.*; import static org.hamcrest.Matchers.*; import java.util.regex.Pattern; import org.junit.Before; import org.junit.Rule; import org.junit.Test; import com.github.restdriver.clientdriver.ClientDriverRequest; import com.github.restdriver.clientdriver.ClientDriverResponse; import com.github.restdriver.clientdriver.ClientDriverRule; import com.github.restdriver.serverdriver.http.response.Response; public class HeadersAcceptanceTest { @Rule public ClientDriverRule driver = new ClientDriverRule(); private String baseUrl; @Before public void getServerDetails() { baseUrl = driver.getBaseUrl(); } @Test public void getRetrievesHeaders() { driver.addExpectation( new ClientDriverRequest("/"), new ClientDriverResponse("", null).withHeader("X-foo", "barrr")); Response response = get(baseUrl); // multiple ways to assert on a header assertThat(response, hasHeaderWithValue("X-foo", equalTo("barrr"))); assertThat(response, hasHeader("X-foo", equalTo("barrr"))); assertThat(response, hasHeader("X-foo", "barrr")); assertThat(response, hasHeader("X-foo: barrr")); assertThat(response, hasHeader(header("X-foo", "barrr"))); assertThat(response, hasHeader(header("X-foo: barrr"))); // header *names* are case insensitive assertThat(response, hasHeader(header("X-FOO: barrr"))); assertThat(response, hasHeader("X-FOO: barrr")); assertThat(response, hasHeader("X-FOO")); // but values are case-sensitive assertThat(response, not(hasHeader(header("X-foo: BARRR")))); assertThat(response, not(hasHeader("X-foo: BARRR"))); } @Test public void serverDriverSendsDefaultUserAgentWhenNoneIsSpecified() { driver.addExpectation( new ClientDriverRequest("/").withHeader("User-Agent", Pattern.compile("rest-server-driver/.*")), new ClientDriverResponse("received header", "text/plain")); Response response = get(baseUrl); assertThat(response.asText(), is("received header")); } @Test public void serverDriverSendsSpecifiedUserAgentWhenGiven() { driver.addExpectation( new ClientDriverRequest("/").withHeader("User-Agent", "provided"), new ClientDriverResponse("received header", "text/plain")); Response response = get(baseUrl, header("User-Agent", "provided")); assertThat(response.asText(), is("received header")); } }