View Javadoc
1   /*
2    * ====================================================================
3    * Licensed to the Apache Software Foundation (ASF) under one
4    * or more contributor license agreements.  See the NOTICE file
5    * distributed with this work for additional information
6    * regarding copyright ownership.  The ASF licenses this file
7    * to you under the Apache License, Version 2.0 (the
8    * "License"); you may not use this file except in compliance
9    * with the License.  You may obtain a copy of the License at
10   *
11   *   http://www.apache.org/licenses/LICENSE-2.0
12   *
13   * Unless required by applicable law or agreed to in writing,
14   * software distributed under the License is distributed on an
15   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
16   * KIND, either express or implied.  See the License for the
17   * specific language governing permissions and limitations
18   * under the License.
19   * ====================================================================
20   *
21   * This software consists of voluntary contributions made by many
22   * individuals on behalf of the Apache Software Foundation.  For more
23   * information on the Apache Software Foundation, please see
24   * <http://www.apache.org/>.
25   *
26   */
27  
28  package org.apache.hc.client5.testing.extension.async;
29  
30  import org.apache.hc.client5.http.impl.async.CloseableHttpAsyncClient;
31  import org.apache.hc.client5.http.impl.async.HttpAsyncClients;
32  import org.apache.hc.client5.http.ssl.DefaultClientTlsStrategy;
33  import org.apache.hc.client5.testing.SSLTestContexts;
34  import org.apache.hc.core5.http.nio.ssl.TlsStrategy;
35  import org.apache.hc.core5.http2.config.H2Config;
36  import org.apache.hc.core5.reactor.IOReactorConfig;
37  import org.apache.hc.core5.util.Timeout;
38  
39  final class H2OnlyMinimalTestClientBuilder implements TestAsyncClientBuilder {
40  
41      private Timeout timeout;
42      private TlsStrategy tlsStrategy;
43      private H2Config h2Config;
44  
45      public H2OnlyMinimalTestClientBuilder() {
46      }
47  
48      @Override
49      public TestAsyncClientBuilder setTimeout(final Timeout timeout) {
50          this.timeout = timeout;
51          return this;
52      }
53  
54      @Override
55      public TestAsyncClientBuilder setTlsStrategy(final TlsStrategy tlsStrategy) {
56          this.tlsStrategy = tlsStrategy;
57          return this;
58      }
59  
60      @Override
61      public ClientProtocolLevel getProtocolLevel() {
62          return ClientProtocolLevel.MINIMAL_H2_ONLY;
63      }
64  
65      @Override
66      public TestAsyncClientBuilder setH2Config(final H2Config h2Config) {
67          this.h2Config = h2Config;
68          return this;
69      }
70  
71      @Override
72      public TestAsyncClient build() throws Exception {
73          final CloseableHttpAsyncClient client = HttpAsyncClients.createHttp2Minimal(
74                  h2Config,
75                  IOReactorConfig.custom()
76                          .setSoTimeout(timeout)
77                          .build(),
78                  tlsStrategy != null ? tlsStrategy : new DefaultClientTlsStrategy(SSLTestContexts.createClientSSLContext()));
79          return new TestAsyncClient(client, null);
80      }
81  
82  }