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 java.util.Map;
|
||||||
|
|
||||||
|
import org.alfresco.transform.base.MtlsTestUtils;
|
||||||
import org.springframework.core.io.ClassPathResource;
|
import org.springframework.core.io.ClassPathResource;
|
||||||
import org.springframework.core.io.Resource;
|
import org.springframework.core.io.Resource;
|
||||||
import org.springframework.http.HttpEntity;
|
import org.springframework.http.HttpEntity;
|
||||||
@@ -27,7 +28,7 @@ import org.springframework.web.client.RestTemplate;
|
|||||||
*/
|
*/
|
||||||
public class HttpClient
|
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(
|
public static ResponseEntity<Resource> sendTRequest(
|
||||||
final String engineUrl, final String sourceFile,
|
final String engineUrl, final String sourceFile,
|
||||||
|
@@ -21,6 +21,7 @@ import java.nio.file.Files;
|
|||||||
import java.nio.file.Paths;
|
import java.nio.file.Paths;
|
||||||
import java.nio.file.StandardCopyOption;
|
import java.nio.file.StandardCopyOption;
|
||||||
|
|
||||||
|
import org.alfresco.transform.base.MtlsTestUtils;
|
||||||
import org.apache.http.HttpEntity;
|
import org.apache.http.HttpEntity;
|
||||||
import org.apache.http.HttpResponse;
|
import org.apache.http.HttpResponse;
|
||||||
import org.apache.http.client.methods.HttpGet;
|
import org.apache.http.client.methods.HttpGet;
|
||||||
@@ -56,7 +57,7 @@ public class SfsClient
|
|||||||
((Logger) LoggerFactory.getLogger("org.apache.http.wire")).setAdditive(false);
|
((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
|
public static String uploadFile(final String fileToUploadName) throws Exception
|
||||||
{
|
{
|
||||||
@@ -75,7 +76,7 @@ public class SfsClient
|
|||||||
.addPart("file", new FileBody(file, ContentType.DEFAULT_BINARY))
|
.addPart("file", new FileBody(file, ContentType.DEFAULT_BINARY))
|
||||||
.build());
|
.build());
|
||||||
|
|
||||||
try (CloseableHttpClient client = HttpClients.createDefault())
|
try (CloseableHttpClient client = MtlsTestUtils.isMtlsEnabled() ? MtlsTestUtils.httpClientWithMtls() : HttpClients.createDefault())
|
||||||
{
|
{
|
||||||
final HttpResponse response = client.execute(post);
|
final HttpResponse response = client.execute(post);
|
||||||
int status = response.getStatusLine().getStatusCode();
|
int status = response.getStatusLine().getStatusCode();
|
||||||
@@ -134,7 +135,8 @@ public class SfsClient
|
|||||||
sfsBaseUrl+"/alfresco/api/-default-/private/sfs/versions/1/file/{0}",
|
sfsBaseUrl+"/alfresco/api/-default-/private/sfs/versions/1/file/{0}",
|
||||||
uuid));
|
uuid));
|
||||||
|
|
||||||
try (CloseableHttpClient client = HttpClients.createDefault())
|
try (CloseableHttpClient client = MtlsTestUtils.isMtlsEnabled() ?
|
||||||
|
MtlsTestUtils.httpClientWithMtls() : HttpClients.createDefault())
|
||||||
{
|
{
|
||||||
final HttpResponse response = client.execute(head);
|
final HttpResponse response = client.execute(head);
|
||||||
final int status = response.getStatusLine().getStatusCode();
|
final int status = response.getStatusLine().getStatusCode();
|
||||||
|
Reference in New Issue
Block a user