X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=cadi%2Faaf%2Fsrc%2Ftest%2Fjava%2Forg%2Fonap%2Faaf%2Fcadi%2Fcm%2Ftest%2FJU_PlaceArtifactInKeystore.java;fp=cadi%2Faaf%2Fsrc%2Ftest%2Fjava%2Forg%2Fonap%2Faaf%2Fcadi%2Fcm%2Ftest%2FJU_PlaceArtifactInKeystore.java;h=02054cbfb00dbfc640d8bcf3081b55aa81668364;hb=a0b791ea5f276087f25b31ba15723162a6d1ca38;hp=0000000000000000000000000000000000000000;hpb=52663b5825635395c423aa65f16833c361ba183b;p=aaf%2Fauthz.git

diff --git a/cadi/aaf/src/test/java/org/onap/aaf/cadi/cm/test/JU_PlaceArtifactInKeystore.java b/cadi/aaf/src/test/java/org/onap/aaf/cadi/cm/test/JU_PlaceArtifactInKeystore.java
new file mode 100644
index 00000000..02054cbf
--- /dev/null
+++ b/cadi/aaf/src/test/java/org/onap/aaf/cadi/cm/test/JU_PlaceArtifactInKeystore.java
@@ -0,0 +1,144 @@
+/**
+ * ============LICENSE_START====================================================
+ * org.onap.aaf
+ * ===========================================================================
+ * Copyright (c) 2018 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.onap.aaf.cadi.cm.test;
+
+import static org.junit.Assert.*;
+import static org.hamcrest.CoreMatchers.*;
+import static org.mockito.Mockito.*;
+
+import java.io.BufferedReader;
+import java.io.ByteArrayOutputStream;
+import java.io.File;
+import java.io.FileNotFoundException;
+import java.io.FileReader;
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.List;
+
+import java.security.cert.CertificateException;
+
+import org.junit.*;
+import org.mockito.*;
+import org.onap.aaf.cadi.CadiException;
+import org.onap.aaf.cadi.cm.PlaceArtifactInKeystore;
+import org.onap.aaf.misc.env.Env;
+import org.onap.aaf.misc.env.TimeTaken;
+import org.onap.aaf.misc.env.Trans;
+
+import certman.v1_0.Artifacts.Artifact;
+import certman.v1_0.CertInfo;
+
+public class JU_PlaceArtifactInKeystore {
+
+	@Mock private Trans transMock;
+	@Mock private CertInfo certInfoMock;
+	@Mock private Artifact artiMock;
+
+	private static final String dirName = "src/test/resources/artifacts";
+	private static final String nsName = "org.onap.test";
+	private static final String mechID = "m12345";
+	private static final String luggagePassword = "12345";  // That's the stupidest combination I've ever heard in my life
+
+	private static String privateKeyString;
+	private static String x509Chain;
+	private static String x509String;
+
+	private List<String> certs;
+
+	@Before
+	public void setup() throws FileNotFoundException, IOException, CertificateException {
+		MockitoAnnotations.initMocks(this);
+
+		x509Chain = fromFile(new File("src/test/resources/cert.pem"));
+		x509String = fromFile(new File("src/test/resources/exampleCertificate.cer"));
+		privateKeyString = fromFile(new File("src/test/resources/key.pem"));
+
+		certs = new ArrayList<>();
+
+		when(certInfoMock.getChallenge()).thenReturn(luggagePassword);
+		when(certInfoMock.getCerts()).thenReturn(certs);
+
+		when(artiMock.getDir()).thenReturn(dirName);
+		when(artiMock.getNs()).thenReturn(nsName);
+		when(artiMock.getMechid()).thenReturn(mechID);
+
+		when(certInfoMock.getPrivatekey()).thenReturn(privateKeyString);
+
+		when(transMock.start("Reconstitute Private Key", Env.SUB)).thenReturn(mock(TimeTaken.class));
+	}
+
+	@AfterClass
+	public static void tearDownOnce() {
+		cleanup();
+		PlaceArtifactInKeystore.clear();
+	}
+
+	@Test
+	public void test() throws CadiException {
+		PlaceArtifactInKeystore placer = new PlaceArtifactInKeystore("pkcs12");
+
+		certs.add(x509String);
+		certs.add(x509Chain);
+		assertThat(placer.place(transMock, certInfoMock, artiMock, "machine"), is(true));
+		for (String ext : new String[] {"chal", "keyfile", "pkcs12", "props", "trust.pkcs12"}) {
+			assertThat(new File(dirName + '/' + nsName + '.' + ext).exists(), is(true));
+		}
+
+		// coverage
+		assertThat(placer.place(transMock, certInfoMock, artiMock, "machine"), is(true));
+		
+		when(certInfoMock.getCerts()).thenReturn(null);
+		try {
+			placer._place(transMock, certInfoMock, artiMock);
+			fail("Should've thrown an exception");
+		} catch (Exception e) {
+		}
+
+	}
+
+	private static void cleanup() {
+		File dir = new File(dirName);
+		if (dir.exists()) {
+			for (File f : dir.listFiles()) {
+				f.delete();
+			}
+			dir.delete();
+		}
+	}
+
+	public String fromFile(File file) throws IOException {
+		BufferedReader br = new BufferedReader(new FileReader(file));
+		ByteArrayOutputStream baos = new ByteArrayOutputStream();
+		String line;
+		baos.write(br.readLine().getBytes());
+		// Here comes the hacky part
+		baos.write("\n".getBytes());
+		while((line=br.readLine())!=null) {
+			if(line.length()>0) {
+				baos.write(line.getBytes());
+				baos.write("\n".getBytes());
+			}
+		}
+		br.close();
+		return baos.toString();
+	}
+}