1
2
3
4
5
6
7
8
9
10
11
12
13
14
15 package org.htmlunit.util;
16
17 import java.util.Collections;
18
19 import org.htmlunit.HttpMethod;
20 import org.htmlunit.SimpleWebTestCase;
21 import org.htmlunit.WebConnection;
22 import org.htmlunit.WebRequest;
23 import org.htmlunit.WebResponse;
24 import org.htmlunit.WebResponseData;
25 import org.htmlunit.http.HttpStatus;
26 import org.junit.Test;
27
28
29
30
31
32
33 public class WebConnectionWrapperTest extends SimpleWebTestCase {
34
35
36
37
38 @Test
39 public void wrapper() throws Exception {
40 final WebResponseData data = new WebResponseData(new byte[]{},
41 HttpStatus.OK_200, HttpStatus.OK_200_MSG, Collections.emptyList());
42 final WebResponse response = new WebResponse(data, URL_FIRST, HttpMethod.GET, 0);
43 final WebRequest wrs = new WebRequest(URL_FIRST);
44
45 final WebConnection realConnection = new WebConnection() {
46 @Override
47 public WebResponse getResponse(final WebRequest request) {
48 assertSame(wrs, request);
49 return response;
50 }
51 @Override
52 public void close() {
53
54 }
55 };
56
57 try (WebConnectionWrapper wrapper = new WebConnectionWrapper(realConnection)) {
58 assertSame(response, wrapper.getResponse(wrs));
59 }
60 }
61
62 }