1 package org.openecomp.sdc.be.components.distribution.engine.rest;
3 import static com.github.tomakehurst.wiremock.client.WireMock.aResponse;
4 import static com.github.tomakehurst.wiremock.client.WireMock.containing;
5 import static com.github.tomakehurst.wiremock.client.WireMock.matchingJsonPath;
6 import static com.github.tomakehurst.wiremock.client.WireMock.patch;
7 import static com.github.tomakehurst.wiremock.client.WireMock.patchRequestedFor;
8 import static com.github.tomakehurst.wiremock.client.WireMock.urlMatching;
9 import static com.github.tomakehurst.wiremock.client.WireMock.verify;
10 import static com.github.tomakehurst.wiremock.core.WireMockConfiguration.options;
11 import static org.assertj.core.api.Assertions.assertThat;
12 import static org.mockito.Mockito.when;
14 import org.apache.http.HttpHeaders;
15 import org.junit.Before;
16 import org.junit.ClassRule;
17 import org.junit.Test;
18 import org.openecomp.sdc.be.components.distribution.engine.DistributionStatusNotificationEnum;
19 import org.openecomp.sdc.be.components.distribution.engine.DummyDistributionConfigurationManager;
20 import org.openecomp.sdc.be.config.ConfigurationManager;
21 import org.openecomp.sdc.common.http.client.api.HttpResponse;
22 import org.openecomp.sdc.common.http.config.BasicAuthorization;
23 import org.openecomp.sdc.common.http.config.ExternalServiceConfig;
24 import org.openecomp.sdc.common.http.config.HttpClientConfig;
25 import org.openecomp.sdc.common.http.config.HttpRequestConfig;
26 import org.openecomp.sdc.common.http.config.Timeouts;
27 import org.openecomp.sdc.security.SecurityUtil;
29 import com.github.tomakehurst.wiremock.client.ResponseDefinitionBuilder;
30 import com.github.tomakehurst.wiremock.junit.WireMockRule;
31 import com.github.tomakehurst.wiremock.matching.AnythingPattern;
32 import com.github.tomakehurst.wiremock.matching.UrlPattern;
34 import fj.data.Either;
36 public class MsoRestClientTest {
38 private static final String MSO_HOST = "127.0.0.1";
39 private static final String MSO_API_URL = "onap/mso/infra/modelDistributions/v1";
40 private static final String DISTRIBUTION_ID = "1000";
42 private MSORestClient msoRestClient;
45 public static WireMockRule msoRestServer = new WireMockRule(options()
47 .bindAddress(MSO_HOST));
50 public void setupMsoServer() throws Exception {
51 String encodedPw = "";
52 Either<String, String> passkey = SecurityUtil.INSTANCE.decrypt(getPwd());
53 if(passkey.isLeft()) {
54 encodedPw = passkey.left().value();
57 throw new IllegalArgumentException(passkey.right().value());
60 msoRestServer.resetToDefaultMappings();
61 msoRestServer.stubFor(patch(urlMatching(String.format("/%s%s/(.*)", MSO_API_URL, getDistributionsUrl())))
62 .withBasicAuth(getUserName(), encodedPw)
63 .withHeader(HttpHeaders.CONTENT_TYPE, containing("application/json"))
64 .withRequestBody(matchingJsonPath("$.status"))
65 .withRequestBody(matchingJsonPath("$.errorReason", new AnythingPattern()))//error reason is not mandatory
66 .willReturn(aResponse().withStatus(200)));
70 public void setUp() throws Exception {
71 DummyDistributionConfigurationManager distributionEngineConfigurationMock = new DummyDistributionConfigurationManager();
72 when(distributionEngineConfigurationMock.getConfigurationMock().getMsoConfig()).thenReturn(new MsoDummyConfig(msoRestServer.port()));
73 msoRestClient = new MSORestClient();
77 public void notifyDistributionComplete_emptyErrReason() throws Exception {
78 HttpResponse<String> response = msoRestClient.notifyDistributionComplete(DISTRIBUTION_ID, DistributionStatusNotificationEnum.DISTRIBUTION_COMPLETE_OK, "");
79 assertThat(response.getStatusCode()).isEqualTo(200);
82 private int getNumOfRetries() {
83 return ConfigurationManager.getConfigurationManager().getDistributionEngineConfiguration().getMsoConfig().getHttpClientConfig().getNumOfRetries();
87 public void notifyDistributionComplete_noErrReason() throws Exception {
88 HttpResponse<String> response = msoRestClient.notifyDistributionComplete(DISTRIBUTION_ID, DistributionStatusNotificationEnum.DISTRIBUTION_COMPLETE_OK, null);
89 assertThat(response.getStatusCode()).isEqualTo(200);
93 public void notifyDistributionComplete_completeWithError() throws Exception {
94 HttpResponse<String> response = msoRestClient.notifyDistributionComplete(DISTRIBUTION_ID, DistributionStatusNotificationEnum.DISTRIBUTION_COMPLETE_ERROR, "my reason");
95 assertThat(response.getStatusCode()).isEqualTo(200);
99 public void testRetries() throws Exception {
100 int readTimeout = ConfigurationManager.getConfigurationManager().getDistributionEngineConfiguration().getMsoConfig().getHttpClientConfig().getTimeouts().getReadTimeoutMs();
101 int expectedNumOfRetries = getNumOfRetries();
103 UrlPattern msoReqUrlPattern = urlMatching(String.format("/%s%s/(.*)", MSO_API_URL, getDistributionsUrl()));
104 msoRestServer.stubFor(patch(msoReqUrlPattern)
105 .willReturn(new ResponseDefinitionBuilder().withFixedDelay(readTimeout + 1)));
106 HttpResponse<String> response = msoRestClient.notifyDistributionComplete(DISTRIBUTION_ID, DistributionStatusNotificationEnum.DISTRIBUTION_COMPLETE_ERROR, "my reason");
107 verify(expectedNumOfRetries + 1, patchRequestedFor(msoReqUrlPattern));
108 assertThat(response.getStatusCode()).isEqualTo(500);
111 private static String getDistributionsUrl() {
112 return "/distributions";
115 private static String getUserName() {
119 private static String getPwd() {
120 return "OTLEp5lfVhYdyw5EAtTUBQ==";
123 private static class MsoDummyConfig extends ExternalServiceConfig {
126 MsoDummyConfig(int port) {
129 BasicAuthorization basicAuthorization = new BasicAuthorization();
130 basicAuthorization.setUserName(MsoRestClientTest.getUserName());
131 basicAuthorization.setPassword(getPwd());
132 HttpClientConfig httpClientConfig = new HttpClientConfig(new Timeouts(500, 2000), basicAuthorization);
133 httpClientConfig.setNumOfRetries(getNumOfRetries());
134 super.setHttpClientConfig(httpClientConfig);
136 HttpRequestConfig httpRequestConfig = new HttpRequestConfig();
137 httpRequestConfig.setServerRootUrl(String.format("http://%s:%s/%s", MSO_HOST, this.port, MSO_API_URL));
138 httpRequestConfig.getResourceNamespaces().put(MSORestClient.DISTRIBUTIONS_RESOURCE_CONFIG_PARAM, getDistributionsUrl());
139 super.setHttpRequestConfig(httpRequestConfig);
142 int getNumOfRetries() {