2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
6 * ================================================================================
7 * Modifications Copyright (C) 2019 Samsung
8 * ================================================================================
9 * Licensed under the Apache License, Version 2.0 (the "License");
10 * you may not use this file except in compliance with the License.
11 * You may obtain a copy of the License at
13 * http://www.apache.org/licenses/LICENSE-2.0
15 * Unless required by applicable law or agreed to in writing, software
16 * distributed under the License is distributed on an "AS IS" BASIS,
17 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
18 * See the License for the specific language governing permissions and
19 * limitations under the License.
20 * ============LICENSE_END=========================================================
22 package org.onap.policy.controller;
24 import static org.junit.Assert.assertTrue;
25 import java.io.UnsupportedEncodingException;
26 import java.util.ArrayList;
27 import java.util.HashSet;
28 import java.util.List;
30 import javax.servlet.http.HttpServletRequest;
31 import org.junit.Before;
32 import org.junit.Test;
33 import org.mockito.Mockito;
34 import org.onap.policy.common.logging.flexlogger.FlexLogger;
35 import org.onap.policy.common.logging.flexlogger.Logger;
36 import org.onap.policy.model.Roles;
37 import org.onap.policy.xacml.api.pap.OnapPDPGroup;
38 import org.onap.policy.xacml.std.pap.StdPDPGroup;
39 import org.onap.policy.xacml.std.pap.StdPDPGroupStatus;
40 import org.springframework.mock.web.MockHttpServletResponse;
42 public class PDPControllerTest extends Mockito {
44 private static Logger logger = FlexLogger.getLogger(PDPControllerTest.class);
45 private Set<OnapPDPGroup> groupsData;
46 private Set<StdPDPGroup> groups;
47 private static List<Object> rolesdata;
50 public void setUp() throws Exception {
51 logger.info("setUp: Entering");
52 rolesdata = new ArrayList<>();
53 Roles roles = new Roles();
54 roles.setLoginId("Test");
55 roles.setRole("super-admin");
56 Roles roles1 = new Roles();
57 roles1.setLoginId("Test");
58 roles1.setRole("admin");
59 roles1.setScope("['com','Test']");
61 rolesdata.add(roles1);
63 groups = new HashSet<>();
64 StdPDPGroup group = new StdPDPGroup();
65 group.setId("default");
66 group.setDefault(true);
67 group.setName("default");
68 group.setDescription("The default group where new PDP's are put.");
69 group.setStatus(new StdPDPGroupStatus());
71 groupsData = new HashSet<>();
72 for (OnapPDPGroup g : this.groups) {
78 public void testPDPGroupData() {
79 HttpServletRequest request = mock(HttpServletRequest.class);
80 MockHttpServletResponse response = new MockHttpServletResponse();
81 PolicyController controller = mock(PolicyController.class);
82 PDPController pdpController = new PDPController();
83 pdpController.setJunit(true);;
84 pdpController.setPolicyController(controller);
85 pdpController.setGroupsData(groupsData);
86 when(controller.getRoles("Test")).thenReturn(rolesdata);
87 pdpController.getPDPGroupEntityData(request, response);
89 assertTrue(response.getContentAsString() != null
90 && response.getContentAsString().contains("data"));
91 } catch (UnsupportedEncodingException e) {
92 logger.error("Exception Occured" + e);