auth/auth-cass/src/test/java/org/onap/aaf/auth/dao/cass/JU_ConfigDAOTest.java

   1 /**
   2  * ============LICENSE_START====================================================
   3  * org.onap.aaf
   4  * ===========================================================================
   5  * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.
   6  * ===========================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END====================================================
  19  *
  20  */
  21
  22 package org.onap.aaf.auth.dao.cass;
  23
  24 import static org.mockito.MockitoAnnotations.initMocks;
  25
  26 import java.io.ByteArrayInputStream;
  27 import java.io.ByteArrayOutputStream;
  28 import java.io.DataInputStream;
  29 import java.io.DataOutputStream;
  30 import java.io.FileInputStream;
  31 import java.io.FileNotFoundException;
  32 import java.io.IOException;
  33 import java.lang.reflect.Constructor;
  34 import java.lang.reflect.Field;
  35 import java.lang.reflect.InvocationTargetException;
  36 import java.lang.reflect.Method;
  37 import java.nio.ByteBuffer;
  38 import java.util.List;
  39
  40 import org.junit.Before;
  41 import org.junit.Test;
  42 import org.mockito.Mock;
  43 import org.mockito.Mockito;
  44 import org.onap.aaf.auth.dao.AbsCassDAO;
  45 import org.onap.aaf.auth.dao.AbsCassDAO.CRUD;
  46 import org.onap.aaf.auth.dao.AbsCassDAO.PSInfo;
  47 import org.onap.aaf.auth.dao.CassAccess;
  48 import org.onap.aaf.auth.dao.cass.ConfigDAO.Data;
  49 import org.onap.aaf.auth.env.AuthzEnv;
  50 import org.onap.aaf.auth.env.AuthzTrans;
  51 import org.onap.aaf.auth.layer.Result;
  52 import org.onap.aaf.cadi.config.Config;
  53 import org.onap.aaf.misc.env.APIException;
  54 import org.onap.aaf.misc.env.Decryptor;
  55 import org.onap.aaf.misc.env.Env;
  56 import org.onap.aaf.misc.env.LogTarget;
  57 import org.onap.aaf.misc.env.TimeTaken;
  58
  59 import com.datastax.driver.core.Cluster;
  60 import com.datastax.driver.core.ResultSet;
  61 import com.datastax.driver.core.Row;
  62 import com.datastax.driver.core.Session;
  63
  64 public class JU_ConfigDAOTest {
  65
  66         @Mock
  67     AuthzTrans trans;
  68         @Mock
  69         Cluster cluster;
  70         @Mock
  71         Session session;
  72         @Mock
  73         AuthzEnv env;
  74         @Mock
  75         LogTarget logTarget;
  76
  77         @Before
  78         public void setUp() throws APIException, IOException {
  79                 initMocks(this);
  80                 Mockito.doReturn(Mockito.mock(LogTarget.class)).when(trans).warn();
  81                 Mockito.doReturn(Mockito.mock(LogTarget.class)).when(trans).error();
  82                 Mockito.doReturn(Mockito.mock(LogTarget.class)).when(trans).debug();
  83                 Mockito.doReturn(Mockito.mock(LogTarget.class)).when(trans).info();
  84                 Mockito.doReturn(Mockito.mock(LogTarget.class)).when(trans).init();
  85                 Mockito.doReturn("100").when(trans).getProperty(Config.CADI_LATITUDE);
  86                 Mockito.doReturn("100").when(trans).getProperty(Config.CADI_LONGITUDE);
  87                 Mockito.doReturn(session).when(cluster).connect("test");
  88         }
  89
  90         @Test
  91         public void testInit() {
  92                 TimeTaken tt = Mockito.mock(TimeTaken.class);
  93                 Mockito.doReturn(tt).when(trans).start("ConfigDAO CREATE", Env.REMOTE);
  94                 Mockito.doReturn(tt).when(trans).start("Clear Reset Deque", Env.SUB);
  95                 Mockito.doReturn(tt).when(trans).start("New Cassandra Session", Env.SUB);
  96                 Mockito.doReturn(tt).when(trans).start("Preparing PSInfo CREATE on ConfigDAO", Env.SUB);
  97                 Mockito.doNothing().when(tt).done();
  98                 ConfigDAO.Data data  = new ConfigDAO.Data();
  99                 PSInfo createPS = Mockito.mock(PSInfo.class);
 100                 Result<ResultSet> rs = new Result<ResultSet>(null,0,"test",new String[0]);
 101                 Mockito.doReturn(rs).when(createPS).exec(trans, "ConfigDAOImpl CREATE", data);
 102
 103                 ConfigDAO daoObj=null;
 104                 try {
 105                         daoObj = new ConfigDAO(trans, cluster, "test");
 106                 } catch (APIException | IOException e) {
 107                         // TODO Auto-generated catch block
 108                         e.printStackTrace();
 109                 }
 110
 111         }
 112
 113         @Test
 114         public void testConfigLoader(){
 115
 116                 Class<?> innerClass = ConfigDAO.class.getDeclaredClasses()[0];
 117         Constructor<?> constructor = innerClass.getDeclaredConstructors()[0];
 118         constructor.setAccessible(true);
 119         try {
 120                         Object obj = constructor.newInstance(1);
 121                         Method innnerClassMtd;
 122
 123                         ConfigDAO.Data data  = new ConfigDAO.Data();
 124                         Row row = Mockito.mock(Row.class);
 125                         ByteBuffer bbObj = ByteBuffer.allocateDirect(10);
 126                         bbObj.limit(7);
 127                         bbObj.put(0, new Byte("0"));
 128                         bbObj.put(1, new Byte("1"));
 129                         bbObj.put(2, new Byte("2"));
 130                         Mockito.doReturn(bbObj).when(row).getBytesUnsafe(1);
 131
 132                         innnerClassMtd = innerClass.getMethod("load", new Class[] {ConfigDAO.Data.class, Row.class});
 133                         innnerClassMtd.invoke(obj, new Object[] {data, row});
 134
 135                         innnerClassMtd = innerClass.getDeclaredMethod("key", new Class[] {ConfigDAO.Data.class, Integer.TYPE, Object[].class });
 136                         innnerClassMtd.invoke(obj, new Object[] {data, 1, new Object[] {"test","test","test"} });
 137 //
 138                         innnerClassMtd = innerClass.getDeclaredMethod("body", new Class[] {ConfigDAO.Data.class, Integer.TYPE, Object[].class });
 139                         innnerClassMtd.invoke(obj, new Object[] {data, 1, new Object[] {"test","test","test","test","test","test","test","test","test","test","test"} });
 140
 141                         ByteArrayOutputStream baos = new ByteArrayOutputStream();
 142                         DataOutputStream dos = new DataOutputStream(baos);
 143                         innnerClassMtd = innerClass.getDeclaredMethod("marshal", new Class[] {ConfigDAO.Data.class, DataOutputStream.class });
 144                         innnerClassMtd.invoke(obj, new Object[] {data, dos });
 145
 146                         ByteArrayInputStream bais = new ByteArrayInputStream(baos.toByteArray());
 147                         DataInputStream dis = new DataInputStream(bais);
 148                         innnerClassMtd = innerClass.getDeclaredMethod("unmarshal", new Class[] {ConfigDAO.Data.class, DataInputStream.class });
 149                         innnerClassMtd.invoke(obj, new Object[] {data, dis });
 150
 151                 } catch (InstantiationException e) {
 152                         // TODO Auto-generated catch block
 153                         e.printStackTrace();
 154                 } catch (IllegalAccessException e) {
 155                         // TODO Auto-generated catch block
 156                         e.printStackTrace();
 157                 } catch (IllegalArgumentException e) {
 158                         // TODO Auto-generated catch block
 159                         e.printStackTrace();
 160                 } catch (InvocationTargetException e) {
 161                         // TODO Auto-generated catch block
 162                         e.printStackTrace();
 163                 } catch (NoSuchMethodException e) {
 164                         // TODO Auto-generated catch block
 165                         e.printStackTrace();
 166                 } catch (SecurityException e) {
 167                         // TODO Auto-generated catch block
 168                         e.printStackTrace();
 169                 }
 170         }
 171
 172         @Test
 173         public void testWasMOdified() {
 174                 TimeTaken tt = Mockito.mock(TimeTaken.class);
 175                 Mockito.doReturn(tt).when(trans).start("ConfigDAO CREATE", Env.REMOTE);
 176                 Mockito.doReturn(tt).when(trans).start("Clear Reset Deque", Env.SUB);
 177                 Mockito.doReturn(tt).when(trans).start("New Cassandra Session", Env.SUB);
 178                 Mockito.doReturn(tt).when(trans).start("Preparing PSInfo CREATE on ConfigDAO", Env.SUB);
 179                 Mockito.doReturn(tt).when(trans).start("DELETE APPROVAL",Env.REMOTE);
 180                 Mockito.doNothing().when(tt).done();
 181                 ConfigDAO.Data data  = new ConfigDAO.Data();
 182                 PSInfo createPS = Mockito.mock(PSInfo.class);
 183
 184                 ConfigDAO daoObj = null;
 185                 try {
 186                         daoObj = new ConfigDAO(trans, cluster, "test");
 187                 } catch (APIException | IOException e) {
 188                         // TODO Auto-generated catch block
 189                         e.printStackTrace();
 190                 }
 191
 192                 daoObj.wasModified(trans, CRUD.create, data, new String[] {"test"});
 193
 194
 195
 196         }
 197
 198         @Test
 199         public void testRead() {
 200                 TimeTaken tt = Mockito.mock(TimeTaken.class);
 201                 Mockito.doReturn(tt).when(trans).start("ConfigDAO CREATE", Env.REMOTE);
 202                 Mockito.doReturn(tt).when(trans).start("ConfigDAO READ", Env.REMOTE);
 203                 Mockito.doReturn(tt).when(trans).start("Clear Reset Deque", Env.SUB);
 204                 Mockito.doReturn(tt).when(trans).start("New Cassandra Session", Env.SUB);
 205                 Mockito.doReturn(tt).when(trans).start("Preparing PSInfo READ on ConfigDAO", Env.SUB);
 206                 Mockito.doReturn(tt).when(trans).start(Mockito.anyString(),Mockito.anyInt());
 207                 Mockito.doReturn("100").when(trans).getProperty(CassAccess.CASSANDRA_CLUSTERS_PORT,"100");
 208                 Mockito.doReturn("100").when(trans).getProperty(CassAccess.CASSANDRA_CLUSTERS_PORT,"9042");
 209                 Mockito.doReturn("100").when(trans).getProperty(CassAccess.CASSANDRA_CLUSTERS_USER_NAME,"100");
 210                 Mockito.doReturn("100").when(trans).getProperty(CassAccess.CASSANDRA_CLUSTERS_USER_NAME,null);
 211                 Mockito.doReturn(Mockito.mock(Decryptor.class)).when(trans).decryptor();
 212                 Mockito.doNothing().when(tt).done();
 213
 214                 Result<List<Data>>  rs1 = new Result<List<Data>>(null,0,"test",new String[0]);
 215
 216
 217                 PSInfo psObj = Mockito.mock(PSInfo.class);
 218                 ConfigDAOImpl daoObj = null;
 219                 try {
 220                         daoObj = new ConfigDAOImpl(trans, cluster, "test",psObj);
 221                 } catch (APIException | IOException e) {
 222                         // TODO Auto-generated catch block
 223                         e.printStackTrace();
 224                 }
 225                 Mockito.doReturn(rs1).when(psObj).read(trans, "ConfigDAO READ", new Object[]{"test"});
 226                 daoObj.readName(trans, "test");
 227
 228
 229         }
 230
 231
 232
 233         @Test
 234         public void testSecondConstructor() {
 235                 TimeTaken tt = Mockito.mock(TimeTaken.class);
 236                 Mockito.doReturn(tt).when(trans).start("ConfigDAO CREATE", Env.REMOTE);
 237                 Mockito.doReturn(tt).when(trans).start("Clear Reset Deque", Env.SUB);
 238                 Mockito.doReturn(tt).when(trans).start("New Cassandra Session", Env.SUB);
 239                 Mockito.doReturn(tt).when(trans).start("Preparing PSInfo CREATE on ConfigDAO", Env.SUB);
 240                 Mockito.doReturn(tt).when(trans).start("DELETE APPROVAL",Env.REMOTE);
 241                 Mockito.doNothing().when(tt).done();
 242                 AbsCassDAO absDAO = Mockito.mock(AbsCassDAO.class);
 243
 244                 try {
 245                         ConfigDAO daoObj = new ConfigDAO(trans, absDAO);
 246                 } catch (APIException | IOException e) {
 247                         // TODO Auto-generated catch block
 248                         e.printStackTrace();
 249                 }
 250         }
 251 }
 252
 253 class ConfigDAOImpl extends ConfigDAO{
 254
 255
 256         public ConfigDAOImpl(AuthzTrans trans, Cluster cluster, String keyspace,PSInfo readPS  ) throws APIException, IOException {
 257                 super(trans, cluster, keyspace);
 258                 setPs(this, readPS, "psName");
 259         }
 260
 261
 262         public void setPs(ConfigDAOImpl ConfigDAOObj, PSInfo psInfoObj, String methodName) {
 263                 Field nsDaoField;
 264                 try {
 265                         nsDaoField = ConfigDAO.class.getDeclaredField(methodName);
 266
 267                         nsDaoField.setAccessible(true);
 268                 // remove final modifier from field
 269                 Field modifiersField = Field.class.getDeclaredField("modifiers");
 270                 modifiersField.setAccessible(true);
 271 //              modifiersField.setInt(nsDaoField, nsDaoField.getModifiers() & ~Modifier.FINAL);
 272
 273                 nsDaoField.set(ConfigDAOObj, psInfoObj);
 274                 } catch (NoSuchFieldException | SecurityException e) {
 275                         // TODO Auto-generated catch block
 276                         e.printStackTrace();
 277                 } catch (IllegalArgumentException e) {
 278                         // TODO Auto-generated catch block
 279                         e.printStackTrace();
 280                 } catch (IllegalAccessException e) {
 281                         // TODO Auto-generated catch block
 282                         e.printStackTrace();
 283                 }
 284         }
 285
 286
 287         public void setSession(ConfigDAOImpl ConfigDAOObj, Session session) {
 288                 Field nsDaoField;
 289                 try {
 290                         nsDaoField = AbsCassDAO.class.getDeclaredField("session");
 291
 292                         nsDaoField.setAccessible(true);
 293                 // remove final modifier from field
 294                 Field modifiersField = Field.class.getDeclaredField("modifiers");
 295                 modifiersField.setAccessible(true);
 296 //              modifiersField.setInt(nsDaoField, nsDaoField.getModifiers() & ~Modifier.FINAL);
 297                 nsDaoField.set(ConfigDAOObj, session);
 298                 } catch (NoSuchFieldException | SecurityException e) {
 299                         // TODO Auto-generated catch block
 300                         e.printStackTrace();
 301                 } catch (IllegalArgumentException e) {
 302                         // TODO Auto-generated catch block
 303                         e.printStackTrace();
 304                 } catch (IllegalAccessException e) {
 305                         // TODO Auto-generated catch block
 306                         e.printStackTrace();
 307                 }
 308         }
 309
 310 }