Newer
Older
Alinga Yeung
committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
/*
************************************************************************
******************* CANADIAN ASTRONOMY DATA CENTRE *******************
************** CENTRE CANADIEN DE DONNÉES ASTRONOMIQUES **************
*
* (c) 2015. (c) 2015.
* Government of Canada Gouvernement du Canada
* National Research Council Conseil national de recherches
* Ottawa, Canada, K1A 0R6 Ottawa, Canada, K1A 0R6
* All rights reserved Tous droits réservés
*
* NRC disclaims any warranties, Le CNRC dénie toute garantie
* expressed, implied, or énoncée, implicite ou légale,
* statutory, of any kind with de quelque nature que ce
* respect to the software, soit, concernant le logiciel,
* including without limitation y compris sans restriction
* any warranty of merchantability toute garantie de valeur
* or fitness for a particular marchande ou de pertinence
* purpose. NRC shall not be pour un usage particulier.
* liable in any event for any Le CNRC ne pourra en aucun cas
* damages, whether direct or être tenu responsable de tout
* indirect, special or general, dommage, direct ou indirect,
* consequential or incidental, particulier ou général,
* arising from the use of the accessoire ou fortuit, résultant
* software. Neither the name de l'utilisation du logiciel. Ni
* of the National Research le nom du Conseil National de
* Council of Canada nor the Recherches du Canada ni les noms
* names of its contributors may de ses participants ne peuvent
* be used to endorse or promote être utilisés pour approuver ou
* products derived from this promouvoir les produits dérivés
* software without specific prior de ce logiciel sans autorisation
* written permission. préalable et particulière
* par écrit.
*
* This file is part of the Ce fichier fait partie du projet
* OpenCADC project. OpenCADC.
*
* OpenCADC is free software: OpenCADC est un logiciel libre ;
* you can redistribute it and/or vous pouvez le redistribuer ou le
* modify it under the terms of modifier suivant les termes de
* the GNU Affero General Public la “GNU Affero General Public
* License as published by the License” telle que publiée
* Free Software Foundation, par la Free Software Foundation
* either version 3 of the : soit la version 3 de cette
* License, or (at your option) licence, soit (à votre gré)
* any later version. toute version ultérieure.
*
* OpenCADC is distributed in the OpenCADC est distribué
* hope that it will be useful, dans l’espoir qu’il vous
* but WITHOUT ANY WARRANTY; sera utile, mais SANS AUCUNE
* without even the implied GARANTIE : sans même la garantie
* warranty of MERCHANTABILITY implicite de COMMERCIALISABILITÉ
* or FITNESS FOR A PARTICULAR ni d’ADÉQUATION À UN OBJECTIF
* PURPOSE. See the GNU Affero PARTICULIER. Consultez la Licence
* General Public License for Générale Publique GNU Affero
* more details. pour plus de détails.
*
* You should have received Vous devriez avoir reçu une
* a copy of the GNU Affero copie de la Licence Générale
* General Public License along Publique GNU Affero avec
* with OpenCADC. If not, see OpenCADC ; si ce n’est
* <http://www.gnu.org/licenses/>. pas le cas, consultez :
* <http://www.gnu.org/licenses/>.
*
*
************************************************************************
*/
package ca.nrc.cadc.ac.server.web;
Alinga Yeung
committed
import ca.nrc.cadc.ac.UserNotFoundException;
Alinga Yeung
committed
import ca.nrc.cadc.ac.server.UserPersistence;
Alinga Yeung
committed
import ca.nrc.cadc.ac.server.ldap.LdapUserDAO;
Alinga Yeung
committed
import ca.nrc.cadc.auth.HttpPrincipal;
import ca.nrc.cadc.util.StringUtil;
import org.junit.Test;
import javax.security.auth.Subject;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.security.Principal;
import java.security.PrivilegedExceptionAction;
Alinga Yeung
committed
import java.util.ArrayList;
import java.util.List;
Alinga Yeung
committed
import static org.easymock.EasyMock.*;
public class ResetPasswordServletTest
{
private static final String EMAIL_ADDRESS = "email@canada.ca";
Alinga Yeung
committed
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
public void testSubjectAndEmailAddress(final Subject subject, final String emailAddress,
int responseStatus) throws Exception
{
@SuppressWarnings("serial")
final ResetPasswordServlet testSubject = new ResetPasswordServlet()
{
@Override
Subject getSubject(final HttpServletRequest request)
{
return subject;
}
};
final HttpServletRequest mockRequest =
createMock(HttpServletRequest.class);
final HttpServletResponse mockResponse =
createMock(HttpServletResponse.class);
expect(mockRequest.getPathInfo()).andReturn("users/CADCtest").once();
expect(mockRequest.getMethod()).andReturn("POST").once();
expect(mockRequest.getRemoteAddr()).andReturn("mysite.com").once();
if (!StringUtil.hasText(emailAddress))
{
expect(mockRequest.getParameter("emailAddress")).andReturn(emailAddress).once();
}
mockResponse.setStatus(responseStatus);
expectLastCall().once();
replay(mockRequest, mockResponse);
Subject.doAs(subject, new PrivilegedExceptionAction<Void>()
{
@Override
public Void run() throws Exception
{
testSubject.doGet(mockRequest, mockResponse);
return null;
}
});
verify(mockRequest, mockResponse);
}
@Test
Alinga Yeung
committed
public void testGetWithNullSubject() throws Exception
Alinga Yeung
committed
{
final Subject subject = null;
testSubjectAndEmailAddress(subject, EMAIL_ADDRESS, HttpServletResponse.SC_UNAUTHORIZED);
Alinga Yeung
committed
}
@Test
Alinga Yeung
committed
public void testGetWithEmptySubject() throws Exception
Alinga Yeung
committed
{
final Subject subject = new Subject();;
testSubjectAndEmailAddress(subject, EMAIL_ADDRESS, HttpServletResponse.SC_UNAUTHORIZED);
Alinga Yeung
committed
}
public void testPrivilegedSubjectAndEmailAddress(final List<Subject> privSubjects,
Alinga Yeung
committed
final Subject subject, int responseStatus,
final UserPersistence<Principal> mockUserPersistence) throws Exception
Alinga Yeung
committed
{
@SuppressWarnings("serial")
final ResetPasswordServlet testSubject = new ResetPasswordServlet()
{
@Override
Alinga Yeung
committed
public void init() throws ServletException
Alinga Yeung
committed
{
privilegedSubjects = privSubjects;
Alinga Yeung
committed
userPersistence = mockUserPersistence;
}
Alinga Yeung
committed
@Override
protected boolean isPrivilegedSubject(final HttpServletRequest request)
{
if (privSubjects == null || privSubjects.isEmpty())
Alinga Yeung
committed
{
return false;
}
else
{
return true;
}
}
Alinga Yeung
committed
@Override
Subject getSubject(final HttpServletRequest request)
{
return subject;
}
};
final HttpServletRequest mockRequest =
createMock(HttpServletRequest.class);
final HttpServletResponse mockResponse =
createMock(HttpServletResponse.class);
expect(mockRequest.getPathInfo()).andReturn("users/CADCtest").once();
expect(mockRequest.getMethod()).andReturn("POST").once();
expect(mockRequest.getRemoteAddr()).andReturn("mysite.com").once();
Alinga Yeung
committed
if (privSubjects != null && !privSubjects.isEmpty())
Alinga Yeung
committed
{
if (mockUserPersistence == null)
{
expect(mockRequest.getParameter("emailAddress")).andReturn("").once();
}
else
{
expect(mockRequest.getParameter("emailAddress")).andReturn(EMAIL_ADDRESS).once();
Alinga Yeung
committed
}
}
mockResponse.setStatus(responseStatus);
Alinga Yeung
committed
expectLastCall().once();
Alinga Yeung
committed
Alinga Yeung
committed
replay(mockRequest, mockResponse);
Alinga Yeung
committed
Subject.doAs(subject, new PrivilegedExceptionAction<Void>()
{
@Override
public Void run() throws Exception
Alinga Yeung
committed
{
testSubject.init();
Alinga Yeung
committed
testSubject.doGet(mockRequest, mockResponse);
return null;
}
});
Alinga Yeung
committed
Alinga Yeung
committed
verify(mockRequest, mockResponse);
}
Alinga Yeung
committed
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
@Test
public void testGetWithNullPrivilegedSubjects() throws Exception
{
final Subject subject = new Subject();;
subject.getPrincipals().add(new HttpPrincipal("CADCtest"));
testPrivilegedSubjectAndEmailAddress(null, subject,
HttpServletResponse.SC_FORBIDDEN, null);
}
@Test
public void testGetWithEmptyPrivilegedSubjects() throws Exception
{
final Subject subject = new Subject();;
subject.getPrincipals().add(new HttpPrincipal("CADCtest"));
testPrivilegedSubjectAndEmailAddress(new ArrayList<Subject>(), subject,
HttpServletResponse.SC_FORBIDDEN, null);
}
@Test
public void testGetWithMissingEmailAddress() throws Exception
{
final Subject subject = new Subject();;
subject.getPrincipals().add(new HttpPrincipal("CADCtest"));
List<Subject> privilegedSubjects = new ArrayList<Subject>();
privilegedSubjects.add(new Subject());
testPrivilegedSubjectAndEmailAddress(privilegedSubjects, subject,
HttpServletResponse.SC_BAD_REQUEST, null);
}
@SuppressWarnings("unchecked")
@Test
public void testGetWithMoreThanOneUserFound() throws Exception
{
final Subject subject = new Subject();;
subject.getPrincipals().add(new HttpPrincipal("CADCtest"));
List<Subject> privilegedSubjects = new ArrayList<Subject>();
privilegedSubjects.add(new Subject());
UserNotFoundException unfe = new UserNotFoundException(LdapUserDAO.EMAIL_ADDRESS_CONFLICT_MESSAGE);
Alinga Yeung
committed
UserPersistence<Principal> mockUserPersistence =
(UserPersistence<Principal>) createMock(UserPersistence.class);
expect(mockUserPersistence.getUserByEmailAddress(EMAIL_ADDRESS)).andThrow(unfe);
Alinga Yeung
committed
testPrivilegedSubjectAndEmailAddress(privilegedSubjects, subject,
HttpServletResponse.SC_CONFLICT, mockUserPersistence);
}
@SuppressWarnings("unchecked")
@Test
public void testGetWithNoUserFound() throws Exception
{
final Subject subject = new Subject();;
subject.getPrincipals().add(new HttpPrincipal("CADCtest"));
List<Subject> privilegedSubjects = new ArrayList<Subject>();
privilegedSubjects.add(new Subject());
UserNotFoundException unfe = new UserNotFoundException("User with email address ");
Alinga Yeung
committed
UserPersistence<Principal> mockUserPersistence =
(UserPersistence<Principal>) createMock(UserPersistence.class);
expect(mockUserPersistence.getUserByEmailAddress(EMAIL_ADDRESS)).andThrow(unfe);
Alinga Yeung
committed
testPrivilegedSubjectAndEmailAddress(privilegedSubjects, subject,
HttpServletResponse.SC_NOT_FOUND, mockUserPersistence);
}
@SuppressWarnings("unchecked")
@Test
public void testGetWithInternalServerError() throws Exception
{
final Subject subject = new Subject();;
subject.getPrincipals().add(new HttpPrincipal("CADCtest"));
List<Subject> privilegedSubjects = new ArrayList<Subject>();
privilegedSubjects.add(new Subject());
RuntimeException rte = new RuntimeException();
Alinga Yeung
committed
UserPersistence<Principal> mockUserPersistence =
(UserPersistence<Principal>) createMock(UserPersistence.class);
expect(mockUserPersistence.getUserByEmailAddress(EMAIL_ADDRESS)).andThrow(rte);
Alinga Yeung
committed
testPrivilegedSubjectAndEmailAddress(privilegedSubjects, subject,
HttpServletResponse.SC_INTERNAL_SERVER_ERROR, mockUserPersistence);
}