mirror of
https://github.com/Alfresco/alfresco-transform-core.git
synced 2025-08-07 17:48:35 +00:00
[ACS-4460] integration tests fix
This commit is contained in:
@@ -0,0 +1,69 @@
|
||||
package org.alfresco.transform.base;
|
||||
|
||||
import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
|
||||
import org.apache.http.conn.ssl.TrustAllStrategy;
|
||||
import org.apache.http.impl.client.CloseableHttpClient;
|
||||
import org.apache.http.impl.client.HttpClients;
|
||||
import org.apache.http.ssl.SSLContextBuilder;
|
||||
import org.springframework.http.client.ClientHttpRequestFactory;
|
||||
import org.springframework.http.client.HttpComponentsClientHttpRequestFactory;
|
||||
import org.springframework.web.client.RestTemplate;
|
||||
|
||||
import javax.net.ssl.SSLContext;
|
||||
import java.io.FileInputStream;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.security.KeyManagementException;
|
||||
import java.security.KeyStore;
|
||||
import java.security.KeyStoreException;
|
||||
import java.security.NoSuchAlgorithmException;
|
||||
import java.security.UnrecoverableKeyException;
|
||||
import java.security.cert.CertificateException;
|
||||
|
||||
public class MtlsTestUtils {
|
||||
|
||||
public static boolean isMtlsEnabled()
|
||||
{
|
||||
return Boolean.parseBoolean(System.getProperty("mtls-enabled"));
|
||||
}
|
||||
|
||||
public static CloseableHttpClient httpClientWithMtls() throws NoSuchAlgorithmException, KeyManagementException, UnrecoverableKeyException, KeyStoreException, IOException, CertificateException {
|
||||
|
||||
String keyStoreFile = System.getProperty("mtls-keystore-file");
|
||||
String keyStoreType = System.getProperty("mtls-keystore-type");
|
||||
char[] keyStorePassword = System.getProperty("mtls-keystore-password").toCharArray();
|
||||
String trustStoreFile = System.getProperty("mtls-truststore-file");
|
||||
String trustStoreType = System.getProperty("mtls-truststore-type");
|
||||
char[] trustStorePassword = System.getProperty("mtls-truststore-password").toCharArray();
|
||||
|
||||
SSLContextBuilder sslContextBuilder = new SSLContextBuilder();
|
||||
KeyStore keyStore = KeyStore.getInstance(keyStoreType);
|
||||
try (InputStream keyStoreInputStream = new FileInputStream(keyStoreFile))
|
||||
{
|
||||
keyStore.load(keyStoreInputStream, keyStorePassword);
|
||||
sslContextBuilder.loadKeyMaterial(keyStore, keyStorePassword);
|
||||
}
|
||||
|
||||
KeyStore trustStore = KeyStore.getInstance(trustStoreType);
|
||||
try (InputStream trustStoreInputStream = new FileInputStream(trustStoreFile))
|
||||
{
|
||||
trustStore.load(trustStoreInputStream, trustStorePassword);
|
||||
sslContextBuilder.loadTrustMaterial(trustStore, TrustAllStrategy.INSTANCE);
|
||||
}
|
||||
|
||||
SSLContext sslContext = sslContextBuilder.build();
|
||||
SSLConnectionSocketFactory sslContextFactory = new SSLConnectionSocketFactory(sslContext);
|
||||
return HttpClients.custom().setSSLSocketFactory(sslContextFactory).build();
|
||||
}
|
||||
|
||||
public static RestTemplate restTemplateWithMtls()
|
||||
{
|
||||
ClientHttpRequestFactory requestFactory = null;
|
||||
try {
|
||||
requestFactory = new HttpComponentsClientHttpRequestFactory(httpClientWithMtls());
|
||||
} catch (Exception e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
return new RestTemplate(requestFactory);
|
||||
}
|
||||
}
|
@@ -13,6 +13,7 @@ import static org.springframework.http.MediaType.MULTIPART_FORM_DATA;
|
||||
|
||||
import java.util.Map;
|
||||
|
||||
import org.alfresco.transform.base.MtlsTestUtils;
|
||||
import org.springframework.core.io.ClassPathResource;
|
||||
import org.springframework.core.io.Resource;
|
||||
import org.springframework.http.HttpEntity;
|
||||
@@ -27,7 +28,7 @@ import org.springframework.web.client.RestTemplate;
|
||||
*/
|
||||
public class HttpClient
|
||||
{
|
||||
private static final RestTemplate REST_TEMPLATE = new RestTemplate();
|
||||
private static final RestTemplate REST_TEMPLATE = MtlsTestUtils.isMtlsEnabled() ? MtlsTestUtils.restTemplateWithMtls() : new RestTemplate();
|
||||
|
||||
public static ResponseEntity<Resource> sendTRequest(
|
||||
final String engineUrl, final String sourceFile,
|
||||
|
@@ -21,6 +21,7 @@ import java.nio.file.Files;
|
||||
import java.nio.file.Paths;
|
||||
import java.nio.file.StandardCopyOption;
|
||||
|
||||
import org.alfresco.transform.base.MtlsTestUtils;
|
||||
import org.apache.http.HttpEntity;
|
||||
import org.apache.http.HttpResponse;
|
||||
import org.apache.http.client.methods.HttpGet;
|
||||
@@ -56,7 +57,7 @@ public class SfsClient
|
||||
((Logger) LoggerFactory.getLogger("org.apache.http.wire")).setAdditive(false);
|
||||
}
|
||||
|
||||
private static final String SFS_BASE_URL = "http://localhost:8099";
|
||||
private static final String SFS_BASE_URL = MtlsTestUtils.isMtlsEnabled() ? "https://localhost:8099" : "http://localhost:8099";
|
||||
|
||||
public static String uploadFile(final String fileToUploadName) throws Exception
|
||||
{
|
||||
@@ -75,7 +76,7 @@ public class SfsClient
|
||||
.addPart("file", new FileBody(file, ContentType.DEFAULT_BINARY))
|
||||
.build());
|
||||
|
||||
try (CloseableHttpClient client = HttpClients.createDefault())
|
||||
try (CloseableHttpClient client = MtlsTestUtils.isMtlsEnabled() ? MtlsTestUtils.httpClientWithMtls() : HttpClients.createDefault())
|
||||
{
|
||||
final HttpResponse response = client.execute(post);
|
||||
int status = response.getStatusLine().getStatusCode();
|
||||
@@ -134,7 +135,8 @@ public class SfsClient
|
||||
sfsBaseUrl+"/alfresco/api/-default-/private/sfs/versions/1/file/{0}",
|
||||
uuid));
|
||||
|
||||
try (CloseableHttpClient client = HttpClients.createDefault())
|
||||
try (CloseableHttpClient client = MtlsTestUtils.isMtlsEnabled() ?
|
||||
MtlsTestUtils.httpClientWithMtls() : HttpClients.createDefault())
|
||||
{
|
||||
final HttpResponse response = client.execute(head);
|
||||
final int status = response.getStatusLine().getStatusCode();
|
||||
|
Reference in New Issue
Block a user