1 /*******************************************************************************
2 * ============LICENSE_START====================================================
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
11 * * http://www.apache.org/licenses/LICENSE-2.0
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====================================================
21 ******************************************************************************/
22 package org.onap.aaf.cadi.test;
25 import org.mockito.Mock;
26 import org.mockito.MockitoAnnotations;
28 import static org.junit.Assert.*;
29 import static org.mockito.Matchers.*;
30 import static org.mockito.Mockito.*;
32 import java.io.ByteArrayOutputStream;
33 import java.io.PrintStream;
34 import java.security.Principal;
35 import java.util.List;
37 import javax.servlet.http.HttpServletRequest;
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;
53 public class JU_CadiWrap {
56 private HttpServletRequest request;
59 private TafResp tafResp;
62 private TaggedPrincipal principle;
68 public void setUp() throws Exception {
69 MockitoAnnotations.initMocks(this);
71 System.setOut(new PrintStream(new ByteArrayOutputStream()));
75 public void tearDown() {
76 System.setOut(System.out);
79 @SuppressWarnings("unchecked")
81 public void testInstantiate() throws CadiException {
82 Access a = new PropAccess();
83 when(tafResp.getAccess()).thenReturn(a);
85 lur.fishAll(isA(Principal.class), (List<Permission>)isA(List.class));
87 EpiLur lur1 = new EpiLur(lur);
89 CadiWrap wrap = new CadiWrap(request, tafResp, lur1);
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);
97 byte[] arr = {'1','2'};
100 assertEquals(arr, wrap.getCred());
102 wrap.setUser("User1");
103 assertEquals("User1", wrap.getUser());
105 wrap.invalidate("1");
107 assertFalse(wrap.isUserInRole(null));
109 wrap.set(tafResp, lur);
111 wrap.invalidate("2");
113 assertFalse(wrap.isUserInRole("User1"));
117 public void testInstantiateWithPermConverter() throws CadiException {
118 Access a = new PropAccess();
119 when(tafResp.getAccess()).thenReturn(a);
120 when(tafResp.getPrincipal()).thenReturn(principle);
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) { }
136 MapPermConverter pc = new MapPermConverter();
138 CadiWrap wrap = new CadiWrap(request, tafResp, lur1, pc);
140 assertNotNull(wrap.getUserPrincipal());
141 assertNull(wrap.getRemoteUser());
142 assertNull(wrap.getUser());
144 byte[] arr = {'1','2'};
147 assertEquals(arr, wrap.getCred());
149 wrap.setUser("User1");
150 assertEquals("User1", wrap.getUser());
152 wrap.invalidate("1");
153 wrap.setPermConverter(new MapPermConverter());
155 assertTrue(wrap.getLur() instanceof CachingLur);
156 assertTrue(wrap.isUserInRole("User1"));
158 wrap.set(tafResp, lur);
159 assertFalse(wrap.isUserInRole("Perm1"));