auth/auth-cass/src/test/java/org/onap/aaf/auth/dao/cached/JU_CachedUserRoleDAO.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 package org.onap.aaf.auth.dao.cached;
  22
  23 import static org.junit.Assert.assertEquals;
  24 import static org.mockito.Mockito.when;
  25 import static org.mockito.MockitoAnnotations.initMocks;
  26
  27 import java.util.List;
  28
  29 import org.junit.Before;
  30 import org.junit.Test;
  31 import org.junit.runner.RunWith;
  32 import org.mockito.Mock;
  33 import org.mockito.Mockito;
  34 import org.onap.aaf.auth.dao.CIDAO;
  35 import org.onap.aaf.auth.dao.cass.PermDAO;
  36 import org.onap.aaf.auth.dao.cass.RoleDAO;
  37 import org.onap.aaf.auth.dao.cass.UserRoleDAO;
  38 import org.onap.aaf.auth.dao.cass.UserRoleDAO.Data;
  39 import org.onap.aaf.auth.env.AuthzEnv;
  40 import org.onap.aaf.auth.env.AuthzTrans;
  41 import org.onap.aaf.auth.env.AuthzTransImpl;
  42 import org.onap.aaf.auth.layer.Result;
  43 import org.onap.aaf.cadi.principal.TaggedPrincipal;
  44 import org.onap.aaf.misc.env.LogTarget;
  45 import org.onap.aaf.misc.env.Slot;
  46 import org.powermock.modules.junit4.PowerMockRunner;
  47
  48 @RunWith(PowerMockRunner.class)
  49 public class JU_CachedUserRoleDAO {
  50
  51         @Mock
  52         UserRoleDAO dao;
  53
  54         @Mock
  55         CIDAO<AuthzTrans> info;
  56
  57         @Mock
  58         AuthzTransImpl trans;
  59
  60         @Mock
  61         RoleDAO.Data data;
  62
  63         @Mock
  64         PermDAO.Data permData;
  65
  66         @Before
  67         public void setUp() throws Exception {
  68                 initMocks(this);
  69                 when(trans.debug()).thenReturn(new LogTarget() {
  70
  71             @Override
  72             public void printf(String fmt, Object... vars) {}
  73
  74             @Override
  75             public void log(Throwable e, Object... msgs) {
  76                 e.getMessage();
  77                 e.printStackTrace();
  78                 msgs.toString();
  79
  80             }
  81
  82             @Override
  83             public void log(Object... msgs) {
  84             }
  85
  86             @Override
  87             public boolean isLoggable() {
  88
  89                 return true;
  90             }
  91         });
  92         }
  93
  94         private class TaggedPrincipalStub extends TaggedPrincipal {
  95                 String name="TaggedPrincipalStub";
  96         public TaggedPrincipalStub() { super(); }
  97         public TaggedPrincipalStub(final TagLookup tl) { super(tl); }
  98         @Override public String getName() { return name; }
  99         @Override public String tag() { return null; }
 100     }
 101
 102         @Test
 103         public void testReadName() {
 104                 CachedUserRoleDAO roleDaoObj =new CachedUserRoleDAO(dao,info, 10L);
 105                 Result<List<Data>> retVal1 = new Result<List<Data>>(null,0,"test4",new String[0]);
 106                 Mockito.doReturn(retVal1).when(dao).readByUser(trans, "test4");
 107 //              Mockito.when(roleDaoObj.get(Mockito.any(), Mockito.any(String.class), Mockito.any())).thenReturn(retVal1);
 108                 Result<List<Data>> retVal = roleDaoObj.readByUser(trans, "test4");
 109                 //System.out.println(retVal.status);
 110                 //retVal.status = 0;
 111                 assertEquals("25", Integer.toString(retVal.status));
 112         }
 113
 114         @Test
 115         public void testReadNameUser() {
 116                 CachedUserRoleDAO roleDaoObj =new CachedUserRoleDAO(dao,info, 10L);
 117                 Result<List<Data>> retVal1 = new Result<List<Data>>(null,1,"TaggedPrincipalStub",new String[0]);
 118                 AuthzEnv env = Mockito.mock(AuthzEnv.class);
 119                 AuthzTransImpl transTemp = new AuthzTransImpl(env) {
 120                         @Override
 121                     public<T> T get(Slot slot, T deflt) {
 122                                 Object o=null;
 123                                 return (T)o;
 124                         }
 125
 126                 };
 127                 transTemp.setUser(new TaggedPrincipalStub());
 128                 Mockito.doReturn(retVal1).when(info).touch(trans, null,null);
 129                 Mockito.doReturn(retVal1).when(dao).readByUser(transTemp, "TaggedPrincipalStub");
 130                 roleDaoObj.invalidate("TaggedPrincipalStub");
 131                 Result<List<Data>> retVal = roleDaoObj.readByUser(transTemp, "TaggedPrincipalStub");
 132 //              System.out.println(retVal.status);
 133                 assertEquals("1", Integer.toString(retVal.status));
 134         }
 135
 136         @Test
 137         public void testReadByRoleSuccess() {
 138                 CachedUserRoleDAO roleDaoObj =new CachedUserRoleDAO(dao,info, 0);//Mockito.mock(CachedRoleDAO.class);//
 139                 Result<List<Data>> retVal1 = new Result<List<Data>>(null,1,"test",new String[0]);
 140                 Mockito.doReturn(retVal1).when(dao).readByRole(trans, "");
 141                 roleDaoObj.invalidate("");
 142                 Result<List<Data>> retVal = roleDaoObj.readByRole(trans, "");
 143                 //System.out.println(retVal.status);
 144                 assertEquals("1", Integer.toString(retVal.status));
 145         }
 146         @Test
 147         public void testReadByRoleFailure() {
 148                 CachedUserRoleDAO roleDaoObj =new CachedUserRoleDAO(dao,info, 0);//Mockito.mock(CachedRoleDAO.class);//
 149                 Result<List<Data>> retVal1 = new Result<List<Data>>(null,0,"test1",new String[0]);
 150                 Mockito.doReturn(retVal1).when(dao).readByRole(trans, "");
 151                 roleDaoObj.invalidate("");
 152                 Result<List<Data>> retVal = roleDaoObj.readByRole(trans, "");
 153                 //System.out.println(retVal.status);
 154                 assertEquals("25", Integer.toString(retVal.status));
 155         }
 156
 157         @Test
 158         public void testReadUserInRole() {
 159                 CachedUserRoleDAO roleDaoObj =new CachedUserRoleDAO(dao,info, 10);//Mockito.mock(CachedRoleDAO.class);//
 160                 Result<List<Data>> retVal1 = new Result<List<Data>>(null,0,"TaggedPrincipalStub",new String[0]);
 161                 AuthzEnv env = Mockito.mock(AuthzEnv.class);
 162                 AuthzTransImpl transTemp = new AuthzTransImpl(env) {
 163                         @Override
 164                     public<T> T get(Slot slot, T deflt) {
 165                                 Object o=null;
 166                                 return (T)o;
 167                         }
 168
 169                 };
 170                 transTemp.setUser(new TaggedPrincipalStub());
 171                 Mockito.doReturn(retVal1).when(info).touch(trans, null,null);
 172                 Mockito.doReturn(retVal1).when(dao).readByUserRole(transTemp, "","");
 173                 Mockito.doReturn(retVal1).when(dao).readByUser(transTemp, "TaggedPrincipalStub");
 174                 Result<List<Data>> retVal = roleDaoObj.readUserInRole(transTemp, "TaggedPrincipalStub","");
 175                 //System.out.println(retVal.status);
 176                 assertEquals("25", Integer.toString(retVal.status));
 177         }
 178
 179
 180 }