tests/club/wpia/gigi/pages/TestMain.java

   1 package club.wpia.gigi.pages;
   2
   3 import static org.hamcrest.CoreMatchers.*;
   4 import static org.junit.Assert.*;
   5
   6 import java.io.IOException;
   7 import java.net.HttpURLConnection;
   8 import java.net.MalformedURLException;
   9 import java.net.URL;
  10 import java.net.URLConnection;
  11 import java.security.GeneralSecurityException;
  12 import java.security.KeyPair;
  13 import java.security.PrivateKey;
  14 import java.security.cert.X509Certificate;
  15
  16 import org.junit.Test;
  17
  18 import club.wpia.gigi.GigiApiException;
  19 import club.wpia.gigi.dbObjects.Certificate;
  20 import club.wpia.gigi.dbObjects.Certificate.CSRType;
  21 import club.wpia.gigi.dbObjects.Digest;
  22 import club.wpia.gigi.testUtils.ClientTest;
  23 import club.wpia.gigi.testUtils.IOUtils;
  24
  25 public class TestMain extends ClientTest {
  26
  27     @Test
  28     public void testPasswordLogin() throws MalformedURLException, IOException {
  29         URLConnection uc = new URL("https://" + getServerName()).openConnection();
  30         uc.addRequestProperty("Cookie", cookie);
  31         String content = IOUtils.readURL(uc);
  32
  33         assertThat(content, not(containsString("via certificate")));
  34
  35         makeAgent(u.getId());
  36         uc = new URL("https://" + getServerName()).openConnection();
  37         uc.addRequestProperty("Cookie", cookie);
  38         content = IOUtils.readURL(uc);
  39         assertThat(content, containsString("For some actions, e.g. add verification, support, you need to be authenticated via certificate."));
  40
  41     }
  42
  43     @Test
  44     public void testCertLogin() throws GeneralSecurityException, IOException, GigiApiException, InterruptedException {
  45         KeyPair kp = generateKeypair();
  46         String csr = generatePEMCSR(kp, "CN=" + u.getPreferredName().toString());
  47         Certificate c = new Certificate(u, u, Certificate.buildDN("CN", u.getPreferredName().toString()), Digest.SHA256, csr, CSRType.CSR, getClientProfile());
  48         final PrivateKey pk = kp.getPrivate();
  49         await(c.issue(null, "2y", u));
  50         final X509Certificate ce = c.cert();
  51         c.setLoginEnabled(true);
  52         cookie = login(pk, ce);
  53         loginCertificate = c;
  54         loginPrivateKey = pk;
  55
  56         URLConnection uc = new URL("https://" + getSecureServerName()).openConnection();
  57         authenticate((HttpURLConnection) uc);
  58         String content = IOUtils.readURL(uc);
  59         assertThat(content, not(containsString("via certificate")));
  60
  61         makeAgent(u.getId());
  62         uc = new URL("https://" + getSecureServerName()).openConnection();
  63         authenticate((HttpURLConnection) uc);
  64         content = IOUtils.readURL(uc);
  65         assertThat(content, containsString("You are authenticated via certificate, so you will be able to perform all actions."));
  66
  67     }
  68 }