Update Batch from Testing

[aaf/authz.git] / auth / auth-batch / src / test / java / org / onap / aaf / auth / batch / actions / test / JU_ActionDAO.java

diff --git a/auth/auth-batch/src/test/java/org/onap/aaf/auth/batch/actions/test/JU_ActionDAO.java b/auth/auth-batch/src/test/java/org/onap/aaf/auth/batch/actions/test/JU_ActionDAO.java
index 9376da4..3974389 100644 (file)
--- a/auth/auth-batch/src/test/java/org/onap/aaf/auth/batch/actions/test/JU_ActionDAO.java
+++ b/auth/auth-batch/src/test/java/org/onap/aaf/auth/batch/actions/test/JU_ActionDAO.java
@@ -21,37 +21,46 @@
 
 package org.onap.aaf.auth.batch.actions.test;
 
-import static org.junit.Assert.*;
-import org.junit.After;
+import static org.junit.Assert.assertTrue;
+import static org.mockito.MockitoAnnotations.initMocks;
+
+import java.io.IOException;
+import java.util.Properties;
+
 import org.junit.Before;
 import org.junit.Test;
 import org.mockito.Mock;
 import org.mockito.Mockito;
 import org.onap.aaf.auth.batch.actions.ActionDAO;
+import org.onap.aaf.auth.common.Define;
+import org.onap.aaf.auth.dao.hl.Function;
+import org.onap.aaf.auth.dao.hl.Question;
 import org.onap.aaf.auth.env.AuthzTrans;
 import org.onap.aaf.auth.layer.Result;
+import org.onap.aaf.cadi.CadiException;
+import org.onap.aaf.cadi.PropAccess;
+import org.onap.aaf.cadi.config.Config;
 import org.onap.aaf.misc.env.APIException;
+import org.onap.aaf.misc.env.LogTarget;
+import org.onap.aaf.misc.env.TimeTaken;
 
 import com.datastax.driver.core.Cluster;
-import com.datastax.driver.core.Configuration;
-import com.datastax.driver.core.Cluster.Initializer;
-import com.datastax.driver.core.Host.StateListener;
-
-import static org.mockito.Mockito.*;
-
-import java.io.IOException;
-import java.net.InetSocketAddress;
-import java.util.Collection;
-import java.util.List;
-
-import org.junit.Test;
+import com.datastax.driver.core.ColumnDefinitions;
+import com.datastax.driver.core.PreparedId;
+import com.datastax.driver.core.PreparedStatement;
+import com.datastax.driver.core.Session;
 
 public class JU_ActionDAO {
     
-    AuthzTrans aTrans;
-    Cluster cluster;
-    ActionDAOStub actionDAOStub;
-    ActionDAOStub actionDAOStub1;
+       @Mock
+    AuthzTrans trans;
+       @Mock
+       Cluster cluster;
+       @Mock
+       PropAccess access;
+       
+       @Mock
+       ActionDAO actionObj;
 
     private class ActionDAOStub extends ActionDAO {
 
@@ -75,6 +84,34 @@ public class JU_ActionDAO {
     
     @Before
     public void setUp() throws APIException, IOException {
+       initMocks(this);
+       Session sessionObj=Mockito.mock(Session.class);
+       PreparedStatement psObj =Mockito.mock(PreparedStatement.class);
+               try {
+                       Mockito.doReturn(Mockito.mock(LogTarget.class)).when(trans).init();
+                       Mockito.doReturn(Mockito.mock(LogTarget.class)).when(trans).warn();
+                       Mockito.doReturn(Mockito.mock(LogTarget.class)).when(trans).debug();
+                       Mockito.doReturn("10").when(trans).getProperty(Config.AAF_USER_EXPIRES, Config.AAF_USER_EXPIRES_DEF);
+                       Mockito.doReturn(Mockito.mock(TimeTaken.class)).when(trans).start(Mockito.anyString(),Mockito.anyInt());
+                       Mockito.doReturn(sessionObj).when(cluster).connect("authz");
+                       Mockito.doReturn(psObj).when(sessionObj).prepare(Mockito.anyString());
+                       
+                       Mockito.doReturn(Mockito.mock(ColumnDefinitions.class)).when(psObj).getVariables();
+                       Mockito.doReturn(Mockito.mock(PreparedId.class)).when(psObj).getPreparedId();
+                       Mockito.doReturn(Mockito.mock(Properties.class)).when(access).getProperties();
+                       Mockito.doReturn("test.test").when(access).getProperty(Config.AAF_ROOT_NS,"org.osaaf.aaf");
+                       Define.set(access);
+                       actionObj = new ActionDAOStub(trans, cluster, true);
+               } catch (APIException e) {
+                       // TODO Auto-generated catch block
+                       e.printStackTrace();
+               } catch (IOException e) {
+                       // TODO Auto-generated catch block
+                       e.printStackTrace();
+               } catch (CadiException e) {
+                       // TODO Auto-generated catch block
+                       e.printStackTrace();
+               }
 //        Cluster.Initializer cInit = mock(Cluster.Initializer.class);
 //        Cluster.Builder cBuild = new Cluster.Builder();
 //        cBuild.addContactPoint("test");
@@ -91,5 +128,44 @@ public class JU_ActionDAO {
 //        actionDAOStub = new ActionDAOStub(aTrans,cluster,true);
 //        actionDAOStub1 = new ActionDAOStub(aTrans, actionDAOStub);
     }
+    
+    @Test
+       public void testGetSession() {
+       try {
+               Session session = actionObj.getSession(trans);
+               assertTrue(session.toString().contains("Mock for Session"));
+               } catch (APIException e) {
+                       // TODO Auto-generated catch block
+                       e.printStackTrace();
+               } catch (IOException e) {
+                       // TODO Auto-generated catch block
+                       e.printStackTrace();
+               } 
+       }
+    
+    @Test
+       public void testQuestion() {
+               Question retVal = actionObj.question();
+               assertTrue(retVal.toString().contains("org.onap.aaf.auth.dao.hl.Question"));
+       }
+    
+    @Test
+       public void testFunction() {
+       Function retVal = actionObj.function();
+               assertTrue(retVal.toString().contains("org.onap.aaf.auth.dao.hl.Function"));
+       }
+    
+    @Test
+       public void testClose() {
+               actionObj.close(trans);
+//             assertTrue(session.toString().contains("Mock for Session"));
+       }
+    
+    @Test
+       public void testCloseFalse() {
+       actionObj = new ActionDAOStub(trans, Mockito.mock(ActionDAO.class));
+                       actionObj.close(trans);
+//                     assertTrue(session.toString().contains("Mock for Session"));
+       }
 
 }