cadi/aaf/src/test/java/org/onap/aaf/cadi/oauth/test/JU_OAuth2Lur.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.cadi.oauth.test;
  23
  24 import static org.mockito.Mockito.when;
  25 import static org.hamcrest.CoreMatchers.is;
  26 import static org.junit.Assert.assertThat;
  27
  28 import java.security.Principal;
  29 import java.util.ArrayList;
  30 import java.util.List;
  31
  32 import org.junit.Before;
  33 import org.junit.Test;
  34 import org.mockito.Mock;
  35 import org.mockito.MockitoAnnotations;
  36 import org.onap.aaf.cadi.Permission;
  37 import org.onap.aaf.cadi.aaf.AAFPermission;
  38 import org.onap.aaf.cadi.oauth.OAuth2Lur;
  39 import org.onap.aaf.cadi.oauth.OAuth2Principal;
  40 import org.onap.aaf.cadi.oauth.TokenMgr;
  41 import org.onap.aaf.cadi.oauth.TokenPerm;
  42 import org.onap.aaf.cadi.principal.BearerPrincipal;
  43
  44 public class JU_OAuth2Lur {
  45
  46         private List<AAFPermission> aafPerms;
  47         private List<Permission> perms;
  48
  49         @Mock private TokenMgr tmMock;
  50         @Mock private AAFPermission pondMock;
  51         @Mock private Principal princMock;
  52         @Mock private OAuth2Principal oauthPrincMock;
  53         @Mock private BearerPrincipal bearPrincMock;
  54         @Mock private TokenPerm tpMock;
  55
  56         @Before
  57         public void setup() {
  58                 MockitoAnnotations.initMocks(this);
  59         }
  60
  61         @Test
  62         public void test() {
  63                 OAuth2Lur lur = new OAuth2Lur(tmMock);
  64                 lur.createPerm("testPerm");
  65                 lur.createPerm("testPerm1|testPerm2|testPerm3");
  66
  67                 assertThat(lur.fish(princMock, pondMock), is(false));
  68                 assertThat(lur.fish(oauthPrincMock, pondMock), is(false));
  69
  70                 when(oauthPrincMock.tokenPerm()).thenReturn(tpMock);
  71                 assertThat(lur.fish(oauthPrincMock, pondMock), is(false));
  72
  73                 aafPerms = new ArrayList<>();
  74                 aafPerms.add(pondMock);
  75                 aafPerms.add(pondMock);
  76                 when(tpMock.perms()).thenReturn(aafPerms);
  77                 when(pondMock.match(pondMock)).thenReturn(false).thenReturn(true);
  78                 assertThat(lur.fish(oauthPrincMock, pondMock), is(true));
  79
  80                 perms = new ArrayList<>();
  81                 perms.add(pondMock);
  82                 perms.add(pondMock);
  83                 lur.fishAll(oauthPrincMock, perms);
  84
  85                 when(oauthPrincMock.tokenPerm()).thenReturn(null);
  86                 lur.fishAll(oauthPrincMock, perms);
  87
  88                 assertThat(lur.handlesExclusively(pondMock), is(false));
  89
  90                 assertThat(lur.handles(null), is(false));
  91                 assertThat(lur.handles(princMock), is(false));
  92                 assertThat(lur.handles(bearPrincMock), is(false));
  93                 when(bearPrincMock.getBearer()).thenReturn("not null :)");
  94                 assertThat(lur.handles(bearPrincMock), is(true));
  95
  96                 lur.destroy();
  97                 lur.clear(null, null);
  98         }
  99
 100 }