cadi/aaf/src/test/java/org/onap/aaf/cadi/aaf/v2_0/JU_AAFAuthn.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.cadi.aaf.v2_0;
  22
  23 import static org.junit.Assert.assertNull;
  24 import static org.junit.Assert.assertTrue;
  25 import static org.mockito.MockitoAnnotations.initMocks;
  26
  27 import java.io.IOException;
  28
  29 import org.junit.Before;
  30 import org.junit.Test;
  31 import org.mockito.Mock;
  32 import org.mockito.Mockito;
  33 import org.onap.aaf.cadi.AbsUserCache;
  34 import org.onap.aaf.cadi.PropAccess;
  35 import org.onap.aaf.cadi.User;
  36 import org.onap.aaf.cadi.aaf.AAFPermission;
  37 import org.onap.aaf.cadi.client.Future;
  38 import org.onap.aaf.cadi.client.Rcli;
  39 import org.onap.aaf.cadi.principal.BasicPrincipal;
  40
  41 public class JU_AAFAuthn {
  42
  43         @Mock
  44         AAFCon con;
  45
  46         @Mock
  47         AbsUserCache<AAFPermission> cache;
  48
  49         @Mock
  50         PropAccess propaccess;
  51
  52         @Before
  53         public void setUp() {
  54                 initMocks(this);
  55         }
  56
  57         @Test
  58         public void testGetRealm() {
  59                 AAFAuthn authnObj = new AAFAuthn(con);
  60                 String realm = authnObj.getRealm();
  61                 assertNull(realm);
  62         }
  63
  64         @Test
  65         public void testValidateFailure() {
  66                 AAFAuthnImplWithGetUserNull authnObj = new AAFAuthnImplWithGetUserNull(con, cache);
  67                 String realm="";
  68                 try {
  69                         Mockito.doReturn("test").when(propaccess).decrypt("test", false);
  70                         realm = authnObj.validate("test", "test");
  71                         assertNull(realm);
  72                 } catch (Exception e) {
  73                         // TODO Auto-generated catch block
  74                         assertNull( e.getLocalizedMessage());
  75                 }
  76         }
  77
  78         @Test
  79         public void testValidate() {
  80                 AAFAuthnImpl authnObj = new AAFAuthnImpl(con);
  81                 String realm="";
  82                 try {
  83                         Mockito.doReturn("test").when(propaccess).decrypt("test", false);
  84                         Rcli rcliObj = Mockito.mock(Rcli.class);
  85                         Mockito.doReturn(rcliObj).when(con).client();
  86                         Mockito.doReturn(rcliObj).when(rcliObj).forUser(null);
  87                         Future<String> futureObj = Mockito.mock(Future.class);
  88                         Mockito.doReturn(futureObj).when(rcliObj).read( "/authn/basicAuth","text/plain");
  89                         realm = authnObj.validate("test", "test","test");
  90                         assertTrue(realm.contains("user/pass combo invalid"));
  91                 } catch (Exception e) {
  92                         // TODO Auto-generated catch block
  93                         e.printStackTrace();
  94                 }
  95         }
  96
  97         @Test
  98         public void testValidateRevalidate() {
  99                 AAFAuthnImpl authnObj = new AAFAuthnImpl(con);
 100                 String realm="";
 101                 try {
 102                         Mockito.doReturn("test").when(propaccess).decrypt("test", false);
 103                         Rcli rcliObj = Mockito.mock(Rcli.class);
 104                         Mockito.doReturn(rcliObj).when(con).client();
 105                         Mockito.doReturn(rcliObj).when(rcliObj).forUser(null);
 106                         Future<String> futureObj = Mockito.mock(Future.class);
 107                         Mockito.doReturn(futureObj).when(rcliObj).read( "/authn/basicAuth","text/plain");
 108                         Mockito.doReturn(true).when(futureObj).get( 0);
 109                         realm = authnObj.validate("test", "test","test");
 110                         assertNull(realm);
 111                 } catch (Exception e) {
 112                         // TODO Auto-generated catch block
 113                         e.printStackTrace();
 114                 }
 115         }
 116
 117         @Test
 118         public void testValidateValidUser() {
 119                 AAFAuthnImplWithGetUser authnObj = new AAFAuthnImplWithGetUser(con);
 120                 String realm="";
 121                 try {
 122                         Mockito.doReturn("test").when(propaccess).decrypt("test", false);
 123                         realm = authnObj.validate("test", "test","test");
 124                         assertTrue(realm.contains("User already denied"));
 125                 } catch (Exception e) {
 126                         // TODO Auto-generated catch block
 127                         e.printStackTrace();
 128                 }
 129         }
 130
 131         @Test
 132         public void testValidateValidUserNull() {
 133                 AAFAuthnImplWithGetUserNull authnObj = new AAFAuthnImplWithGetUserNull(con);
 134                 String realm="";
 135                 try {
 136                         Mockito.doReturn("test").when(propaccess).decrypt("test", false);
 137                         realm = authnObj.validate("test", "test","test");
 138                         assertNull(realm);
 139                 } catch (Exception e) {
 140                         // TODO Auto-generated catch block
 141                         e.printStackTrace();
 142                 }
 143         }
 144         class AAFAuthnImpl extends AAFAuthn{
 145                 AAFAuthnImpl(AAFCon con) {
 146                         super(con);
 147                         this.access = propaccess;
 148                         // TODO Auto-generated constructor stub
 149                 }
 150
 151                 AAFAuthnImpl(AAFCon con, AbsUserCache cache) {
 152                         super(con, cache);
 153                         this.access = propaccess;
 154                         // TODO Auto-generated constructor stub
 155                 }
 156
 157
 158         }
 159
 160         class AAFAuthnImplWithGetUser extends AAFAuthn{
 161                 AAFAuthnImplWithGetUser(AAFCon con) {
 162                         super(con);
 163                         this.access = propaccess;
 164                         // TODO Auto-generated constructor stub
 165                 }
 166
 167                 AAFAuthnImplWithGetUser(AAFCon con, AbsUserCache cache) {
 168                         super(con, cache);
 169                         this.access = propaccess;
 170                         // TODO Auto-generated constructor stub
 171                 }
 172
 173                 @Override
 174                 protected User getUser(String user, byte[] cred) {
 175                         return new User<>("test",new byte[] {});
 176                 }
 177         }
 178
 179         class AAFAuthnImplWithGetUserNull extends AAFAuthn{
 180                 AAFAuthnImplWithGetUserNull(AAFCon con) {
 181                         super(con);
 182                         this.access = propaccess;
 183                         // TODO Auto-generated constructor stub
 184                 }
 185
 186                 AAFAuthnImplWithGetUserNull(AAFCon con, AbsUserCache cache) {
 187                         super(con, cache);
 188                         this.access = propaccess;
 189                         // TODO Auto-generated constructor stub
 190                 }
 191
 192                 @Override
 193                 protected User getUser(String user, byte[] cred) {
 194                         User user1 = null;
 195                         try {
 196                                 user1 = new User(new BasicPrincipal("test","test"));
 197                         } catch (IOException e) {
 198                                 // TODO Auto-generated catch block
 199                                 e.printStackTrace();
 200                         }
 201                         return user1;
 202                 }
 203         }
 204 }