1 files changed, 116 insertions, 0 deletions
diff --git a/common/src/test/java/org/openecomp/mso/client/sdno/SDNOValidatorTest.java b/common/src/test/java/org/openecomp/mso/client/sdno/SDNOValidatorTest.java
new file mode 100644
index 0000000000..cd84e7854d
--- /dev/null
+++ b/common/src/test/java/org/openecomp/mso/client/sdno/SDNOValidatorTest.java
@@ -0,0 +1,116 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * ONAP - SO
+ * ================================================================================
+ * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.openecomp.mso.client.sdno;
+
+import static org.junit.Assert.assertEquals;
+import static org.mockito.Matchers.any;
+import static org.mockito.Mockito.doReturn;
+import static org.mockito.Mockito.spy;
+import static org.mockito.Mockito.when;
+
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Paths;
+import java.util.Arrays;
+import java.util.UUID;
+
+import org.hamcrest.core.StringContains;
+import org.junit.Before;
+import org.junit.Ignore;
+import org.junit.Rule;
+import org.junit.Test;
+import org.junit.rules.ExpectedException;
+import org.mockito.Mock;
+import org.mockito.MockitoAnnotations;
+import org.mockito.Spy;
+import org.onap.aai.domain.yang.GenericVnf;
+import org.openecomp.mso.client.dmaap.Consumer;
+import org.openecomp.mso.client.dmaap.exceptions.DMaaPConsumerFailure;
+import org.openecomp.mso.client.exceptions.SDNOException;
+import org.openecomp.mso.client.sdno.beans.SDNO;
+import org.openecomp.mso.client.sdno.dmaap.SDNOHealthCheckDmaapConsumer;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+
+
+public class SDNOValidatorTest {
+
+	
+	@Mock private Consumer mrConsumer;
+	@Spy private SDNOHealthCheckDmaapConsumer dmaapConsumer;
+	private final String fileLocation = "src/test/resources/org/openecomp/mso/client/sdno/";
+	private final String uuid = "xyz123";
+	@Rule public ExpectedException thrown = ExpectedException.none();
+	
+	@Before
+	public void setUpTests() {
+		MockitoAnnotations.initMocks(this);
+	}
+	
+	@Test
+	public void success() throws IOException, Exception {
+		when(dmaapConsumer.getConsumer()).thenReturn(mrConsumer);
+		when(mrConsumer.fetch()).thenReturn(Arrays.asList(new String[]{getJson("response.json"), getJson("output-success.json")}));
+		
+		SDNOValidatorImpl validator = new SDNOValidatorImpl();
+		SDNOValidatorImpl spy = spy(validator);
+		when(dmaapConsumer.getRequestId()).thenReturn("xyz123");
+		doReturn(dmaapConsumer).when(spy).getConsumer(any(String.class));
+		boolean result = spy.pollForResponse("xyz123");
+		assertEquals("result is true", result, true);
+	}
+	
+	@Test
+	public void failure() throws IOException, Exception {
+		when(dmaapConsumer.getConsumer()).thenReturn(mrConsumer);
+		when(mrConsumer.fetch()).thenReturn(Arrays.asList(new String[]{getJson("response.json"), getJson("output-failure.json")}));
+		
+		SDNOValidatorImpl validator = new SDNOValidatorImpl();
+		SDNOValidatorImpl spy = spy(validator);
+		when(dmaapConsumer.getRequestId()).thenReturn("xyz123");
+		doReturn(dmaapConsumer).when(spy).getConsumer(any(String.class));
+		thrown.expect(SDNOException.class);
+		thrown.expectMessage(new StringContains("my error message"));
+		boolean result = spy.pollForResponse("xyz123");
+		
+	}
+	@Ignore
+	@Test
+	public void run() throws Exception {
+		SDNOValidatorImpl validator = new SDNOValidatorImpl();
+		UUID uuid = UUID.randomUUID();
+		GenericVnf vnf = new GenericVnf();
+		vnf.setVnfId("test");
+		vnf.setIpv4OamAddress("1.2.3.4");
+		vnf.setVnfType("VPE");
+		SDNO request = validator.buildRequestDiagnostic(vnf, uuid, "mechid");
+		ObjectMapper mapper = new ObjectMapper();
+		String json = mapper.writeValueAsString(request);
+		validator.submitRequest(json);
+		thrown.expect(DMaaPConsumerFailure.class);
+		boolean result = validator.pollForResponse(uuid.toString());
+		System.out.println(json);
+
+	}
+	private String getJson(String filename) throws IOException {
+		return new String(Files.readAllBytes(Paths.get(fileLocation + filename)));
+	}
+}