f1d4aa050ba292d677c3c164c490a74556f93bf5
[portal/sdk.git] /
1 /*
2  * ============LICENSE_START==========================================
3  * ONAP Portal SDK
4  * ===================================================================
5  * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
6  * ===================================================================
7  *
8  * Unless otherwise specified, all software contained herein is licensed
9  * under the Apache License, Version 2.0 (the "License");
10  * you may not use this software except in compliance with the License.
11  * You may obtain a copy of the License at
12  *
13  *             http://www.apache.org/licenses/LICENSE-2.0
14  *
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  *
21  * Unless otherwise specified, all documentation contained herein is licensed
22  * under the Creative Commons License, Attribution 4.0 Intl. (the "License");
23  * you may not use this documentation except in compliance with the License.
24  * You may obtain a copy of the License at
25  *
26  *             https://creativecommons.org/licenses/by/4.0/
27  *
28  * Unless required by applicable law or agreed to in writing, documentation
29  * distributed under the License is distributed on an "AS IS" BASIS,
30  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
31  * See the License for the specific language governing permissions and
32  * limitations under the License.
33  *
34  * ============LICENSE_END============================================
35  *
36  * ECOMP is a trademark and service mark of AT&T Intellectual Property.
37  */
38 package org.onap.portalapp.controller.core;
39
40 import static org.junit.Assert.assertEquals;
41
42 import java.net.URLDecoder;
43 import java.net.URLEncoder;
44 import java.util.ArrayList;
45 import java.util.List;
46
47 import javax.servlet.http.Cookie;
48 import javax.servlet.http.HttpServletRequest;
49 import javax.servlet.http.HttpServletResponse;
50
51 import org.junit.Before;
52 import org.junit.Test;
53 import org.junit.runner.RunWith;
54 import org.mockito.InjectMocks;
55 import org.mockito.Matchers;
56 import org.mockito.Mock;
57 import org.mockito.Mockito;
58 import org.mockito.MockitoAnnotations;
59 import org.onap.portalapp.framework.MockitoTestSuite;
60 import org.onap.portalsdk.core.auth.LoginStrategy;
61 import org.onap.portalsdk.core.command.LoginBean;
62 import org.onap.portalsdk.core.domain.RoleFunction;
63 import org.onap.portalsdk.core.domain.User;
64 import org.onap.portalsdk.core.menu.MenuProperties;
65 import org.onap.portalsdk.core.onboarding.util.PortalApiConstants;
66 import org.onap.portalsdk.core.onboarding.util.PortalApiProperties;
67 import org.onap.portalsdk.core.service.LoginService;
68 import org.onap.portalsdk.core.service.RoleService;
69 import org.onap.portalsdk.core.util.SystemProperties;
70 import org.onap.portalsdk.core.web.support.UserUtils;
71 import org.powermock.api.mockito.PowerMockito;
72 import org.powermock.core.classloader.annotations.PrepareForTest;
73 import org.powermock.modules.junit4.PowerMockRunner;
74 import org.springframework.web.servlet.ModelAndView;
75 import org.springframework.web.util.WebUtils;
76
77 @RunWith(PowerMockRunner.class)
78 @PrepareForTest({ URLDecoder.class, SystemProperties.class, URLEncoder.class, PortalApiProperties.class, WebUtils.class,
79                 UserUtils.class })
80 public class SingleSignOnControllerTest {
81
82         @InjectMocks
83         SingleSignOnController singleSignOnController = new SingleSignOnController();
84
85         @Mock
86         RoleService roleService;
87
88         @Mock
89         LoginService loginService;
90
91         @Mock
92         LoginStrategy loginStrategy;
93
94         @Mock
95         URLDecoder uRLDecoder;
96
97         @Before
98         public void setup() {
99                 MockitoAnnotations.initMocks(this);
100         }
101
102         MockitoTestSuite mockitoTestSuite = new MockitoTestSuite();
103
104         HttpServletRequest mockedRequest = mockitoTestSuite.getMockedRequest();
105         HttpServletResponse mockedResponse = mockitoTestSuite.getMockedResponse();
106         NullPointerException nullPointerException = new NullPointerException();
107
108         @Test(expected = java.lang.SecurityException.class)
109         public void singleSignOnLoginExceptionTest() throws Exception {
110                 Mockito.when(mockedRequest.getParameter("forwardURL")).thenReturn("Test");
111                 PowerMockito.mockStatic(URLDecoder.class);
112                 PowerMockito.mockStatic(SystemProperties.class);
113                 Mockito.when(URLDecoder.decode(null, "UTF-8")).thenReturn("Test");
114                 Mockito.when(SystemProperties.containsProperty(SystemProperties.APP_BASE_URL)).thenReturn(true);
115                 Mockito.when(SystemProperties.getProperty(SystemProperties.APP_BASE_URL)).thenReturn("http://TestUrl");
116                 Mockito.when(SystemProperties.getProperty(SystemProperties.COOKIE_DOMAIN)).thenReturn("te");
117                 singleSignOnController.singleSignOnLogin(mockedRequest);
118
119         }
120
121         @Test
122         public void singleSignOnLoginTest() throws Exception {
123                 Mockito.when(mockedRequest.getParameter("forwardURL")).thenReturn("Test");
124                 PowerMockito.mockStatic(URLDecoder.class);
125                 PowerMockito.mockStatic(SystemProperties.class);
126                 PowerMockito.mockStatic(URLEncoder.class);
127                 PowerMockito.mockStatic(PortalApiProperties.class);
128                 Mockito.when(URLDecoder.decode(null, "UTF-8")).thenReturn("Test");
129                 Mockito.when(SystemProperties.containsProperty(SystemProperties.APP_BASE_URL)).thenReturn(true);
130                 Mockito.when(SystemProperties.getProperty(SystemProperties.APP_BASE_URL)).thenReturn("http://TestUrl");
131                 Mockito.when(SystemProperties.getProperty(SystemProperties.COOKIE_DOMAIN)).thenReturn("TestUrl");
132                 Mockito.when(URLEncoder.encode("http://TestUrl/Test", "UTF-8")).thenReturn("encodeTestUrl");
133                 Mockito.when(PortalApiProperties.getProperty(PortalApiConstants.UEB_APP_KEY)).thenReturn("uebkey");
134                 Mockito.when(PortalApiProperties.getProperty(PortalApiConstants.ECOMP_REDIRECT_URL))
135                                 .thenReturn("http://test.com/roles");
136                 ModelAndView expectedResults = singleSignOnController.singleSignOnLogin(mockedRequest);
137                 assertEquals(expectedResults.getViewName(),
138                                 "redirect:http://test.com/process_csp?uebAppKey=uebkey&redirectUrl=http%3A%2F%2FTestUrl%2FTest");
139         }
140
141         @Test
142         public void singleSignOnLoginIfUrlIsNotAppBasedTest() throws Exception {
143                 Mockito.when(mockedRequest.getParameter("forwardURL")).thenReturn("Test");
144                 PowerMockito.mockStatic(URLDecoder.class);
145                 PowerMockito.mockStatic(SystemProperties.class);
146                 PowerMockito.mockStatic(URLEncoder.class);
147                 PowerMockito.mockStatic(PortalApiProperties.class);
148                 Mockito.when(URLDecoder.decode(null, "UTF-8")).thenReturn("Test");
149                 Mockito.when(SystemProperties.containsProperty(SystemProperties.APP_BASE_URL)).thenReturn(false);
150                 Mockito.when(SystemProperties.getProperty(SystemProperties.COOKIE_DOMAIN)).thenReturn("test.com");
151                 Mockito.when(PortalApiProperties.getProperty(PortalApiConstants.UEB_APP_KEY)).thenReturn("uebkey");
152                 Mockito.when(PortalApiProperties.getProperty(PortalApiConstants.ECOMP_REDIRECT_URL))
153                                 .thenReturn("http://test.com/roles");
154                 StringBuffer stringBuffer = new StringBuffer("http://test.com/testSDK");
155                 Mockito.when(mockedRequest.getRequestURL()).thenReturn(stringBuffer);
156                 ModelAndView expectedResults = singleSignOnController.singleSignOnLogin(mockedRequest);
157                 assertEquals(expectedResults.getViewName(),
158                                 "redirect:http://test.com/process_csp?uebAppKey=uebkey&redirectUrl=http%3A%2F%2Ftest.com%2FtestSDK");
159         }
160
161         @Test
162         public void singleSignOnTest() throws Exception {
163                 Mockito.when(mockedRequest.getParameter("forwardURL")).thenReturn("http://Test.com");
164                 PowerMockito.mockStatic(URLDecoder.class);
165                 PowerMockito.mockStatic(SystemProperties.class);
166                 PowerMockito.mockStatic(WebUtils.class);
167                 Mockito.when(URLDecoder.decode(null, "UTF-8")).thenReturn("http://Test.com");
168                 Mockito.when(mockedRequest.getParameter("redirectToPortal")).thenReturn(null);
169                 Mockito.when(SystemProperties.containsProperty(SystemProperties.APP_BASE_URL)).thenReturn(true);
170                 Mockito.when(SystemProperties.getProperty(SystemProperties.APP_BASE_URL)).thenReturn("http://TestUrl");
171                 Mockito.when(SystemProperties.getProperty(SystemProperties.COOKIE_DOMAIN)).thenReturn("Test.com");
172                 Mockito.when(WebUtils.getCookie(mockedRequest, "EPService")).thenReturn(new Cookie("test", "test"));
173                 User user = new User();
174                 user.setOrgUserId("test12");
175                 Mockito.when(UserUtils.getUserSession(mockedRequest)).thenReturn(user);
176                 ModelAndView expectedResults = singleSignOnController.singleSignOnLogin(mockedRequest);
177                 assertEquals(expectedResults.getViewName(), "redirect:http://Test.com");
178
179         }
180
181         @Test
182         public void singleSignOnIfUserNullTest() throws Exception {
183                 User user = null;
184                 Mockito.when(UserUtils.getUserSession(mockedRequest)).thenReturn(user);
185                 Mockito.when(mockedRequest.getParameter("forwardURL")).thenReturn("http://Test.com");
186                 PowerMockito.mockStatic(URLDecoder.class);
187                 PowerMockito.mockStatic(WebUtils.class);
188                 PowerMockito.mockStatic(SystemProperties.class);
189
190                 Mockito.when(URLDecoder.decode(null, "UTF-8")).thenReturn("http://Test.com");
191                 Mockito.when(WebUtils.getCookie(mockedRequest, "EPService")).thenReturn(new Cookie("test", "test"));
192                 Mockito.when(UserUtils.getUserSession(mockedRequest)).thenReturn(user);
193                 Mockito.when(SystemProperties.getProperty(SystemProperties.AUTHENTICATION_MECHANISM)).thenReturn("testauth");
194                 Mockito.when(loginStrategy.getUserId(mockedRequest)).thenReturn("test1234");
195                 Mockito.when(mockedRequest.getAttribute(MenuProperties.MENU_PROPERTIES_FILENAME_KEY)).thenReturn("test");
196                 LoginBean commandBean = new LoginBean();
197                 commandBean.setUserid("test1234");
198                 commandBean.setUser(null);
199                 Mockito.when(loginService.findUser(Matchers.any(), Matchers.anyString(), Matchers.anyMap()))
200                                 .thenReturn(commandBean);
201                 List<RoleFunction> roleFunctionList = new ArrayList<>();
202                 Mockito.when(roleService.getRoleFunctions("test1234")).thenReturn(roleFunctionList);
203                 ModelAndView expectedResults = singleSignOnController.singleSignOnLogin(mockedRequest);
204                 assertEquals(expectedResults.getViewName(), "redirect:null?noUserError=Yes");
205         }
206
207         @Test
208         public void singleSignOnIfUserNotNullTest() throws Exception {
209                 User user = null;
210                 Mockito.when(UserUtils.getUserSession(mockedRequest)).thenReturn(user);
211                 Mockito.when(mockedRequest.getParameter("forwardURL")).thenReturn("http://Test.com");
212                 PowerMockito.mockStatic(URLDecoder.class);
213                 PowerMockito.mockStatic(WebUtils.class);
214                 PowerMockito.mockStatic(SystemProperties.class);
215                 PowerMockito.mockStatic(UserUtils.class);
216
217                 Mockito.when(URLDecoder.decode(null, "UTF-8")).thenReturn("http://Test.com");
218                 Mockito.when(WebUtils.getCookie(mockedRequest, "EPService")).thenReturn(new Cookie("test", "test"));
219                 Mockito.when(UserUtils.getUserSession(mockedRequest)).thenReturn(user);
220                 Mockito.when(SystemProperties.getProperty(SystemProperties.AUTHENTICATION_MECHANISM)).thenReturn("testauth");
221                 Mockito.when(loginStrategy.getUserId(mockedRequest)).thenReturn("test1234");
222                 Mockito.when(mockedRequest.getAttribute(MenuProperties.MENU_PROPERTIES_FILENAME_KEY)).thenReturn("test");
223                 LoginBean commandBean = new LoginBean();
224                 commandBean.setUserid("test1234");
225                 User user1 = new User();
226                 user1.setId((long) 1);
227                 commandBean.setUser(user1);
228                 Mockito.when(loginService.findUser(Matchers.any(), Matchers.anyString(), Matchers.anyMap()))
229                                 .thenReturn(commandBean);
230                 List<RoleFunction> roleFunctionList = new ArrayList<>();
231                 Mockito.when(roleService.getRoleFunctions("test1234")).thenReturn(roleFunctionList);
232                 ModelAndView expectedResults = singleSignOnController.singleSignOnLogin(mockedRequest);
233                 assertEquals(expectedResults.getViewName(), "redirect:http://Test.com");
234         }
235
236         @Test
237         public void singleSignOnIfUserNotNullAndAuthNullTest() throws Exception {
238                 User user = null;
239                 Mockito.when(UserUtils.getUserSession(mockedRequest)).thenReturn(user);
240                 Mockito.when(mockedRequest.getParameter("forwardURL")).thenReturn("http://Test.com");
241                 PowerMockito.mockStatic(URLDecoder.class);
242                 PowerMockito.mockStatic(WebUtils.class);
243                 PowerMockito.mockStatic(SystemProperties.class);
244                 PowerMockito.mockStatic(UserUtils.class);
245
246                 Mockito.when(URLDecoder.decode(null, "UTF-8")).thenReturn("http://Test.com");
247                 Mockito.when(WebUtils.getCookie(mockedRequest, "EPService")).thenReturn(new Cookie("test", "test"));
248                 Mockito.when(UserUtils.getUserSession(mockedRequest)).thenReturn(user);
249                 Mockito.when(SystemProperties.getProperty(SystemProperties.AUTHENTICATION_MECHANISM)).thenReturn(null);
250                 Mockito.when(loginStrategy.getUserId(mockedRequest)).thenReturn("test1234");
251                 Mockito.when(mockedRequest.getAttribute(MenuProperties.MENU_PROPERTIES_FILENAME_KEY)).thenReturn("test");
252                 LoginBean commandBean = new LoginBean();
253                 commandBean.setUserid("test1234");
254                 User user1 = new User();
255                 user1.setId((long) 1);
256                 commandBean.setUser(user1);
257                 Mockito.when(loginService.findUser(Matchers.any(), Matchers.anyString(), Matchers.anyMap()))
258                                 .thenReturn(commandBean);
259                 List<RoleFunction> roleFunctionList = new ArrayList<>();
260                 Mockito.when(roleService.getRoleFunctions("test1234")).thenReturn(roleFunctionList);
261                 ModelAndView expectedResults = singleSignOnController.singleSignOnLogin(mockedRequest);
262                 assertEquals(expectedResults.getViewName(), "redirect:http://Test.com");
263         }
264
265         @Test
266         public void singleSignOnIfUserNotNullAndAuthCSPTest() throws Exception {
267                 singleSignOnController.setViewName("test");
268                 singleSignOnController.setWelcomeView("welcome");
269                 assertEquals(singleSignOnController.getViewName(), "test");
270                 assertEquals(singleSignOnController.getWelcomeView(), "welcome");
271                 User user = null;
272                 Mockito.when(UserUtils.getUserSession(mockedRequest)).thenReturn(user);
273                 Mockito.when(mockedRequest.getParameter("forwardURL")).thenReturn("http://Test.com");
274                 PowerMockito.mockStatic(URLDecoder.class);
275                 PowerMockito.mockStatic(WebUtils.class);
276                 PowerMockito.mockStatic(SystemProperties.class);
277                 PowerMockito.mockStatic(UserUtils.class);
278                 Mockito.when(URLDecoder.decode(null, "UTF-8")).thenReturn("http://Test.com");
279                 Mockito.when(WebUtils.getCookie(mockedRequest, "EPService")).thenReturn(new Cookie("test", "test"));
280                 Mockito.when(UserUtils.getUserSession(mockedRequest)).thenReturn(user);
281                 Mockito.when(SystemProperties.getProperty(SystemProperties.AUTHENTICATION_MECHANISM)).thenReturn("CSP");
282                 Mockito.when(loginStrategy.getUserId(mockedRequest)).thenReturn("test1234");
283                 Mockito.when(mockedRequest.getAttribute(MenuProperties.MENU_PROPERTIES_FILENAME_KEY)).thenReturn("test");
284                 LoginBean commandBean = new LoginBean();
285                 commandBean.setUserid("test1234");
286                 User user1 = new User();
287                 user1.setId((long) 1);
288                 commandBean.setUser(user1);
289                 Mockito.when(loginService.findUser(Matchers.any(), Matchers.anyString(), Matchers.anyMap()))
290                                 .thenReturn(commandBean);
291                 List<RoleFunction> roleFunctionList = new ArrayList<>();
292                 Mockito.when(roleService.getRoleFunctions("test1234")).thenReturn(roleFunctionList);
293                 ModelAndView expectedResults = singleSignOnController.singleSignOnLogin(mockedRequest);
294                 assertEquals(expectedResults.getViewName(), "redirect:http://Test.com");
295         }
296
297 }