Improve coverage of cadi-aaf

[aaf/authz.git] / cadi / aaf / src / test / java / org / onap / aaf / cadi / cm / test / JU_PlaceArtifactInKeystore.java

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 (file)
index 0000000..02054cb
--- /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();
+       }
+}