1 package club.wpia.gigi.pages;
3 import static org.hamcrest.CoreMatchers.*;
4 import static org.junit.Assert.*;
6 import java.io.IOException;
7 import java.net.HttpURLConnection;
8 import java.net.MalformedURLException;
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;
16 import org.junit.Test;
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;
25 public class TestMain extends ClientTest {
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);
33 assertThat(content, not(containsString("via certificate")));
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."));
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);
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")));
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."));