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.core5.testing.extension;
29
30 import org.apache.hc.core5.testing.SocksProxy;
31 import org.apache.hc.core5.util.TimeValue;
32 import org.junit.jupiter.api.Assertions;
33 import org.junit.jupiter.api.extension.AfterEachCallback;
34 import org.junit.jupiter.api.extension.BeforeEachCallback;
35 import org.junit.jupiter.api.extension.ExtensionContext;
36 import org.slf4j.Logger;
37 import org.slf4j.LoggerFactory;
38
39 public class SocksProxyResource implements BeforeEachCallback, AfterEachCallback {
40
41 private static final Logger LOG = LoggerFactory.getLogger(SocksProxyResource.class);
42
43 private SocksProxy proxy;
44
45 @Override
46 public void beforeEach(final ExtensionContext extensionContext) throws Exception {
47 LOG.debug("Starting up SOCKS proxy");
48 proxy = new SocksProxy();
49 proxy.start();
50 }
51
52 @Override
53 public void afterEach(final ExtensionContext extensionContext) throws Exception {
54 LOG.debug("Shutting down SOCKS proxy");
55 if (proxy != null) {
56 try {
57 proxy.shutdown(TimeValue.ofSeconds(5));
58 } catch (final Exception ignore) {
59 }
60 }
61 }
62
63 public SocksProxy proxy() {
64 Assertions.assertNotNull(proxy);
65 return proxy;
66 }
67
68 }