1 files changed, 135 insertions, 0 deletions
diff --git a/catalog-be/src/test/java/org/openecomp/sdc/be/components/distribution/engine/rest/MsoRestClientTest.java b/catalog-be/src/test/java/org/openecomp/sdc/be/components/distribution/engine/rest/MsoRestClientTest.java
new file mode 100644
index 0000000000..9d0e8cd683
--- /dev/null
+++ b/catalog-be/src/test/java/org/openecomp/sdc/be/components/distribution/engine/rest/MsoRestClientTest.java
@@ -0,0 +1,135 @@
+package org.openecomp.sdc.be.components.distribution.engine.rest;
+
+import com.github.tomakehurst.wiremock.client.ResponseDefinitionBuilder;
+import com.github.tomakehurst.wiremock.junit.WireMockRule;
+import com.github.tomakehurst.wiremock.matching.AnythingPattern;
+import com.github.tomakehurst.wiremock.matching.UrlPattern;
+import fj.data.Either;
+import org.apache.http.HttpHeaders;
+import org.junit.Before;
+import org.junit.ClassRule;
+import org.junit.Test;
+import org.openecomp.sdc.be.components.distribution.engine.DistributionStatusNotificationEnum;
+import org.openecomp.sdc.be.components.distribution.engine.DummyDistributionConfigurationManager;
+import org.openecomp.sdc.be.config.ConfigurationManager;
+import org.openecomp.sdc.common.http.client.api.HttpResponse;
+import org.openecomp.sdc.common.http.config.*;
+import org.openecomp.sdc.security.SecurityUtil;
+
+import static com.github.tomakehurst.wiremock.client.WireMock.*;
+import static com.github.tomakehurst.wiremock.core.WireMockConfiguration.options;
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.mockito.Mockito.when;
+
+public class MsoRestClientTest {
+
+    private static final String MSO_HOST = "127.0.0.1";
+    private static final String MSO_API_URL = "onap/mso/infra/modelDistributions/v1";
+    private static final String DISTRIBUTION_ID = "1000";
+
+    private MSORestClient msoRestClient;
+
+    @ClassRule
+    public static WireMockRule msoRestServer = new WireMockRule(options()
+            .dynamicPort()
+            .bindAddress(MSO_HOST));
+
+    @Before
+    public void setupMsoServer() throws Exception {
+        String encodedPw = "";
+        Either<String, String> passkey = SecurityUtil.INSTANCE.decrypt(getPwd());
+        if(passkey.isLeft()) {
+            encodedPw = passkey.left().value();
+        }
+        else {
+            throw new IllegalArgumentException(passkey.right().value());
+        }
+        
+        msoRestServer.resetToDefaultMappings();
+        msoRestServer.stubFor(patch(urlMatching(String.format("/%s%s/(.*)", MSO_API_URL, getDistributionsUrl())))
+                .withBasicAuth(getUserName(), encodedPw)
+                .withHeader(HttpHeaders.CONTENT_TYPE, containing("application/json"))
+                .withRequestBody(matchingJsonPath("$.status"))
+                .withRequestBody(matchingJsonPath("$.errorReason", new AnythingPattern()))//error reason is not mandatory
+                .willReturn(aResponse().withStatus(200)));
+    }
+
+    @Before
+    public void setUp() throws Exception {
+        DummyDistributionConfigurationManager distributionEngineConfigurationMock = new DummyDistributionConfigurationManager();
+        when(distributionEngineConfigurationMock.getConfigurationMock().getMsoConfig()).thenReturn(new MsoDummyConfig(msoRestServer.port()));
+        msoRestClient = new MSORestClient();
+    }
+
+    @Test
+    public void notifyDistributionComplete_emptyErrReason() throws Exception {
+        HttpResponse<String> response = msoRestClient.notifyDistributionComplete(DISTRIBUTION_ID, DistributionStatusNotificationEnum.DISTRIBUTION_COMPLETE_OK, "");
+        assertThat(response.getStatusCode()).isEqualTo(200);
+    }
+
+    private int getNumOfRetries() {
+        return ConfigurationManager.getConfigurationManager().getDistributionEngineConfiguration().getMsoConfig().getHttpClientConfig().getNumOfRetries();
+    }
+
+    @Test
+    public void notifyDistributionComplete_noErrReason() throws Exception {
+        HttpResponse<String> response = msoRestClient.notifyDistributionComplete(DISTRIBUTION_ID, DistributionStatusNotificationEnum.DISTRIBUTION_COMPLETE_OK, null);
+        assertThat(response.getStatusCode()).isEqualTo(200);
+    }
+
+    @Test
+    public void notifyDistributionComplete_completeWithError() throws Exception {
+        HttpResponse<String> response = msoRestClient.notifyDistributionComplete(DISTRIBUTION_ID, DistributionStatusNotificationEnum.DISTRIBUTION_COMPLETE_ERROR, "my reason");
+        assertThat(response.getStatusCode()).isEqualTo(200);
+    }
+
+    @Test
+    public void testRetries() throws Exception {
+        int readTimeout = ConfigurationManager.getConfigurationManager().getDistributionEngineConfiguration().getMsoConfig().getHttpClientConfig().getTimeouts().getReadTimeoutMs();
+        int expectedNumOfRetries = getNumOfRetries();
+
+        UrlPattern msoReqUrlPattern = urlMatching(String.format("/%s%s/(.*)", MSO_API_URL, getDistributionsUrl()));
+        msoRestServer.stubFor(patch(msoReqUrlPattern)
+                .willReturn(new ResponseDefinitionBuilder().withFixedDelay(readTimeout + 1)));
+        HttpResponse<String> response = msoRestClient.notifyDistributionComplete(DISTRIBUTION_ID, DistributionStatusNotificationEnum.DISTRIBUTION_COMPLETE_ERROR, "my reason");
+        verify(expectedNumOfRetries + 1, patchRequestedFor(msoReqUrlPattern));
+        assertThat(response.getStatusCode()).isEqualTo(500);
+    }
+
+    private static String getDistributionsUrl() {
+        return "/distributions";
+    }
+
+    private static String getUserName() {
+        return "asdc";
+    }
+
+    private static String getPwd() {
+        return "OTLEp5lfVhYdyw5EAtTUBQ==";
+    }
+
+    private static class MsoDummyConfig extends ExternalServiceConfig {
+        int port;
+
+        MsoDummyConfig(int port) {
+            this.port = port;
+
+            BasicAuthorization basicAuthorization = new BasicAuthorization();
+            basicAuthorization.setUserName(MsoRestClientTest.getUserName());
+            basicAuthorization.setPassword(getPwd());
+            HttpClientConfig httpClientConfig = new HttpClientConfig(new Timeouts(500, 2000), basicAuthorization);
+            httpClientConfig.setNumOfRetries(getNumOfRetries());
+            super.setHttpClientConfig(httpClientConfig);
+
+            HttpRequestConfig httpRequestConfig = new HttpRequestConfig();
+            httpRequestConfig.setServerRootUrl(String.format("http://%s:%s/%s", MSO_HOST, this.port, MSO_API_URL));
+            httpRequestConfig.getResourceNamespaces().put(MSORestClient.DISTRIBUTIONS_RESOURCE_CONFIG_PARAM, getDistributionsUrl());
+            super.setHttpRequestConfig(httpRequestConfig);
+        }
+
+        int getNumOfRetries() {
+            return 1;
+        }
+    }
+
+}