auth/auth-oauth/src/test/java/org/onap/aaf/auth/oauth/service/JU_JSONPermLoaderFactoryTest.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.oauth.service;
  22
  23 import static org.junit.Assert.assertEquals;
  24 import static org.mockito.Mockito.only;
  25 import static org.mockito.Mockito.verify;
  26 import static org.mockito.Mockito.when;
  27 import static org.mockito.MockitoAnnotations.initMocks;
  28
  29 import java.util.ArrayList;
  30 import java.util.HashSet;
  31 import java.util.List;
  32 import java.util.Set;
  33
  34 import javax.xml.ws.handler.MessageContext.Scope;
  35
  36 import org.junit.Before;
  37 import org.junit.Test;
  38 import org.mockito.Mock;
  39 import org.onap.aaf.auth.common.Define;
  40 import org.onap.aaf.auth.dao.cass.NsSplit;
  41 import org.onap.aaf.auth.dao.cass.PermDAO;
  42 import org.onap.aaf.auth.dao.hl.Question;
  43 import org.onap.aaf.auth.env.AuthzEnv;
  44 import org.onap.aaf.auth.env.AuthzTrans;
  45 import org.onap.aaf.auth.layer.Result;
  46 import org.onap.aaf.cadi.Access;
  47 import org.onap.aaf.cadi.CadiException;
  48 import org.onap.aaf.cadi.aaf.v2_0.AAFCon;
  49 import org.onap.aaf.cadi.client.Future;
  50 import org.onap.aaf.cadi.client.Rcli;
  51 import org.onap.aaf.cadi.config.Config;
  52 import org.onap.aaf.misc.env.APIException;
  53 import org.onap.aaf.misc.env.Env;
  54 import org.onap.aaf.misc.env.TimeTaken;
  55
  56 public class JU_JSONPermLoaderFactoryTest {
  57     @Mock
  58     private AAFCon<?> aafcon;
  59     @Mock
  60     private AuthzTrans trans;
  61     @Mock
  62     private TimeTaken tt;
  63     @Mock
  64     Rcli c;
  65     @Mock
  66     private Future fs;
  67     @Mock
  68     private Question question;
  69     @Mock
  70     private Result<NsSplit> rdns;
  71     private NsSplit nss;
  72
  73     private Access access;
  74
  75     @Before
  76     public void setup() throws CadiException {
  77         access = new AuthzEnv();
  78         Define.set(access);
  79         initMocks(this);
  80         nss = new NsSplit("APPLICATION", "APPLICATION");
  81     }
  82
  83     @Test
  84     public void testRemoteWithTimeOut() throws APIException, CadiException {
  85         when(trans.start("Call AAF Service", Env.REMOTE)).thenReturn(tt);
  86         when(aafcon.clientAs(Config.AAF_DEFAULT_VERSION, trans.getUserPrincipal())).thenReturn(c);
  87         when(c.read("/authz/perms/user/null?scopes=APPLICATION:HANDLER",
  88                 "application/Perms+json;charset=utf-8;version=2.0")).thenReturn(fs);
  89         when(fs.get(0)).thenReturn(true);
  90
  91         Set<String> scopes = new HashSet<String>();
  92         scopes.add(Scope.APPLICATION.toString());
  93         scopes.add(Scope.HANDLER.toString());
  94
  95         JSONPermLoader factory = JSONPermLoaderFactory.remote(aafcon, 0);
  96
  97         Result<String> loadJSONPerms = factory.loadJSONPerms(trans, null, scopes);
  98
  99         assertEquals(0, loadJSONPerms.status);
 100
 101         verify(tt, only()).done();
 102     }
 103
 104     @Test
 105     public void testRemoteWith404() throws APIException, CadiException {
 106         when(trans.start("Call AAF Service", Env.REMOTE)).thenReturn(tt);
 107         when(aafcon.clientAs(Config.AAF_DEFAULT_VERSION, trans.getUserPrincipal())).thenReturn(c);
 108         when(c.read("/authz/perms/user/null?scopes=APPLICATION:HANDLER",
 109                 "application/Perms+json;charset=utf-8;version=2.0")).thenReturn(fs);
 110         when(fs.get(0)).thenReturn(false);
 111         when(fs.code()).thenReturn(404);
 112
 113         Set<String> scopes = new HashSet<String>();
 114         scopes.add(Scope.APPLICATION.toString());
 115         scopes.add(Scope.HANDLER.toString());
 116
 117         JSONPermLoader factory = JSONPermLoaderFactory.remote(aafcon, 0);
 118
 119         Result<String> loadJSONPerms = factory.loadJSONPerms(trans, null, scopes);
 120
 121         assertEquals(Result.ERR_NotFound, loadJSONPerms.status);
 122
 123         verify(tt, only()).done();
 124     }
 125
 126     @Test
 127     public void testRemote() throws APIException, CadiException {
 128         when(trans.start("Call AAF Service", Env.REMOTE)).thenReturn(tt);
 129         when(aafcon.clientAs(Config.AAF_DEFAULT_VERSION, trans.getUserPrincipal())).thenReturn(c);
 130         when(c.read("/authz/perms/user/null?scopes=APPLICATION:HANDLER",
 131                 "application/Perms+json;charset=utf-8;version=2.0")).thenReturn(fs);
 132         when(fs.get(0)).thenReturn(false);
 133
 134         Set<String> scopes = new HashSet<String>();
 135         scopes.add(Scope.APPLICATION.toString());
 136         scopes.add(Scope.HANDLER.toString());
 137
 138         JSONPermLoader factory = JSONPermLoaderFactory.remote(aafcon, 0);
 139
 140         Result<String> loadJSONPerms = factory.loadJSONPerms(trans, null, scopes);
 141
 142         assertEquals(Result.ERR_Backend, loadJSONPerms.status);
 143
 144         verify(tt, only()).done();
 145     }
 146
 147     @Test
 148     public void testDirectWhenPdNotOk() throws APIException, CadiException {
 149
 150         Result<List<PermDAO.Data>> pd = Result.create(null, Result.ERR_Backend, "details", "vars");
 151
 152         when(question.getPermsByUser(trans, "user", false)).thenReturn(pd);
 153         when(trans.start("Cached DB Perm lookup", Env.SUB)).thenReturn(tt);
 154
 155         Set<String> scopes = new HashSet<String>();
 156         scopes.add(Scope.APPLICATION.toString());
 157         scopes.add(Scope.HANDLER.toString());
 158
 159         JSONPermLoader factory = JSONPermLoaderFactory.direct(question);
 160
 161         Result<String> loadJSONPerms = factory.loadJSONPerms(trans, "user", scopes);
 162
 163         assertEquals(Result.ERR_Backend, loadJSONPerms.status);
 164
 165         verify(tt, only()).done();
 166     }
 167
 168     @Test
 169     public void testDirectWhenPdOk() throws APIException, CadiException {
 170
 171         when(trans.start("Cached DB Perm lookup", Env.SUB)).thenReturn(tt);
 172         when(question.deriveNsSplit(trans, "name")).thenReturn(rdns);
 173         when(rdns.isOKhasData()).thenReturn(false);
 174
 175         List<PermDAO.Data> list = new ArrayList<PermDAO.Data>();
 176         list.add(new PermDAO.Data(nss, "instance", "action"));
 177         list.add(new PermDAO.Data(nss, "instance", "action"));
 178
 179         Result<List<PermDAO.Data>> pd = Result.create(list, Result.OK, "details", "vars");
 180
 181         when(question.getPermsByUser(trans, "user", false)).thenReturn(pd);
 182
 183         Set<String> scopes = new HashSet<String>();
 184         scopes.add(Scope.APPLICATION.toString());
 185         scopes.add(Scope.HANDLER.toString());
 186
 187         JSONPermLoader factory = JSONPermLoaderFactory.direct(question);
 188
 189         Result<String> loadJSONPerms = factory.loadJSONPerms(trans, "user", scopes);
 190
 191         assertEquals(Result.OK, loadJSONPerms.status);
 192         assertEquals("Success", loadJSONPerms.details);
 193         assertEquals(
 194                 "{\"perm\":[{\"ns\":\"APPLICATION\",\"type\":\"APPLICATION\",\"instance\":\"instance\",\"action\":\"action\"},{\"ns\":\"APPLICATION\",\"type\":\"APPLICATION\",\"instance\":\"instance\",\"action\":\"action\"}]}",
 195                 loadJSONPerms.value);
 196
 197         verify(tt, only()).done();
 198     }
 199
 200 }