1 package org.cacert.gigi.pages.wot;
3 import static org.hamcrest.CoreMatchers.*;
4 import static org.junit.Assert.*;
6 import java.io.IOException;
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;
13 import java.sql.SQLException;
14 import java.text.SimpleDateFormat;
15 import java.util.Date;
16 import java.util.regex.Pattern;
18 import org.cacert.gigi.pages.account.MyDetails;
19 import org.cacert.gigi.testUtils.IOUtils;
20 import org.cacert.gigi.testUtils.ManagedTest;
21 import org.junit.Before;
22 import org.junit.Test;
24 public class TestAssurance extends ManagedTest {
26 private String assurerM;
28 private String assureeM;
30 private String cookie;
33 public void setup() throws IOException {
34 assurerM = createUniqueName() + "@cacert-test.org";
35 assureeM = createUniqueName() + "@cacert-test.org";
37 createAssuranceUser("a", "b", assurerM, TEST_PASSWORD);
38 createVerifiedUser("a", "c", assureeM, TEST_PASSWORD);
40 cookie = login(assurerM, TEST_PASSWORD);
44 public void testAssureSearch() throws IOException {
45 String loc = search("email=" + URLEncoder.encode(assureeM, "UTF-8") + "&day=1&month=1&year=1910");
46 assertTrue(loc, loc.contains("type=\"checkbox\" name=\"CCAAgreed\""));
50 public void testAssureSearchEmail() throws IOException {
51 String loc = search("email=1" + URLEncoder.encode(assureeM, "UTF-8") + "&day=1&month=1&year=1910");
52 assertTrue(loc, !loc.contains("type=\"checkbox\" name=\"CCAAgreed\""));
56 public void testAssureSearchDob() throws IOException {
57 String loc = search("email=" + URLEncoder.encode(assureeM, "UTF-8") + "&day=2&month=1&year=1910");
58 assertTrue(loc, !loc.contains("type=\"checkbox\" name=\"CCAAgreed\""));
59 loc = search("email=" + URLEncoder.encode(assureeM, "UTF-8") + "&day=1&month=2&year=1910");
60 assertTrue(loc, !loc.contains("type=\"checkbox\" name=\"CCAAgreed\""));
61 loc = search("email=" + URLEncoder.encode(assureeM, "UTF-8") + "&day=1&month=1&year=1911");
62 assertTrue(loc, !loc.contains("type=\"checkbox\" name=\"CCAAgreed\""));
65 private String search(String query) throws MalformedURLException, IOException, UnsupportedEncodingException {
66 URL u = new URL("https://" + getServerName() + AssurePage.PATH);
67 URLConnection uc = u.openConnection();
69 uc.addRequestProperty("Cookie", cookie);
70 uc.getOutputStream().write(("search&" + query).getBytes("UTF-8"));
71 uc.getOutputStream().flush();
73 return IOUtils.readURL(uc);
77 public void testAssureForm() throws IOException {
78 String error = getError("date=2000-01-01&location=testcase&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10");
83 public void testAssureFormContanisData() throws IOException {
84 URLConnection uc = buildupAssureFormConnection(true);
85 uc.getOutputStream().write(("date=2000-01-01&location=testcase&rules=1&CCAAgreed=1&assertion=1&points=10").getBytes("UTF-8"));
86 uc.getOutputStream().flush();
87 String data = IOUtils.readURL(uc);
88 assertThat(data, containsString("2000-01-01"));
89 assertThat(data, containsString("testcase"));
93 public void testAssureFormNoCSRF() throws IOException {
95 HttpURLConnection uc = (HttpURLConnection) buildupAssureFormConnection(false);
96 uc.getOutputStream().write(("date=2000-01-01&location=testcase&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10").getBytes("UTF-8"));
97 uc.getOutputStream().flush();
98 assertEquals(500, uc.getResponseCode());
102 public void testAssureFormWrongCSRF() throws IOException {
104 HttpURLConnection uc = (HttpURLConnection) buildupAssureFormConnection(false);
105 uc.getOutputStream().write(("date=2000-01-01&location=testcase&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10&csrf=aragc").getBytes("UTF-8"));
106 uc.getOutputStream().flush();
107 assertEquals(500, uc.getResponseCode());
111 public void testAssureFormRaceName() throws IOException, SQLException {
112 testAssureFormRace(true);
116 public void testAssureFormRaceDoB() throws IOException, SQLException {
117 testAssureFormRace(false);
120 public void testAssureFormRace(boolean name) throws IOException, SQLException {
121 URLConnection uc = buildupAssureFormConnection(true);
123 String assureeCookie = login(assureeM, TEST_PASSWORD);
124 String newName = "lname=" + (name ? "c" : "a") + "&fname=a&mname=&suffix=";
125 String newDob = "day=1&month=1&year=" + (name ? 1910 : 1911);
127 assertNull(executeBasicWebInteraction(assureeCookie, MyDetails.PATH, newName + "&" + newDob + "&processDetails", 0));
129 uc.getOutputStream().write(("date=2000-01-01&location=testcase&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10").getBytes("UTF-8"));
130 uc.getOutputStream().flush();
131 String error = fetchStartErrorMessage(IOUtils.readURL(uc));
132 assertTrue(error, !error.startsWith("</div>"));
136 public void testAssureFormFuture() throws IOException {
137 SimpleDateFormat sdf = new SimpleDateFormat("yyyy");
138 int year = Integer.parseInt(sdf.format(new Date(System.currentTimeMillis()))) + 2;
139 String error = getError("date=" + year + "-01-01&location=testcase&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10");
140 assertTrue(error, !error.startsWith("</div>"));
144 public void testAssureFormNoLoc() throws IOException {
145 String error = getError("date=2000-01-01&location=a&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10");
146 assertTrue(error, !error.startsWith("</div>"));
147 error = getError("date=2000-01-01&location=&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10");
148 assertTrue(error, !error.startsWith("</div>"));
152 public void testAssureFormInvalDate() throws IOException {
153 String error = getError("date=20000101&location=testcase&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10");
154 assertTrue(error, !error.startsWith("</div>"));
155 error = getError("date=&location=testcase&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10");
156 assertTrue(error, !error.startsWith("</div>"));
160 public void testAssureFormBoxes() throws IOException {
161 String error = getError("date=2000-01-01&location=testcase&certify=0&rules=1&CCAAgreed=1&assertion=1&points=10");
162 assertTrue(error, !error.startsWith("</div>"));
163 error = getError("date=2000-01-01&location=testcase&certify=1&rules=&CCAAgreed=1&assertion=1&points=10");
164 assertTrue(error, !error.startsWith("</div>"));
165 error = getError("date=2000-01-01&location=testcase&certify=1&rules=1&CCAAgreed=a&assertion=1&points=10");
166 assertTrue(error, !error.startsWith("</div>"));
167 error = getError("date=2000-01-01&location=testcase&certify=1&rules=1&CCAAgreed=1&assertion=z&points=10");
168 assertTrue(error, !error.startsWith("</div>"));
172 public void testAssureListingValid() throws IOException {
173 String uniqueLoc = createUniqueName();
174 String error = getError("date=2000-01-01&location=" + uniqueLoc + "&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10");
176 String cookie = login(assureeM, TEST_PASSWORD);
177 URLConnection url = new URL("https://" + getServerName() + MyPoints.PATH).openConnection();
178 url.setRequestProperty("Cookie", cookie);
179 String resp = IOUtils.readURL(url);
180 resp = resp.split(Pattern.quote("</table>"))[0];
181 assertThat(resp, containsString(uniqueLoc));
185 public void testAssurerListingValid() throws IOException {
186 String uniqueLoc = createUniqueName();
187 String error = getError("date=2000-01-01&location=" + uniqueLoc + "&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10");
189 String cookie = login(assurerM, TEST_PASSWORD);
190 URLConnection url = new URL("https://" + getServerName() + MyPoints.PATH).openConnection();
191 url.setRequestProperty("Cookie", cookie);
192 String resp = IOUtils.readURL(url);
193 resp = resp.split(Pattern.quote("</table>"))[1];
194 assertThat(resp, containsString(uniqueLoc));
197 private String getError(String query) throws MalformedURLException, IOException {
198 URLConnection uc = buildupAssureFormConnection(true);
199 uc.getOutputStream().write((query).getBytes("UTF-8"));
200 uc.getOutputStream().flush();
201 String error = fetchStartErrorMessage(IOUtils.readURL(uc));
205 private URLConnection buildupAssureFormConnection(boolean doCSRF) throws MalformedURLException, IOException {
206 URL u = new URL("https://" + getServerName() + AssurePage.PATH);
207 URLConnection uc = u.openConnection();
208 uc.addRequestProperty("Cookie", cookie);
209 uc.setDoOutput(true);
210 uc.getOutputStream().write(("email=" + URLEncoder.encode(assureeM, "UTF-8") + "&day=1&month=1&year=1910&search").getBytes("UTF-8"));
212 String csrf = getCSRF(uc);
213 uc = u.openConnection();
214 uc.addRequestProperty("Cookie", cookie);
215 uc.setDoOutput(true);
217 uc.getOutputStream().write(("csrf=" + csrf + "&").getBytes("UTF-8"));