1 /*
2 * Copyright (c) 2011-2022, jcabi.com
3 * All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met: 1) Redistributions of source code must retain the above
8 * copyright notice, this list of conditions and the following
9 * disclaimer. 2) Redistributions in binary form must reproduce the above
10 * copyright notice, this list of conditions and the following
11 * disclaimer in the documentation and/or other materials provided
12 * with the distribution. 3) Neither the name of the jcabi.com nor
13 * the names of its contributors may be used to endorse or promote
14 * products derived from this software without specific prior written
15 * permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
18 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT
19 * NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
20 * FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL
21 * THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
22 * INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
23 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
24 * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
26 * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
27 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
28 * OF THE POSSIBILITY OF SUCH DAMAGE.
29 */
30 package com.jcabi.http.wire;
31
32 import com.jcabi.http.mock.MkAnswer;
33 import com.jcabi.http.mock.MkContainer;
34 import com.jcabi.http.mock.MkGrizzlyContainer;
35 import com.jcabi.http.request.JdkRequest;
36 import com.jcabi.http.response.RestResponse;
37 import jakarta.ws.rs.core.HttpHeaders;
38 import java.net.HttpURLConnection;
39 import org.hamcrest.MatcherAssert;
40 import org.hamcrest.Matchers;
41 import org.junit.jupiter.api.Test;
42
43 /**
44 * Test case for {@link UserAgentWire}.
45 * @since 1.2
46 */
47 final class UserAgentWireTest {
48
49 @Test
50 void addsDefaultUserAgentHeader() throws Exception {
51 final MkContainer container = new MkGrizzlyContainer().next(
52 new MkAnswer.Simple("")
53 ).start();
54 new JdkRequest(container.home())
55 .through(UserAgentWire.class)
56 .fetch()
57 .as(RestResponse.class)
58 .assertStatus(HttpURLConnection.HTTP_OK);
59 container.stop();
60 MatcherAssert.assertThat(
61 "must add default User-Agent HTTP header",
62 container.take().headers(),
63 Matchers.hasEntry(
64 Matchers.is(HttpHeaders.USER_AGENT),
65 Matchers.contains(
66 Matchers.startsWith("jcabi-")
67 )
68 )
69 );
70 }
71
72 @Test
73 void addsCustomUserAgentHeader() throws Exception {
74 final MkContainer container = new MkGrizzlyContainer().next(
75 new MkAnswer.Simple("")
76 ).start();
77 final String agent = "Mozilla/5.0";
78 new JdkRequest(container.home())
79 .through(UserAgentWire.class, agent)
80 .fetch()
81 .as(RestResponse.class)
82 .assertStatus(HttpURLConnection.HTTP_OK);
83 container.stop();
84 MatcherAssert.assertThat(
85 "must add custom User-Agent HTTP header",
86 container.take().headers(),
87 Matchers.hasEntry(
88 Matchers.is(HttpHeaders.USER_AGENT),
89 Matchers.contains(agent)
90 )
91 );
92 }
93
94 }