auth/auth-cass/src/test/java/org/onap/aaf/auth/dao/cass/JU_ApprovalDAO.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.assertNull;
  25 import static org.junit.Assert.assertTrue;
  26 import static org.mockito.MockitoAnnotations.initMocks;
  27
  28 import java.io.IOException;
  29 import java.lang.reflect.Field;
  30 import java.util.Iterator;
  31 import java.util.List;
  32 import java.util.UUID;
  33
  34 import org.junit.Before;
  35 import org.junit.Test;
  36 import org.mockito.Mock;
  37 import org.mockito.Mockito;
  38 import org.onap.aaf.auth.dao.AbsCassDAO;
  39 import org.onap.aaf.auth.dao.AbsCassDAO.CRUD;
  40 import org.onap.aaf.auth.dao.AbsCassDAO.PSInfo;
  41 import org.onap.aaf.auth.dao.cass.ApprovalDAO.Data;
  42 import org.onap.aaf.auth.env.AuthzTrans;
  43 import org.onap.aaf.auth.layer.Result;
  44 import org.onap.aaf.misc.env.APIException;
  45 import org.onap.aaf.misc.env.Env;
  46 import org.onap.aaf.misc.env.LogTarget;
  47 import org.onap.aaf.misc.env.TimeTaken;
  48
  49 import com.datastax.driver.core.Cluster;
  50 import com.datastax.driver.core.ColumnDefinitions;
  51 import com.datastax.driver.core.ExecutionInfo;
  52 import com.datastax.driver.core.ResultSet;
  53 import com.datastax.driver.core.Row;
  54 import com.datastax.driver.core.Session;
  55 import com.google.common.util.concurrent.ListenableFuture;
  56
  57 public class JU_ApprovalDAO {
  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                 Mockito.doReturn(Mockito.mock(LogTarget.class)).when(trans).warn();
  68                 Mockito.doReturn(Mockito.mock(LogTarget.class)).when(trans).error();
  69                 Mockito.doReturn(Mockito.mock(LogTarget.class)).when(trans).debug();
  70                 Mockito.doReturn(Mockito.mock(LogTarget.class)).when(trans).info();
  71         }
  72
  73         @Test
  74         public void testInit() {
  75                 TimeTaken tt = Mockito.mock(TimeTaken.class);
  76                 Mockito.doReturn(tt).when(trans).start("ApprovalDAO CREATE", Env.REMOTE);
  77                 Mockito.doReturn(tt).when(trans).start("Clear Reset Deque", Env.SUB);
  78                 Mockito.doReturn(tt).when(trans).start("New Cassandra Session", Env.SUB);
  79                 Mockito.doReturn(tt).when(trans).start("Preparing PSInfo CREATE on ApprovalDAO", Env.SUB);
  80                 Mockito.doNothing().when(tt).done();
  81                 ApprovalDAO.Data data  = new ApprovalDAO.Data();
  82                 PSInfo createPS = Mockito.mock(PSInfo.class);
  83                 Result<ResultSet> rs = new Result<ResultSet>(null,0,"test",new String[0]);
  84                 Mockito.doReturn(rs).when(createPS).exec(trans, "ApprovalDAOImpl CREATE", data);
  85
  86                 ApprovalDAOImpl daoObj = new ApprovalDAOImpl(trans, cluster, "test",data, createPS);
  87 //              data.id
  88                 Result<Data> retVal = daoObj.create(trans, data);
  89                 assertTrue(retVal.status == 0);
  90
  91                 rs = new Result<ResultSet>(null,1,"test",new String[0]);
  92                 Mockito.doReturn(rs).when(createPS).exec(trans, "ApprovalDAOImpl CREATE", data);
  93                 retVal = daoObj.create(trans, data);
  94                 assertTrue(retVal.status == 1);
  95
  96                 Result<List<ApprovalDAO.Data>> rs1 = new Result<List<ApprovalDAO.Data>>(null,0,"test",new String[0]);
  97                 Mockito.doReturn(rs1).when(createPS).read(trans, "ApprovalDAOImpl CREATE", new Object[]{"testUser"});
  98                 Result<List<ApprovalDAO.Data>> retVal1 = daoObj.readByUser(trans, "testUser");
  99                 assertNull(retVal1);
 100
 101                 Mockito.doReturn(rs1).when(createPS).read(trans, "ApprovalDAOImpl CREATE", new Object[]{"testApprover"});
 102                 retVal1 = daoObj.readByApprover(trans, "testApprover");
 103                 assertNull(retVal1);
 104
 105                 Mockito.doReturn(rs1).when(createPS).read(trans, "ApprovalDAOImpl CREATE", new Object[]{new UUID(0, 0)});
 106                 retVal1 = daoObj.readByTicket(trans, new UUID(0, 0));
 107                 assertNull(retVal1);
 108
 109                 Mockito.doReturn(rs1).when(createPS).read(trans, "ApprovalDAOImpl CREATE", new Object[]{"testStatus"});
 110                 retVal1 = daoObj.readByStatus(trans, "testStatus");
 111                 assertNull(retVal1);
 112         }
 113
 114         @Test
 115         public void testDelete() {
 116                 TimeTaken tt = Mockito.mock(TimeTaken.class);
 117                 Mockito.doReturn(tt).when(trans).start("ApprovalDAO CREATE", Env.REMOTE);
 118                 Mockito.doReturn(tt).when(trans).start("Clear Reset Deque", Env.SUB);
 119                 Mockito.doReturn(tt).when(trans).start("New Cassandra Session", Env.SUB);
 120                 Mockito.doReturn(tt).when(trans).start("Preparing PSInfo CREATE on ApprovalDAO", Env.SUB);
 121                 Mockito.doReturn(tt).when(trans).start("DELETE APPROVAL",Env.REMOTE);
 122                 Mockito.doNothing().when(tt).done();
 123                 ApprovalDAO.Data data  = new ApprovalDAO.Data();
 124
 125                 HistoryDAO historyDAO = Mockito.mock(HistoryDAO.class);
 126                 Result<ResultSet> rs1 = new Result<ResultSet>(null,0,"test",new String[0]);
 127                 Mockito.doReturn(rs1).when(historyDAO).create(Mockito.any(), Mockito.any());
 128
 129                 PSInfo createPS = Mockito.mock(PSInfo.class);
 130                 ResultSet rsObj = new ResultSetImpl();
 131                 Result<ResultSet> rs = new Result<ResultSet>(rsObj,0,"test",new String[0]);
 132                 Mockito.doReturn(rs).when(createPS).exec(trans, "ApprovalDAOImpl READ", data);
 133                 Mockito.doReturn(rs).when(createPS).exec(trans, "ApprovalDAOImpl DELETE", data);
 134
 135                 ApprovalDAOImpl daoObj = new ApprovalDAOImpl(trans, cluster, "test", createPS, historyDAO);
 136 //              data.id
 137                 Result<Void> retVal = daoObj.delete(trans, data, true);
 138                 assertTrue(retVal.status == 0);
 139
 140                 rs = new Result<ResultSet>(rsObj,1,"test",new String[0]);
 141                 Mockito.doReturn(rs).when(createPS).exec(trans, "ApprovalDAOImpl READ", data);
 142                 retVal = daoObj.delete(trans, data, true);
 143                 assertTrue(retVal.status == 1);
 144
 145                 data.status="approved";
 146                 data.memo="test";
 147                 retVal = daoObj.delete(trans, data, false);
 148                 assertTrue(retVal.status == 0);
 149
 150                 daoObj.async(true);
 151                 data.status="denied";
 152                 retVal = daoObj.delete(trans, data, false);
 153                 assertTrue(retVal.status == 0);
 154
 155                 data.status=null;
 156                 retVal = daoObj.delete(trans, data, false);
 157         }
 158
 159         @Test
 160         public void testWasMOdified() {
 161                 TimeTaken tt = Mockito.mock(TimeTaken.class);
 162                 Mockito.doReturn(tt).when(trans).start("ApprovalDAO CREATE", Env.REMOTE);
 163                 Mockito.doReturn(tt).when(trans).start("Clear Reset Deque", Env.SUB);
 164                 Mockito.doReturn(tt).when(trans).start("New Cassandra Session", Env.SUB);
 165                 Mockito.doReturn(tt).when(trans).start("Preparing PSInfo CREATE on ApprovalDAO", Env.SUB);
 166                 Mockito.doReturn(tt).when(trans).start("DELETE APPROVAL",Env.REMOTE);
 167                 Mockito.doNothing().when(tt).done();
 168                 ApprovalDAO.Data data  = new ApprovalDAO.Data();
 169                 PSInfo createPS = Mockito.mock(PSInfo.class);
 170
 171                 HistoryDAO historyDAO = Mockito.mock(HistoryDAO.class);
 172                 Result<ResultSet> rs1 = new Result<ResultSet>(null,0,"test",new String[0]);
 173                 Mockito.doReturn(rs1).when(historyDAO).create(Mockito.any(), Mockito.any());
 174
 175                 ApprovalDAOImpl daoObj = new ApprovalDAOImpl(trans, cluster, "test", createPS, historyDAO);
 176                 daoObj.wasModified(trans, CRUD.create, data, new String[] {"test"});
 177
 178                 daoObj.wasModified(trans, CRUD.create, data, new String[] {});
 179                 daoObj.wasModified(trans, CRUD.create, data, new String[] {null});
 180                 daoObj.wasModified(trans, CRUD.create, data, new String[] {"test",null});
 181                 daoObj.wasModified(trans, CRUD.create, data, new String[] {"test","test"});
 182
 183                 rs1 = new Result<ResultSet>(null,1,"test",new String[0]);
 184                 Mockito.doReturn(rs1).when(historyDAO).create(Mockito.any(), Mockito.any());
 185                 daoObj.wasModified(trans, CRUD.create, data, new String[] {"test","test"});
 186         }
 187
 188         @Test
 189         public void testSecondConstructor() {
 190                 TimeTaken tt = Mockito.mock(TimeTaken.class);
 191                 Mockito.doReturn(tt).when(trans).start("ApprovalDAO CREATE", Env.REMOTE);
 192                 Mockito.doReturn(tt).when(trans).start("Clear Reset Deque", Env.SUB);
 193                 Mockito.doReturn(tt).when(trans).start("New Cassandra Session", Env.SUB);
 194                 Mockito.doReturn(tt).when(trans).start("Preparing PSInfo CREATE on ApprovalDAO", Env.SUB);
 195                 Mockito.doReturn(tt).when(trans).start("DELETE APPROVAL",Env.REMOTE);
 196                 Mockito.doNothing().when(tt).done();
 197                 ApprovalDAO.Data data  = new ApprovalDAO.Data();
 198                 HistoryDAO historyDAO = Mockito.mock(HistoryDAO.class);
 199
 200                 ApprovalDAO daoObj = new ApprovalDAO(trans, historyDAO);
 201         }
 202 }
 203
 204 class ResultSetImpl implements ResultSet{
 205
 206         @Override
 207         public boolean isExhausted() {
 208                 // TODO Auto-generated method stub
 209                 return false;
 210         }
 211
 212         @Override
 213         public boolean isFullyFetched() {
 214                 // TODO Auto-generated method stub
 215                 return false;
 216         }
 217
 218         @Override
 219         public int getAvailableWithoutFetching() {
 220                 // TODO Auto-generated method stub
 221                 return 0;
 222         }
 223
 224         @Override
 225         public ListenableFuture<ResultSet> fetchMoreResults() {
 226                 // TODO Auto-generated method stub
 227                 return null;
 228         }
 229
 230         @Override
 231         public List<Row> all() {
 232                 // TODO Auto-generated method stub
 233                 return null;
 234         }
 235
 236         @Override
 237         public Iterator<Row> iterator() {
 238                 // TODO Auto-generated method stub
 239                 return null;
 240         }
 241
 242         @Override
 243         public ExecutionInfo getExecutionInfo() {
 244                 // TODO Auto-generated method stub
 245                 return null;
 246         }
 247
 248         @Override
 249         public List<ExecutionInfo> getAllExecutionInfo() {
 250                 // TODO Auto-generated method stub
 251                 return null;
 252         }
 253
 254         @Override
 255         public Row one() {
 256                 // TODO Auto-generated method stub
 257                 Row rowObj = Mockito.mock(Row.class);
 258                 Mockito.doReturn(Mockito.mock(ColumnDefinitions.class)).when(rowObj).getColumnDefinitions();
 259                 return rowObj;
 260         }
 261
 262         @Override
 263         public ColumnDefinitions getColumnDefinitions() {
 264                 // TODO Auto-generated method stub
 265                 return null;
 266         }
 267
 268         @Override
 269         public boolean wasApplied() {
 270                 // TODO Auto-generated method stub
 271                 return false;
 272         }
 273
 274 }
 275
 276 class ApprovalDAOImpl extends ApprovalDAO{
 277
 278         public ApprovalDAOImpl(AuthzTrans trans, Cluster cluster, String keyspace,ApprovalDAO.Data data,PSInfo createPS  ) {
 279                 super(trans, cluster, keyspace);
 280                 this.createPS = createPS;
 281                 setPs(this, createPS, "psByUser");
 282                 setPs(this, createPS, "psByApprover");
 283                 setPs(this, createPS, "psByTicket");
 284                 setPs(this, createPS, "psByStatus");
 285         }
 286
 287         public ApprovalDAOImpl(AuthzTrans trans, Cluster cluster, String keyspace,PSInfo readPS  ) {
 288                 super(trans, cluster, keyspace);
 289                 this.readPS = readPS;
 290         }
 291
 292         public ApprovalDAOImpl(AuthzTrans trans, Cluster cluster, String keyspace,PSInfo readPS, HistoryDAO historyDAO  ) {
 293                 super(trans, cluster, keyspace);
 294                 this.deletePS = readPS;
 295                 this.readPS = readPS;
 296                 setHistoryDao(this, historyDAO);
 297                 setSession(this, Mockito.mock(Session.class));
 298         }
 299
 300         public void setPs(ApprovalDAOImpl approvalDaoObj, PSInfo psInfoObj, String methodName) {
 301                 Field nsDaoField;
 302                 try {
 303                         nsDaoField = ApprovalDAO.class.getDeclaredField(methodName);
 304
 305                         nsDaoField.setAccessible(true);
 306                 // remove final modifier from field
 307                 Field modifiersField = Field.class.getDeclaredField("modifiers");
 308                 modifiersField.setAccessible(true);
 309 //              modifiersField.setInt(nsDaoField, nsDaoField.getModifiers() & ~Modifier.FINAL);
 310
 311                 nsDaoField.set(approvalDaoObj, psInfoObj);
 312                 } catch (NoSuchFieldException | SecurityException e) {
 313                         // TODO Auto-generated catch block
 314                         e.printStackTrace();
 315                 } catch (IllegalArgumentException e) {
 316                         // TODO Auto-generated catch block
 317                         e.printStackTrace();
 318                 } catch (IllegalAccessException e) {
 319                         // TODO Auto-generated catch block
 320                         e.printStackTrace();
 321                 }
 322         }
 323
 324         public void setHistoryDao(ApprovalDAOImpl approvalDaoObj, HistoryDAO historyDAO) {
 325                 Field nsDaoField;
 326                 try {
 327                         nsDaoField = ApprovalDAO.class.getDeclaredField("historyDAO");
 328
 329                         nsDaoField.setAccessible(true);
 330                 // remove final modifier from field
 331                 Field modifiersField = Field.class.getDeclaredField("modifiers");
 332                 modifiersField.setAccessible(true);
 333 //              modifiersField.setInt(nsDaoField, nsDaoField.getModifiers() & ~Modifier.FINAL);
 334
 335                 nsDaoField.set(approvalDaoObj, historyDAO);
 336                 } catch (NoSuchFieldException | SecurityException e) {
 337                         // TODO Auto-generated catch block
 338                         e.printStackTrace();
 339                 } catch (IllegalArgumentException e) {
 340                         // TODO Auto-generated catch block
 341                         e.printStackTrace();
 342                 } catch (IllegalAccessException e) {
 343                         // TODO Auto-generated catch block
 344                         e.printStackTrace();
 345                 }
 346         }
 347         public void setSession(ApprovalDAOImpl approvalDaoObj, Session session) {
 348                 Field nsDaoField;
 349                 try {
 350                         nsDaoField = AbsCassDAO.class.getDeclaredField("session");
 351
 352                         nsDaoField.setAccessible(true);
 353                 // remove final modifier from field
 354                 Field modifiersField = Field.class.getDeclaredField("modifiers");
 355                 modifiersField.setAccessible(true);
 356 //              modifiersField.setInt(nsDaoField, nsDaoField.getModifiers() & ~Modifier.FINAL);
 357
 358                 nsDaoField.set(approvalDaoObj, session);
 359                 } catch (NoSuchFieldException | SecurityException e) {
 360                         // TODO Auto-generated catch block
 361                         e.printStackTrace();
 362                 } catch (IllegalArgumentException e) {
 363                         // TODO Auto-generated catch block
 364                         e.printStackTrace();
 365                 } catch (IllegalAccessException e) {
 366                         // TODO Auto-generated catch block
 367                         e.printStackTrace();
 368                 }
 369         }
 370 }