auth/auth-cass/src/test/java/org/onap/aaf/auth/dao/cass/JU_LocateDAO.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.junit.Assert.assertTrue;
  25 import static org.mockito.MockitoAnnotations.initMocks;
  26
  27 import java.io.ByteArrayInputStream;
  28 import java.io.ByteArrayOutputStream;
  29 import java.io.DataInputStream;
  30 import java.io.DataOutputStream;
  31 import java.io.IOException;
  32 import java.lang.reflect.Constructor;
  33 import java.lang.reflect.Field;
  34 import java.lang.reflect.InvocationTargetException;
  35 import java.lang.reflect.Method;
  36 import java.nio.ByteBuffer;
  37 import java.util.HashSet;
  38 import java.util.List;
  39 import java.util.Set;
  40 import java.util.TreeSet;
  41
  42 import org.junit.Before;
  43 import org.junit.Test;
  44 import org.mockito.Mock;
  45 import org.mockito.Mockito;
  46 import org.onap.aaf.auth.dao.AbsCassDAO;
  47 import org.onap.aaf.auth.dao.AbsCassDAO.CRUD;
  48 import org.onap.aaf.auth.dao.AbsCassDAO.PSInfo;
  49 import org.onap.aaf.auth.dao.CassDAOImpl;
  50 import org.onap.aaf.auth.env.AuthzTrans;
  51 import org.onap.aaf.auth.layer.Result;
  52 import org.onap.aaf.misc.env.APIException;
  53
  54 import com.datastax.driver.core.Cluster;
  55 import com.datastax.driver.core.Row;
  56
  57 public class JU_LocateDAO {
  58
  59     @Mock
  60     AuthzTrans trans;
  61     @Mock
  62     Cluster cluster;
  63
  64     @Before
  65     public void setUp() throws APIException, IOException {
  66         initMocks(this);
  67     }
  68
  69     @Test
  70     public void testInit() {
  71         try {
  72             LocateDAO daoObj = new LocateDAO(trans, cluster, "test");
  73         } catch (APIException | IOException e) {
  74             // TODO Auto-generated catch block
  75             e.printStackTrace();
  76         }
  77     }
  78     @Test
  79     public void testReadByStartAndTarget() {
  80         LocateDAO daoObj = null;
  81         try {
  82             daoObj = new LocateDAO(trans, cluster, "test");
  83         } catch (APIException | IOException e) {
  84             // TODO Auto-generated catch block
  85             e.printStackTrace();
  86         }
  87
  88         PSInfo psObj = Mockito.mock(PSInfo.class);
  89         setPsByStartAndTarget(daoObj, psObj, "psName");
  90
  91         Result<List<LocateDAO.Data>>  rs1 = new Result<List<LocateDAO.Data>>(null,0,"test",new String[0]);
  92         Mockito.doReturn(rs1).when(psObj).read(trans, "LocateDAO READ", new Object[]{"test"});
  93
  94         daoObj.readByName(trans, "test");
  95     }
  96
  97
  98     public void setPsByStartAndTarget(LocateDAO LocateDAOObj, PSInfo psInfoObj, String fieldName) {
  99         Field nsDaoField;
 100         try {
 101             nsDaoField = LocateDAO.class.getDeclaredField(fieldName);
 102
 103             nsDaoField.setAccessible(true);
 104             // remove final modifier from field
 105             Field modifiersField = Field.class.getDeclaredField("modifiers");
 106             modifiersField.setAccessible(true);
 107 //            modifiersField.setInt(nsDaoField, nsDaoField.getModifiers() & ~Modifier.FINAL);
 108
 109             nsDaoField.set(LocateDAOObj, psInfoObj);
 110         } catch (NoSuchFieldException | SecurityException e) {
 111             // TODO Auto-generated catch block
 112             e.printStackTrace();
 113         } catch (IllegalArgumentException e) {
 114             // TODO Auto-generated catch block
 115             e.printStackTrace();
 116         } catch (IllegalAccessException e) {
 117             // TODO Auto-generated catch block
 118             e.printStackTrace();
 119         }
 120     }
 121
 122     @Test
 123     public void testWasMOdified() {
 124
 125         LocateDAO.Data data  = new LocateDAO.Data();
 126
 127         HistoryDAO historyDAO = Mockito.mock(HistoryDAO.class);
 128
 129         LocateDAO daoObj = null;
 130         try {
 131             daoObj = new LocateDAO(trans, historyDAO );
 132         } catch (APIException | IOException e) {
 133             // TODO Auto-generated catch block
 134             e.printStackTrace();
 135         }
 136         daoObj.wasModified(trans, CRUD.create, data, new String[] {"test"});
 137
 138
 139     }
 140
 141     @Test
 142     public void testSecondConstructor() {
 143         AbsCassDAO historyDAO = Mockito.mock(AbsCassDAO.class);
 144
 145         try {
 146             LocateDAO daoObj = new LocateDAO(trans, historyDAO);
 147         } catch (APIException | IOException e) {
 148             // TODO Auto-generated catch block
 149             e.printStackTrace();
 150         }
 151
 152     }
 153
 154     @Test
 155     public void testLocateLoader(){
 156         Class<?> innerClass = null;
 157         Class<?>[] innerClassArr = LocateDAO.class.getDeclaredClasses();
 158         for(Class indCls:innerClassArr) {
 159             if(indCls.getName().contains("LocateLoader")) {
 160                 innerClass = indCls;
 161                 break;
 162             }
 163         }
 164
 165         Constructor<?> constructor = innerClass.getDeclaredConstructors()[0];
 166         constructor.setAccessible(true);
 167
 168         try {
 169
 170             Object obj = constructor.newInstance(1);
 171             Method innnerClassMtd;
 172
 173             LocateDAO.Data data  = new LocateDAO.Data();
 174             Row row = Mockito.mock(Row.class);
 175             ByteBuffer bbObj = ByteBuffer.allocateDirect(10);
 176             bbObj.limit(7);
 177             bbObj.put(0, new Byte("0"));
 178             bbObj.put(1, new Byte("1"));
 179             bbObj.put(2, new Byte("2"));
 180             Mockito.doReturn(bbObj).when(row).getBytesUnsafe(1);
 181
 182             innnerClassMtd = innerClass.getMethod("load", new Class[] {LocateDAO.Data.class, Row.class});
 183             innnerClassMtd.invoke(obj, new Object[] {data, row});
 184
 185             innnerClassMtd = innerClass.getDeclaredMethod("key", new Class[] {LocateDAO.Data.class, Integer.TYPE, Object[].class });
 186             innnerClassMtd.invoke(obj, new Object[] {data, 0, new Object[] {"test","test","test"} });
 187
 188             innnerClassMtd = innerClass.getDeclaredMethod("body", new Class[] {LocateDAO.Data.class, Integer.TYPE, Object[].class });
 189             innnerClassMtd.invoke(obj, new Object[] {data, 1, new Object[] {"test","test","test","test","test","test","test","test","test","test","test"} });
 190
 191             ByteArrayOutputStream baos = new ByteArrayOutputStream();
 192             DataOutputStream dos = new DataOutputStream(baos);
 193             innnerClassMtd = innerClass.getDeclaredMethod("marshal", new Class[] {LocateDAO.Data.class, DataOutputStream.class });
 194             innnerClassMtd.invoke(obj, new Object[] {data, dos });
 195
 196             ByteArrayInputStream bais = new ByteArrayInputStream(baos.toByteArray());
 197             DataInputStream dis = new DataInputStream(bais);
 198             innnerClassMtd = innerClass.getDeclaredMethod("unmarshal", new Class[] {LocateDAO.Data.class, DataInputStream.class });
 199             innnerClassMtd.invoke(obj, new Object[] {data, dis });
 200
 201         } catch (InstantiationException e) {
 202             // TODO Auto-generated catch block
 203             e.printStackTrace();
 204         } catch (IllegalAccessException e) {
 205             // TODO Auto-generated catch block
 206             e.printStackTrace();
 207         } catch (IllegalArgumentException e) {
 208             // TODO Auto-generated catch block
 209             e.printStackTrace();
 210         } catch (InvocationTargetException e) {
 211             // TODO Auto-generated catch block
 212             e.printStackTrace();
 213         } catch (NoSuchMethodException e) {
 214             // TODO Auto-generated catch block
 215             e.printStackTrace();
 216         } catch (SecurityException e) {
 217             // TODO Auto-generated catch block
 218             e.printStackTrace();
 219         }
 220     }
 221
 222     @Test
 223     public void testData(){
 224         LocateDAO.Data data  = new LocateDAO.Data();
 225         data.name="name";
 226         data.hostname="hostname";
 227         try {
 228             data.bytify();
 229         } catch (IOException e) {
 230             // TODO Auto-generated catch block
 231             e.printStackTrace();
 232         }
 233
 234         data.subprotocol(true);
 235
 236         Set<String> subProt = new HashSet<String>();
 237         Field protField;
 238         try {
 239             protField = LocateDAO.Data.class.getDeclaredField("subprotocol");
 240
 241             protField.setAccessible(true);
 242
 243             protField.set(data, subProt);
 244         } catch (NoSuchFieldException | SecurityException e) {
 245             // TODO Auto-generated catch block
 246             e.printStackTrace();
 247         } catch (IllegalArgumentException e) {
 248             // TODO Auto-generated catch block
 249             e.printStackTrace();
 250         } catch (IllegalAccessException e) {
 251             // TODO Auto-generated catch block
 252             e.printStackTrace();
 253         }
 254
 255         data.subprotocol(true);
 256         subProt = new TreeSet<String>();
 257         subProt.add("test");
 258         try {
 259             protField = LocateDAO.Data.class.getDeclaredField("subprotocol");
 260
 261             protField.setAccessible(true);
 262
 263             protField.set(data, subProt);
 264         } catch (NoSuchFieldException | SecurityException e) {
 265             // TODO Auto-generated catch block
 266             e.printStackTrace();
 267         } catch (IllegalArgumentException e) {
 268             // TODO Auto-generated catch block
 269             e.printStackTrace();
 270         } catch (IllegalAccessException e) {
 271             // TODO Auto-generated catch block
 272             e.printStackTrace();
 273         }
 274
 275         data.subprotocol(true);
 276         data.subprotocol(false);
 277
 278         LocateDAO.Data newDate = data.copy();
 279         assertTrue(data.name.equals(newDate.name));
 280     }
 281
 282 }
 283
 284 class LocateDAOImpl extends LocateDAO{
 285
 286
 287     public LocateDAOImpl(AuthzTrans trans, HistoryDAO historyDAO,PSInfo readPS  ) throws APIException, IOException {
 288         super(trans, historyDAO);
 289         setPs(this, readPS, "createPS");
 290     }
 291
 292
 293     public void setPs(LocateDAOImpl LocateDAOObj, PSInfo psInfoObj, String methodName) {
 294         Field nsDaoField;
 295         try {
 296             nsDaoField = CassDAOImpl.class.getDeclaredField(methodName);
 297
 298             nsDaoField.setAccessible(true);
 299             // remove final modifier from field
 300             Field modifiersField = Field.class.getDeclaredField("modifiers");
 301             modifiersField.setAccessible(true);
 302 //            modifiersField.setInt(nsDaoField, nsDaoField.getModifiers() & ~Modifier.FINAL);
 303
 304             nsDaoField.set(LocateDAOObj, psInfoObj);
 305         } catch (NoSuchFieldException | SecurityException e) {
 306             // TODO Auto-generated catch block
 307             e.printStackTrace();
 308         } catch (IllegalArgumentException e) {
 309             // TODO Auto-generated catch block
 310             e.printStackTrace();
 311         } catch (IllegalAccessException e) {
 312             // TODO Auto-generated catch block
 313             e.printStackTrace();
 314         }
 315     }
 316
 317 }