1 package org.cacert.gigi.pages.account;
3 import static org.junit.Assert.*;
5 import java.io.IOException;
6 import java.io.OutputStream;
7 import java.io.UnsupportedEncodingException;
8 import java.net.HttpURLConnection;
9 import java.net.MalformedURLException;
11 import java.net.URLConnection;
12 import java.net.URLEncoder;
14 import org.cacert.gigi.dbObjects.User;
15 import org.cacert.gigi.pages.PasswordResetPage;
16 import org.cacert.gigi.pages.wot.TestAssurance;
17 import org.cacert.gigi.testUtils.ClientTest;
18 import org.cacert.gigi.testUtils.IOUtils;
19 import org.cacert.gigi.testUtils.TestEmailReceiver.TestMail;
20 import org.cacert.gigi.util.RandomToken;
21 import org.junit.Test;
23 public class TestPasswordResetExternal extends ClientTest {
26 public void testByAssurance() throws IOException {
27 User u = User.getById(createAssuranceUser("fn", "ln", createUniqueName() + "@example.com", TEST_PASSWORD));
28 String cookie2 = login(u.getEmail(), TEST_PASSWORD);
29 URLConnection uc = TestAssurance.buildupAssureFormConnection(cookie2, email, true);
30 String avalue = RandomToken.generateToken(32);
31 uc.getOutputStream().write(("date=1910-01-01&location=testcase&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10&passwordReset=1&passwordResetValue=" + URLEncoder.encode(avalue, "UTF-8")).getBytes("UTF-8"));
32 uc.getOutputStream().flush();
33 String error = fetchStartErrorMessage(IOUtils.readURL(uc));
36 TestMail mail = getMailReciever().receive();
37 String link = mail.extractLink();
38 String npw = TEST_PASSWORD + "'";
39 System.out.println(link);
40 assertNotNull(toPasswordReset(avalue, link, npw, npw + "'"));
41 assertNotNull(toPasswordReset(avalue + "'", link, npw, npw));
42 assertNotNull(toPasswordReset(avalue, link, "a", "a"));
43 assertNull(toPasswordReset(avalue, link, npw, npw));
44 assertNotNull(login(email, npw));
47 private String toPasswordReset(String avalue, String link, String npw, String npw2) throws IOException, MalformedURLException, UnsupportedEncodingException {
48 URLConnection uc2 = new URL(link).openConnection();
49 String csrf = getCSRF(uc2);
50 String headerField = uc2.getHeaderField("Set-Cookie");
51 assertNotNull(headerField);
52 String cookie3 = stripCookie(headerField);
53 uc2 = new URL("https://" + getServerName() + PasswordResetPage.PATH).openConnection();
55 uc2.setDoOutput(true);
56 OutputStream o = uc2.getOutputStream();
57 o.write(("csrf=" + csrf + "&pword1=" + URLEncoder.encode(npw, "UTF-8") + "&pword2=" + URLEncoder.encode(npw2, "UTF-8") + "&private_token=" + URLEncoder.encode(avalue, "UTF-8")).getBytes("UTF-8"));
58 System.out.println(((HttpURLConnection) uc2).getResponseCode());
59 String readURL = IOUtils.readURL(uc2);
60 return fetchStartErrorMessage(readURL);