add junits to auth cass

[aaf/authz.git] / auth / auth-cass / src / test / java / org / onap / aaf / auth / direct / test / JU_DirectCertIdentity.java

diff --git a/auth/auth-cass/src/test/java/org/onap/aaf/auth/direct/test/JU_DirectCertIdentity.java b/auth/auth-cass/src/test/java/org/onap/aaf/auth/direct/test/JU_DirectCertIdentity.java
index 3377c7e..2f64e09 100644 (file)
--- a/auth/auth-cass/src/test/java/org/onap/aaf/auth/direct/test/JU_DirectCertIdentity.java
+++ b/auth/auth-cass/src/test/java/org/onap/aaf/auth/direct/test/JU_DirectCertIdentity.java
@@ -22,11 +22,14 @@
 
 package org.onap.aaf.auth.direct.test;
 
-import static org.junit.Assert.*;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertTrue;
 
 import java.security.Principal;
 import java.security.cert.CertificateException;
 import java.security.cert.X509Certificate;
+import java.util.ArrayList;
+import java.util.List;
 
 import javax.servlet.http.HttpServletRequest;
 
@@ -34,39 +37,64 @@ import org.junit.Before;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.mockito.Mock;
+import org.mockito.Mockito;
 import org.onap.aaf.auth.dao.cached.CachedCertDAO;
+import org.onap.aaf.auth.dao.cass.CertDAO;
 import org.onap.aaf.auth.direct.DirectCertIdentity;
+import org.onap.aaf.auth.env.AuthzTrans;
+import org.onap.aaf.auth.layer.Result;
 import org.powermock.modules.junit4.PowerMockRunner;
 
 @RunWith(PowerMockRunner.class)
 public class JU_DirectCertIdentity {
-    
-    public DirectCertIdentity directCertIdentity;
-    
-    @Before
-    public void setUp(){
-        directCertIdentity = new DirectCertIdentity();
-    }
 
+       public DirectCertIdentity directCertIdentity;
 
-    @Mock
-    HttpServletRequest req;
-    X509Certificate cert;
-    byte[] _certBytes;
-    
-    @Test
-    public void testidentity(){
-        
-        try {
-        Principal p = directCertIdentity.identity(req, cert, _certBytes);
-        assertEquals(( (p) == null),true);
-            
-        } catch (CertificateException e) {
-            // TODO Auto-generated catch block
-            e.printStackTrace();
-        }
-        //assertTrue(true);
-        
-    }
+       @Before
+       public void setUp() {
+               directCertIdentity = new DirectCertIdentity();
+       }
+
+       @Mock
+       HttpServletRequest req;
+       X509Certificate cert;
+       byte[] _certBytes;
+
+       @Test
+       public void testidentity() {
+
+               try {
+                       Principal p = directCertIdentity.identity(req, cert, _certBytes);
+                       assertEquals(((p) == null), true);
+                       
+                       cert = Mockito.mock(X509Certificate.class);
+                       Mockito.when(cert.getEncoded()).thenReturn(new byte[128]);
+                       
+                       Result<List<CertDAO.Data>> rs = new Result<List<CertDAO.Data>>(null, 1, "test", new Object[0]);
+                       
+                       CachedCertDAO cacheDao = Mockito.mock(CachedCertDAO.class);
+                       Mockito.when(cacheDao.read(Mockito.any(AuthzTrans.class),Mockito.any(Object[].class))).thenReturn(rs);
+                       DirectCertIdentity.set(cacheDao);
+                       p = directCertIdentity.identity(req, cert, _certBytes);
+                       
+                       _certBytes = new byte[128];
+                       List<CertDAO.Data> dataAL = new ArrayList<>();
+                       CertDAO.Data data = new CertDAO.Data();
+                       dataAL.add(data);
+                       rs = new Result<List<CertDAO.Data>>(dataAL, 0, "test", new Object[0]);
+                       Mockito.when(cacheDao.read(Mockito.any(AuthzTrans.class),Mockito.any(Object[].class))).thenReturn(rs);
+                       DirectCertIdentity.set(cacheDao);
+                       p = directCertIdentity.identity(req, cert, _certBytes);
+                       assertTrue(p.toString().contains("X509 Authentication for null"));
+                       
+                       cert = null;
+                       directCertIdentity.identity(req, cert, _certBytes);
+               } catch (CertificateException e) {
+                       // TODO Auto-generated catch block
+                       e.printStackTrace();
+               }
+               // assertTrue(true);
+
+       }
 
 }