|
29 | 29 | import javax.ws.rs.core.Context; |
30 | 30 | import javax.ws.rs.core.HttpHeaders; |
31 | 31 | import javax.ws.rs.core.Response; |
| 32 | +import javax.ws.rs.core.UriInfo; |
32 | 33 |
|
33 | 34 | import javax.inject.Singleton; |
34 | 35 |
|
|
40 | 41 | import org.apache.http.auth.AuthScope; |
41 | 42 | import org.apache.http.auth.UsernamePasswordCredentials; |
42 | 43 | import org.apache.http.client.CredentialsProvider; |
| 44 | +import org.apache.http.impl.conn.PoolingHttpClientConnectionManager; |
43 | 45 | import org.junit.Ignore; |
44 | 46 | import org.junit.Test; |
45 | 47 | import static org.junit.Assert.assertEquals; |
@@ -116,6 +118,8 @@ public void testPreemptiveAuthPost() { |
116 | 118 | public static class AuthResource { |
117 | 119 |
|
118 | 120 | int requestCount = 0; |
| 121 | + int queryParamsBasicRequestCount = 0; |
| 122 | + int queryParamsDigestRequestCount = 0; |
119 | 123 |
|
120 | 124 | @GET |
121 | 125 | public String get(@Context HttpHeaders h) { |
@@ -144,6 +148,25 @@ public String getFilter(@Context HttpHeaders h) { |
144 | 148 | return "GET"; |
145 | 149 | } |
146 | 150 |
|
| 151 | + @GET |
| 152 | + @Path("noauth") |
| 153 | + public String get() { |
| 154 | + return "GET"; |
| 155 | + } |
| 156 | + |
| 157 | + @GET |
| 158 | + @Path("digest") |
| 159 | + public String getDigest(@Context HttpHeaders h) { |
| 160 | + String value = h.getRequestHeaders().getFirst("Authorization"); |
| 161 | + if (value == null) { |
| 162 | + throw new WebApplicationException( |
| 163 | + Response.status(401).header("WWW-Authenticate", "Digest realm=\"WallyWorld\"") |
| 164 | + .entity("Forbidden").build()); |
| 165 | + } |
| 166 | + |
| 167 | + return "GET"; |
| 168 | + } |
| 169 | + |
147 | 170 | @POST |
148 | 171 | public String post(@Context HttpHeaders h, String e) { |
149 | 172 | requestCount++; |
@@ -205,6 +228,30 @@ public String deleteFilterWithEntity(@Context HttpHeaders h, String e) { |
205 | 228 |
|
206 | 229 | return e; |
207 | 230 | } |
| 231 | + |
| 232 | + @GET |
| 233 | + @Path("queryParamsBasic") |
| 234 | + public String getQueryParamsBasic(@Context HttpHeaders h, @Context UriInfo uriDetails) { |
| 235 | + queryParamsBasicRequestCount++; |
| 236 | + String value = h.getRequestHeaders().getFirst("Authorization"); |
| 237 | + if (value == null) { |
| 238 | + throw new WebApplicationException( |
| 239 | + Response.status(401).header("WWW-Authenticate", "Basic realm=\"WallyWorld\"").build()); |
| 240 | + } |
| 241 | + return "GET " + queryParamsBasicRequestCount; |
| 242 | + } |
| 243 | + |
| 244 | + @GET |
| 245 | + @Path("queryParamsDigest") |
| 246 | + public String getQueryParamsDigest(@Context HttpHeaders h, @Context UriInfo uriDetails) { |
| 247 | + queryParamsDigestRequestCount++; |
| 248 | + String value = h.getRequestHeaders().getFirst("Authorization"); |
| 249 | + if (value == null) { |
| 250 | + throw new WebApplicationException( |
| 251 | + Response.status(401).header("WWW-Authenticate", "Digest realm=\"WallyWorld\"").build()); |
| 252 | + } |
| 253 | + return "GET " + queryParamsDigestRequestCount; |
| 254 | + } |
208 | 255 | } |
209 | 256 |
|
210 | 257 | @Test |
@@ -254,6 +301,35 @@ public void testAuthGetWithClientFilter() { |
254 | 301 | assertEquals("GET", r.request().get(String.class)); |
255 | 302 | } |
256 | 303 |
|
| 304 | + @Test |
| 305 | + public void testAuthGetBasicNoChallenge() { |
| 306 | + ClientConfig cc = new ClientConfig(); |
| 307 | + cc.connectorProvider(new ApacheConnectorProvider()); |
| 308 | + Client client = ClientBuilder.newClient(cc); |
| 309 | + client.register(HttpAuthenticationFeature.basicBuilder().build()); |
| 310 | + WebTarget r = client.target(getBaseUri()).path("test/noauth"); |
| 311 | + |
| 312 | + assertEquals("GET", r.request().get(String.class)); |
| 313 | + } |
| 314 | + |
| 315 | + @Test |
| 316 | + public void testAuthGetWithDigestFilter() { |
| 317 | + ClientConfig cc = new ClientConfig(); |
| 318 | + PoolingHttpClientConnectionManager cm = new PoolingHttpClientConnectionManager(); |
| 319 | + cc.connectorProvider(new ApacheConnectorProvider()); |
| 320 | + cc.property(ApacheClientProperties.CONNECTION_MANAGER, cm); |
| 321 | + Client client = ClientBuilder.newClient(cc); |
| 322 | + client.register(HttpAuthenticationFeature.universal("name", "password")); |
| 323 | + WebTarget r = client.target(getBaseUri()).path("test/digest"); |
| 324 | + |
| 325 | + assertEquals("GET", r.request().get(String.class)); |
| 326 | + |
| 327 | + // Verify the connection that was used for the request is available for reuse |
| 328 | + // and no connections are leased |
| 329 | + assertEquals(cm.getTotalStats().getAvailable(), 1); |
| 330 | + assertEquals(cm.getTotalStats().getLeased(), 0); |
| 331 | + } |
| 332 | + |
257 | 333 | @Test |
258 | 334 | @Ignore("JERSEY-1750: Cannot retry request with a non-repeatable request entity. How to buffer the entity?" |
259 | 335 | + " Allow repeatable write in jersey?") |
@@ -348,4 +424,40 @@ public void testAuthInteractivePost() { |
348 | 424 |
|
349 | 425 | assertEquals("POST", r.request().post(Entity.text("POST"), String.class)); |
350 | 426 | } |
| 427 | + |
| 428 | + @Test |
| 429 | + public void testAuthGetQueryParamsBasic() { |
| 430 | + ClientConfig cc = new ClientConfig(); |
| 431 | + cc.connectorProvider(new ApacheConnectorProvider()); |
| 432 | + Client client = ClientBuilder.newClient(cc); |
| 433 | + client.register(HttpAuthenticationFeature.universal("name", "password")); |
| 434 | + |
| 435 | + WebTarget r = client.target(getBaseUri()).path("test/queryParamsBasic"); |
| 436 | + assertEquals("GET 2", r.request().get(String.class)); |
| 437 | + |
| 438 | + r = client.target(getBaseUri()) |
| 439 | + .path("test/queryParamsBasic") |
| 440 | + .queryParam("param1", "value1") |
| 441 | + .queryParam("param2", "value2"); |
| 442 | + assertEquals("GET 3", r.request().get(String.class)); |
| 443 | + |
| 444 | + } |
| 445 | + |
| 446 | + @Test |
| 447 | + public void testAuthGetQueryParamsDigest() { |
| 448 | + ClientConfig cc = new ClientConfig(); |
| 449 | + cc.connectorProvider(new ApacheConnectorProvider()); |
| 450 | + Client client = ClientBuilder.newClient(cc); |
| 451 | + client.register(HttpAuthenticationFeature.universal("name", "password")); |
| 452 | + |
| 453 | + WebTarget r = client.target(getBaseUri()).path("test/queryParamsDigest"); |
| 454 | + assertEquals("GET 2", r.request().get(String.class)); |
| 455 | + |
| 456 | + r = client.target(getBaseUri()) |
| 457 | + .path("test/queryParamsDigest") |
| 458 | + .queryParam("param1", "value1") |
| 459 | + .queryParam("param2", "value2"); |
| 460 | + assertEquals("GET 3", r.request().get(String.class)); |
| 461 | + |
| 462 | + } |
351 | 463 | } |
0 commit comments