1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28 package org.apache.hc.client5.testing.extension.sync;
29
30 import org.apache.hc.client5.http.config.ConnectionConfig;
31 import org.apache.hc.client5.http.impl.classic.HttpClients;
32 import org.apache.hc.client5.http.impl.classic.MinimalHttpClient;
33 import org.apache.hc.client5.http.impl.io.PoolingHttpClientConnectionManagerBuilder;
34 import org.apache.hc.client5.http.io.HttpClientConnectionManager;
35 import org.apache.hc.client5.http.ssl.DefaultClientTlsStrategy;
36 import org.apache.hc.client5.testing.SSLTestContexts;
37 import org.apache.hc.core5.http.io.SocketConfig;
38 import org.apache.hc.core5.util.Timeout;
39
40 final class MinimalTestClientBuilder implements TestClientBuilder {
41
42 private Timeout timeout;
43
44 private HttpClientConnectionManager connectionManager;
45
46 public MinimalTestClientBuilder() {
47 }
48
49 @Override
50 public ClientProtocolLevel getProtocolLevel() {
51 return ClientProtocolLevel.MINIMAL;
52 }
53
54 @Override
55 public TestClientBuilder setTimeout(final Timeout timeout) {
56 this.timeout = timeout;
57 return this;
58 }
59
60 @Override
61 public TestClientBuilder setConnectionManager(final HttpClientConnectionManager connectionManager) {
62 this.connectionManager = connectionManager;
63 return this;
64 }
65
66 @Override
67 public TestClient build() throws Exception {
68 final HttpClientConnectionManager connectionManagerCopy = connectionManager != null ? connectionManager :
69 PoolingHttpClientConnectionManagerBuilder.create()
70 .setTlsSocketStrategy(new DefaultClientTlsStrategy(SSLTestContexts.createClientSSLContext()))
71 .setDefaultSocketConfig(SocketConfig.custom()
72 .setSoTimeout(timeout)
73 .build())
74 .setDefaultConnectionConfig(ConnectionConfig.custom()
75 .setConnectTimeout(timeout)
76 .build())
77 .build();
78
79 final MinimalHttpClient minimalClient = HttpClients.createMinimal(connectionManagerCopy);
80 return new TestClient(minimalClient, connectionManagerCopy);
81 }
82
83 }