tests/org/cacert/gigi/pages/wot/TestAssurance.java

   1 package org.cacert.gigi.pages.wot;
   2
   3 import static org.junit.Assert.*;
   4
   5 import java.io.IOException;
   6 import java.io.UnsupportedEncodingException;
   7 import java.net.HttpURLConnection;
   8 import java.net.MalformedURLException;
   9 import java.net.URL;
  10 import java.net.URLConnection;
  11 import java.net.URLEncoder;
  12 import java.sql.PreparedStatement;
  13 import java.sql.SQLException;
  14 import java.text.SimpleDateFormat;
  15 import java.util.Date;
  16 import java.util.regex.Pattern;
  17
  18 import org.cacert.gigi.database.DatabaseConnection;
  19 import org.cacert.gigi.testUtils.IOUtils;
  20 import org.cacert.gigi.testUtils.ManagedTest;
  21 import org.junit.Before;
  22 import org.junit.Test;
  23
  24 public class TestAssurance extends ManagedTest {
  25
  26     private String assurerM;
  27
  28     private String assureeM;
  29
  30     private int assurer;
  31
  32     private int assuree;
  33
  34     private String cookie;
  35
  36     @Before
  37     public void setup() throws IOException {
  38         assurerM = createUniqueName() + "@cacert-test.org";
  39         assureeM = createUniqueName() + "@cacert-test.org";
  40         assurer = createAssuranceUser("a", "b", assurerM, TEST_PASSWORD);
  41         assuree = createAssuranceUser("a", "c", assureeM, TEST_PASSWORD);
  42         cookie = login(assurerM, TEST_PASSWORD);
  43
  44     }
  45
  46     @Test
  47     public void testAssureSearch() throws IOException {
  48         String loc = search("email=" + URLEncoder.encode(assureeM, "UTF-8") + "&day=1&month=1&year=1910");
  49         assertTrue(loc, loc.contains("type=\"checkbox\" name=\"CCAAgreed\""));
  50     }
  51
  52     @Test
  53     public void testAssureSearchEmail() throws IOException {
  54         String loc = search("email=1" + URLEncoder.encode(assureeM, "UTF-8") + "&day=1&month=1&year=1910");
  55         assertTrue(loc, !loc.contains("type=\"checkbox\" name=\"CCAAgreed\""));
  56     }
  57
  58     @Test
  59     public void testAssureSearchDob() throws IOException {
  60         String loc = search("email=" + URLEncoder.encode(assureeM, "UTF-8") + "&day=2&month=1&year=1910");
  61         assertTrue(loc, !loc.contains("type=\"checkbox\" name=\"CCAAgreed\""));
  62         loc = search("email=" + URLEncoder.encode(assureeM, "UTF-8") + "&day=1&month=2&year=1910");
  63         assertTrue(loc, !loc.contains("type=\"checkbox\" name=\"CCAAgreed\""));
  64         loc = search("email=" + URLEncoder.encode(assureeM, "UTF-8") + "&day=1&month=1&year=1911");
  65         assertTrue(loc, !loc.contains("type=\"checkbox\" name=\"CCAAgreed\""));
  66     }
  67
  68     private String search(String query) throws MalformedURLException, IOException, UnsupportedEncodingException {
  69         URL u = new URL("https://" + getServerName() + AssurePage.PATH);
  70         URLConnection uc = u.openConnection();
  71         uc.setDoOutput(true);
  72         uc.addRequestProperty("Cookie", cookie);
  73         uc.getOutputStream().write(("search&" + query).getBytes());
  74         uc.getOutputStream().flush();
  75
  76         return IOUtils.readURL(uc);
  77     }
  78
  79     @Test
  80     public void testAssureForm() throws IOException {
  81         String error = getError("date=2000-01-01&location=testcase&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10");
  82         assertNull(error);
  83     }
  84
  85     @Test
  86     public void testAssureFormNoCSRF() throws IOException {
  87         // override csrf
  88         HttpURLConnection uc = (HttpURLConnection) buildupAssureFormConnection(false);
  89         uc.getOutputStream().write(("date=2000-01-01&location=testcase&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10").getBytes());
  90         uc.getOutputStream().flush();
  91         assertEquals(500, uc.getResponseCode());
  92     }
  93
  94     @Test
  95     public void testAssureFormWrongCSRF() throws IOException {
  96         // override csrf
  97         HttpURLConnection uc = (HttpURLConnection) buildupAssureFormConnection(false);
  98         uc.getOutputStream().write(("date=2000-01-01&location=testcase&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10&csrf=aragc").getBytes());
  99         uc.getOutputStream().flush();
 100         assertEquals(500, uc.getResponseCode());
 101     }
 102
 103     @Test
 104     public void testAssureFormRace() throws IOException, SQLException {
 105         URLConnection uc = buildupAssureFormConnection(true);
 106         PreparedStatement ps = DatabaseConnection.getInstance().prepare("UPDATE `users` SET email='changed' WHERE id=?");
 107         ps.setInt(1, assuree);
 108         ps.execute();
 109         uc.getOutputStream().write(("date=2000-01-01&location=testcase&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10").getBytes());
 110         uc.getOutputStream().flush();
 111         String error = fetchStartErrorMessage(IOUtils.readURL(uc));
 112         assertTrue(error, !error.startsWith("</div>"));
 113     }
 114
 115     @Test
 116     public void testAssureFormFuture() throws IOException {
 117         SimpleDateFormat sdf = new SimpleDateFormat("YYYY");
 118         int year = Integer.parseInt(sdf.format(new Date(System.currentTimeMillis()))) + 2;
 119         String error = getError("date=" + year + "-01-01&location=testcase&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10");
 120         assertTrue(error, !error.startsWith("</div>"));
 121     }
 122
 123     @Test
 124     public void testAssureFormNoLoc() throws IOException {
 125         String error = getError("date=2000-01-01&location=a&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10");
 126         assertTrue(error, !error.startsWith("</div>"));
 127         error = getError("date=2000-01-01&location=&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10");
 128         assertTrue(error, !error.startsWith("</div>"));
 129     }
 130
 131     @Test
 132     public void testAssureFormInvalDate() throws IOException {
 133         String error = getError("date=20000101&location=testcase&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10");
 134         assertTrue(error, !error.startsWith("</div>"));
 135         error = getError("date=&location=testcase&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10");
 136         assertTrue(error, !error.startsWith("</div>"));
 137     }
 138
 139     @Test
 140     public void testAssureFormBoxes() throws IOException {
 141         String error = getError("date=2000-01-01&location=testcase&certify=0&rules=1&CCAAgreed=1&assertion=1&points=10");
 142         assertTrue(error, !error.startsWith("</div>"));
 143         error = getError("date=2000-01-01&location=testcase&certify=1&rules=&CCAAgreed=1&assertion=1&points=10");
 144         assertTrue(error, !error.startsWith("</div>"));
 145         error = getError("date=2000-01-01&location=testcase&certify=1&rules=1&CCAAgreed=a&assertion=1&points=10");
 146         assertTrue(error, !error.startsWith("</div>"));
 147         error = getError("date=2000-01-01&location=testcase&certify=1&rules=1&CCAAgreed=1&assertion=z&points=10");
 148         assertTrue(error, !error.startsWith("</div>"));
 149     }
 150
 151     @Test
 152     public void testAssureListingValid() throws IOException {
 153         String uniqueLoc = createUniqueName();
 154         String error = getError("date=2000-01-01&location=" + uniqueLoc + "&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10");
 155         assertNull(error);
 156         String cookie = login(assureeM, TEST_PASSWORD);
 157         URLConnection url = new URL("https://" + getServerName() + MyPoints.PATH).openConnection();
 158         url.setRequestProperty("Cookie", cookie);
 159         String resp = IOUtils.readURL(url);
 160         resp = resp.split(Pattern.quote("</table>"))[0];
 161         assertTrue(resp.contains(uniqueLoc));
 162     }
 163
 164     @Test
 165     public void testAssurerListingValid() throws IOException {
 166         String uniqueLoc = createUniqueName();
 167         String error = getError("date=2000-01-01&location=" + uniqueLoc + "&certify=1&rules=1&CCAAgreed=1&assertion=1&points=10");
 168         assertNull(error);
 169         String cookie = login(assurerM, TEST_PASSWORD);
 170         URLConnection url = new URL("https://" + getServerName() + MyPoints.PATH).openConnection();
 171         url.setRequestProperty("Cookie", cookie);
 172         String resp = IOUtils.readURL(url);
 173         resp = resp.split(Pattern.quote("</table>"))[1];
 174         assertTrue(resp.contains(uniqueLoc));
 175     }
 176
 177     private String getError(String query) throws MalformedURLException, IOException {
 178         URLConnection uc = buildupAssureFormConnection(true);
 179         uc.getOutputStream().write((query).getBytes());
 180         uc.getOutputStream().flush();
 181         String error = fetchStartErrorMessage(IOUtils.readURL(uc));
 182         return error;
 183     }
 184
 185     private URLConnection buildupAssureFormConnection(boolean doCSRF) throws MalformedURLException, IOException {
 186         URL u = new URL("https://" + getServerName() + AssurePage.PATH);
 187         URLConnection uc = u.openConnection();
 188         uc.addRequestProperty("Cookie", cookie);
 189         uc.setDoOutput(true);
 190         uc.getOutputStream().write(("email=" + URLEncoder.encode(assureeM, "UTF-8") + "&day=1&month=1&year=1910&search").getBytes());
 191
 192         String csrf = getCSRF(uc);
 193         uc = u.openConnection();
 194         uc.addRequestProperty("Cookie", cookie);
 195         uc.setDoOutput(true);
 196         if (doCSRF) {
 197             uc.getOutputStream().write(("csrf=" + csrf + "&").getBytes());
 198         }
 199         return uc;
 200     }
 201
 202 }