1 package org.cacert.gigi.api;
3 import static org.junit.Assert.*;
5 import java.io.IOException;
6 import java.io.UnsupportedEncodingException;
7 import java.net.HttpURLConnection;
8 import java.net.MalformedURLException;
9 import java.net.URLEncoder;
10 import java.security.GeneralSecurityException;
11 import java.security.KeyManagementException;
12 import java.security.NoSuchAlgorithmException;
14 import org.cacert.gigi.GigiApiException;
15 import org.cacert.gigi.dbObjects.CATS.CATSType;
16 import org.cacert.gigi.dbObjects.Certificate;
17 import org.cacert.gigi.dbObjects.Certificate.CSRType;
18 import org.cacert.gigi.dbObjects.Certificate.SANType;
19 import org.cacert.gigi.dbObjects.CertificateProfile;
20 import org.cacert.gigi.dbObjects.Digest;
21 import org.cacert.gigi.dbObjects.User;
22 import org.cacert.gigi.testUtils.IOUtils;
23 import org.cacert.gigi.testUtils.RestrictedApiTest;
24 import org.junit.Test;
26 public class ImportCATSResult extends RestrictedApiTest {
29 public void testLookupSerial() throws GigiApiException, IOException, GeneralSecurityException, InterruptedException {
30 Certificate target2 = new Certificate(u, u, Certificate.buildDN("EMAIL", u.getEmail()), Digest.SHA256, generatePEMCSR(generateKeypair(), "EMAIL=" + u.getEmail()), CSRType.CSR, CertificateProfile.getByName("client"), new Certificate.SubjectAlternateName(SANType.EMAIL, "cats@cacert.org"));
31 await(target2.issue(null, "2y", u));
32 target2.setLoginEnabled(true);
34 assertEquals(u.getId(), Integer.parseInt(apiLookup(target2)));
38 public void testImportCATS() throws GigiApiException, IOException, GeneralSecurityException, InterruptedException {
40 assertEquals(1, u.getTrainings().length);
41 apiImport(u, "Test Training");
42 assertEquals(2, u.getTrainings().length);
44 User u2 = User.getById(createVerifiedUser("fn", "ln", createUniqueName() + "@example.com", TEST_PASSWORD));
45 assertEquals(0, u2.getTrainings().length);
46 assertFalse(u2.hasPassedCATS());
47 apiImport(u2, "Test Training");
48 assertEquals(1, u2.getTrainings().length);
49 assertFalse(u2.hasPassedCATS());
50 apiImport(u2, CATSType.ASSURER_CHALLENGE.getDisplayName());
51 assertEquals(2, u2.getTrainings().length);
52 assertTrue(u2.hasPassedCATS());
57 public void testImportCATSFailures() throws GigiApiException, IOException, GeneralSecurityException, InterruptedException {
58 assertEquals(1, u.getTrainings().length);
59 assertNotEquals(200, executeImportQuery("").getResponseCode());
60 assertNotEquals(200, executeImportQuery("mid=" + u.getId()).getResponseCode());
61 assertNotEquals(200, executeImportQuery("mid=" + u.getId() + "&variant=Test+Training").getResponseCode());
62 assertNotEquals(200, executeImportQuery("mid=" + u.getId() + "&variant=Test+Training&date=" + System.currentTimeMillis()).getResponseCode());
63 assertNotEquals(200, executeImportQuery("mid=" + u.getId() + "&variant=Test+Training&date=" + System.currentTimeMillis() + "&language=en").getResponseCode());
64 assertNotEquals(200, executeImportQuery("mid=" + u.getId() + "&variant=Test+Training&date=" + System.currentTimeMillis() + "&version=1.0").getResponseCode());
65 assertEquals(1, u.getTrainings().length);
66 apiImport(u, "Test Training");
67 assertEquals(2, u.getTrainings().length);
71 private void apiImport(User target, String test) throws IOException, MalformedURLException, NoSuchAlgorithmException, KeyManagementException, UnsupportedEncodingException, GeneralSecurityException {
72 HttpURLConnection connection = executeImportQuery("mid=" + target.getId() + "&variant=" + URLEncoder.encode(test, "UTF-8") + "&date=" + System.currentTimeMillis() + "&language=en&version=1.0");
73 if (connection.getResponseCode() != 200) {
74 throw new Error(connection.getResponseMessage());
78 private HttpURLConnection executeImportQuery(String query) throws IOException, GeneralSecurityException {
79 return doApi(CATSImport.PATH, query);
82 private String apiLookup(Certificate target) throws IOException, GeneralSecurityException {
83 HttpURLConnection connection = doApi(CATSResolve.PATH, "serial=" + target.cert().getSerialNumber().toString(16).toLowerCase());
84 if (connection.getResponseCode() != 200) {
85 throw new Error(connection.getResponseMessage());
87 return IOUtils.readURL(connection);