X-Git-Url: https://gerrit.onap.org/r/gitweb?a=blobdiff_plain;f=auth%2Fauth-batch%2Fsrc%2Ftest%2Fjava%2Forg%2Fonap%2Faaf%2Fauth%2Fbatch%2Factions%2Ftest%2FJU_ActionDAO.java;h=397438968093692568475a14421b8ee1927e7714;hb=e3163b2d5609753f874e2f52fd5ef67fa063af7a;hp=9376da4b9f0531eb435645ebccbe94c79f3fe9f9;hpb=343481da5dac494c0b063f0b5b0ddb865fa1f214;p=aaf%2Fauthz.git

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 9376da4b..39743896 100644
--- 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"));
+   	}
 
 }