cadi/core/src/test/java/org/onap/aaf/cadi/test/JU_CadiWrap.java

   1 /*******************************************************************************
   2  * ============LICENSE_START====================================================
   3  * * org.onap.aaf
   4  * * ===========================================================================
   5  * * Copyright © 2017 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.test;
  23
  24 import org.junit.*;
  25 import org.mockito.Mock;
  26 import org.mockito.MockitoAnnotations;
  27
  28 import static org.junit.Assert.*;
  29 import static org.mockito.Matchers.*;
  30 import static org.mockito.Mockito.*;
  31
  32 import java.io.ByteArrayOutputStream;
  33 import java.io.PrintStream;
  34 import java.security.Principal;
  35 import java.util.List;
  36
  37 import javax.servlet.http.HttpServletRequest;
  38
  39 import org.onap.aaf.cadi.Access;
  40 import org.onap.aaf.cadi.CachingLur;
  41 import org.onap.aaf.cadi.CadiException;
  42 import org.onap.aaf.cadi.CadiWrap;
  43 import org.onap.aaf.cadi.Lur;
  44 import org.onap.aaf.cadi.Permission;
  45 import org.onap.aaf.cadi.PropAccess;
  46 import org.onap.aaf.cadi.User;
  47 import org.onap.aaf.cadi.CachedPrincipal.Resp;
  48 import org.onap.aaf.cadi.filter.MapPermConverter;
  49 import org.onap.aaf.cadi.lur.EpiLur;
  50 import org.onap.aaf.cadi.principal.TaggedPrincipal;
  51 import org.onap.aaf.cadi.taf.TafResp;
  52
  53 public class JU_CadiWrap {
  54
  55         @Mock
  56         private HttpServletRequest request;
  57
  58         @Mock
  59         private TafResp tafResp;
  60
  61         @Mock
  62         private TaggedPrincipal principle;
  63
  64         @Mock
  65         private Lur lur;
  66
  67         @Before
  68         public void setUp() throws Exception {
  69                 MockitoAnnotations.initMocks(this);
  70
  71                 System.setOut(new PrintStream(new ByteArrayOutputStream()));
  72         }
  73
  74         @After
  75         public void tearDown() {
  76                 System.setOut(System.out);
  77         }
  78
  79         @SuppressWarnings("unchecked")
  80         @Test
  81         public void testInstantiate() throws CadiException {
  82                 Access a = new PropAccess();
  83                 when(tafResp.getAccess()).thenReturn(a);
  84
  85                 lur.fishAll(isA(Principal.class), (List<Permission>)isA(List.class));
  86
  87                 EpiLur lur1 = new EpiLur(lur);
  88
  89                 CadiWrap wrap = new CadiWrap(request, tafResp, lur1);
  90
  91                 assertNull(wrap.getUserPrincipal());
  92                 assertNull(wrap.getRemoteUser());
  93                 assertNull(wrap.getUser());
  94                 assertEquals(wrap.getPermissions(principle).size(), 0);
  95                 assertTrue(wrap.access() instanceof PropAccess);
  96
  97                 byte[] arr = {'1','2'};
  98                 wrap.setCred(arr);
  99
 100                 assertEquals(arr, wrap.getCred());
 101
 102                 wrap.setUser("User1");
 103                 assertEquals("User1", wrap.getUser());
 104
 105                 wrap.invalidate("1");
 106
 107                 assertFalse(wrap.isUserInRole(null));
 108
 109                 wrap.set(tafResp, lur);
 110
 111                 wrap.invalidate("2");
 112
 113                 assertFalse(wrap.isUserInRole("User1"));
 114         }
 115
 116         @Test
 117         public void testInstantiateWithPermConverter() throws CadiException {
 118                 Access a = new PropAccess();
 119                 when(tafResp.getAccess()).thenReturn(a);
 120                 when(tafResp.getPrincipal()).thenReturn(principle);
 121
 122                 // Anonymous object for testing purposes
 123                 CachingLur<Permission> lur1 = new CachingLur<Permission>() {
 124                         @Override public Permission createPerm(String p) { return null; }
 125                         @Override public boolean fish(Principal bait, Permission pond) { return true; }
 126                         @Override public void fishAll(Principal bait, List<Permission> permissions) { }
 127                         @Override public void destroy() { }
 128                         @Override public boolean handlesExclusively(Permission pond) { return false; }
 129                         @Override public boolean handles(Principal principal) { return false; }
 130                         @Override public void remove(String user) { }
 131                         @Override public Resp reload(User<Permission> user) { return null; }
 132                         @Override public void setDebug(String commaDelimIDsOrNull) { }
 133                         @Override public void clear(Principal p, StringBuilder sb) { }
 134                 };
 135
 136                 MapPermConverter pc = new MapPermConverter();
 137
 138                 CadiWrap wrap = new CadiWrap(request, tafResp, lur1, pc);
 139
 140                 assertNotNull(wrap.getUserPrincipal());
 141                 assertNull(wrap.getRemoteUser());
 142                 assertNull(wrap.getUser());
 143
 144                 byte[] arr = {'1','2'};
 145                 wrap.setCred(arr);
 146
 147                 assertEquals(arr, wrap.getCred());
 148
 149                 wrap.setUser("User1");
 150                 assertEquals("User1", wrap.getUser());
 151
 152                 wrap.invalidate("1");
 153                 wrap.setPermConverter(new MapPermConverter());
 154
 155                 assertTrue(wrap.getLur() instanceof CachingLur);
 156                 assertTrue(wrap.isUserInRole("User1"));
 157
 158                 wrap.set(tafResp, lur);
 159                 assertFalse(wrap.isUserInRole("Perm1"));
 160         }
 161 }