Improve coverage of cadi-aaf

Issue-ID: AAF-223
Change-Id: I8690aa8cc0448cd82c5ee53361d617724bd41cc9
Signed-off-by: IanHowell <ian.howell@att.com>

diff --git a/cadi/aaf/src/test/java/org/onap/aaf/cadi/cm/test/JU_ArtifactDir.java b/cadi/aaf/src/test/java/org/onap/aaf/cadi/cm/test/JU_ArtifactDir.java
new file mode 100644 (file)
index 0000000..1f68cf6
--- /dev/null
+++ b/cadi/aaf/src/test/java/org/onap/aaf/cadi/cm/test/JU_ArtifactDir.java
@@ -0,0 +1,162 @@
+/**
+ * ============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.File;
+import java.io.IOException;
+import java.security.KeyStore;
+import java.security.KeyStoreException;
+import java.security.NoSuchAlgorithmException;
+import java.security.cert.CertificateException;
+import java.util.ArrayList;
+import java.util.List;
+
+import org.junit.*;
+import org.mockito.*;
+
+import org.onap.aaf.cadi.CadiException;
+import org.onap.aaf.cadi.cm.ArtifactDir;
+import org.onap.aaf.cadi.util.Chmod;
+import org.onap.aaf.misc.env.Trans;
+
+import certman.v1_0.Artifacts.Artifact;
+import certman.v1_0.CertInfo;
+
+public class JU_ArtifactDir {
+
+       @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 luggagePassword = "12345";  // That's the stupidest combination I've ever heard in my life
+
+       private List<String> issuers;
+
+       @Before
+       public void setup() {
+               MockitoAnnotations.initMocks(this);
+               
+               issuers = new ArrayList<>();
+               issuers.add("issuer1");
+               issuers.add("issuer2");
+       }
+       
+       @AfterClass
+       public static void tearDownOnce() {
+               cleanup();
+       }
+
+       @Test
+       public void test() throws CadiException, IOException, KeyStoreException, NoSuchAlgorithmException, CertificateException {
+               ArtifactDirStud artiDir = new ArtifactDirStud();
+
+               try {
+                       artiDir.place(transMock, certInfoMock, artiMock, "machine");
+                       fail("Should've thrown an exception");
+               } catch (CadiException e) {
+                       assertThat(e.getMessage(), is("File Artifacts require a path\nFile Artifacts require an AAF Namespace"));
+               }
+               
+               when(artiMock.getDir()).thenReturn(dirName);
+               try {
+                       artiDir.place(transMock, certInfoMock, artiMock, "machine");
+                       fail("Should've thrown an exception");
+               } catch (CadiException e) {
+                       assertThat(e.getMessage(), is("File Artifacts require an AAF Namespace"));
+               }
+               
+               when(artiMock.getNs()).thenReturn(nsName);
+               when(certInfoMock.getCaIssuerDNs()).thenReturn(issuers);
+               when(certInfoMock.getChallenge()).thenReturn(luggagePassword);
+               artiDir.place(transMock, certInfoMock, artiMock, "machine");
+               
+               File writableFile = new File(dirName + '/' + nsName + "writable.txt");
+               artiDir.write(writableFile, Chmod.to755, "first data point", "second data point");
+               try {
+                       artiDir.write(writableFile, Chmod.to755, (String[])null);
+                       fail("Should've thrown an exception");
+               } catch(NullPointerException e) {
+               }
+               
+               KeyStore ks = KeyStore.getInstance("pkcs12");
+               try {
+                       artiDir.write(writableFile, Chmod.to755, ks, luggagePassword.toCharArray());
+                       fail("Should've thrown an exception");
+               } catch(CadiException e) {
+               }
+               
+               ks.load(null, null);
+               artiDir.write(writableFile, Chmod.to755, ks, luggagePassword.toCharArray());
+               
+               ArtifactDirStud artiDir2 = new ArtifactDirStud();
+               artiDir2.place(transMock, certInfoMock, artiMock, "machine");
+
+               // coverage
+               artiDir.place(transMock, certInfoMock, artiMock, "machine");
+
+               ArtifactDir.clear();
+               artiDir.place(transMock, certInfoMock, artiMock, "machine");
+       
+       }
+
+       @Test(expected = CadiException.class)
+       public void throwsTest() throws CadiException {
+               ArtifactDirStud artiDir = new ArtifactDirStud();
+               when(artiMock.getDir()).thenReturn(dirName);
+               when(artiMock.getNs()).thenReturn(nsName);
+               artiDir.place(transMock, certInfoMock, artiMock, "machine");
+       }
+
+       private class ArtifactDirStud extends ArtifactDir {
+               @Override
+               protected boolean _place(Trans trans, CertInfo certInfo, Artifact arti) throws CadiException {
+                       // This is only here so that we have a concrete class to test
+                       return false;
+               }
+               
+               // Expose the protected methods
+
+               public  void write(File f, Chmod c, String ... data) throws IOException {
+                       super.write(f, c, data);
+               }
+               public void write(File f, Chmod c, KeyStore ks, char[] pass ) throws IOException, CadiException {
+                       super.write(f, c, ks, pass);
+               }
+       }
+
+       private static void cleanup() {
+               File dir = new File(dirName);
+               if (dir.exists()) {
+                       for (File f : dir.listFiles()) {
+                               f.delete();
+                       }
+                       dir.delete();
+               }
+       }
+
+}