[UPDATE-CONFIG] Use 3 hosts www, secure and static.

[gigi.git] / tests / org / cacert / gigi / testUtils / ManagedTest.java

diff --git a/tests/org/cacert/gigi/testUtils/ManagedTest.java b/tests/org/cacert/gigi/testUtils/ManagedTest.java
index c85ce451aac27e5385e9d071f73485a99aba305a..869d417a47b7b932795e3bb6d32e20f075345cb0 100644 (file)
--- a/tests/org/cacert/gigi/testUtils/ManagedTest.java
+++ b/tests/org/cacert/gigi/testUtils/ManagedTest.java
@@ -8,6 +8,7 @@ import java.io.DataOutputStream;
 import java.io.FileInputStream;
 import java.io.IOException;
 import java.io.InputStreamReader;
+import java.io.OutputStream;
 import java.io.UnsupportedEncodingException;
 import java.net.HttpURLConnection;
 import java.net.InetSocketAddress;
@@ -15,12 +16,12 @@ import java.net.URL;
 import java.net.URLEncoder;
 import java.nio.file.Files;
 import java.nio.file.Paths;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
 import java.sql.SQLException;
 import java.util.Properties;
 
 import org.cacert.gigi.DevelLauncher;
-import org.cacert.gigi.IOUtils;
-import org.cacert.gigi.InitTruststore;
 import org.cacert.gigi.database.DatabaseConnection;
 import org.cacert.gigi.testUtils.TestEmailReciever.TestMail;
 import org.cacert.gigi.util.DatabaseManager;
@@ -66,13 +67,19 @@ public class ManagedTest {
                                                Integer.parseInt(parts[1])));
                                return;
                        }
-                       url = "localhost:" + testProps.getProperty("serverPort");
+                       url = testProps.getProperty("name.www") + ":"
+                                       + testProps.getProperty("serverPort");
                        gigi = Runtime.getRuntime().exec(testProps.getProperty("java"));
                        DataOutputStream toGigi = new DataOutputStream(
                                        gigi.getOutputStream());
                        System.out.println("... starting server");
                        Properties mainProps = new Properties();
                        mainProps.setProperty("host", "127.0.0.1");
+                       mainProps.setProperty("name.secure", "sec");
+                       mainProps
+                                       .setProperty("name.www", testProps.getProperty("name.www"));
+                       mainProps.setProperty("name.static", "stat");
+
                        mainProps.setProperty("port", testProps.getProperty("serverPort"));
                        mainProps.setProperty("emailProvider",
                                        "org.cacert.gigi.email.TestEmailProvider");
@@ -90,15 +97,10 @@ public class ManagedTest {
                        byte[] keystore = Files.readAllBytes(Paths
                                        .get("config/keystore.pkcs12"));
 
-                       DevelLauncher.writeGigiConfig(toGigi, new byte[]{},
+                       DevelLauncher.writeGigiConfig(toGigi, "changeit".getBytes(),
                                        "changeit".getBytes(), mainProps, cacerts, keystore);
                        toGigi.flush();
-                       // TODO wait for ready
-                       try {
-                               Thread.sleep(3000);
-                       } catch (InterruptedException e) {
-                               e.printStackTrace();
-                       }
+
                        final BufferedReader br = new BufferedReader(new InputStreamReader(
                                        gigi.getErrorStream()));
                        String line;
@@ -166,8 +168,7 @@ public class ManagedTest {
                String d = IOUtils.readURL(uc);
                return d;
        }
-       public String fetchStartErrorMessage(String query) throws IOException {
-               String d = runRegister(query);
+       public String fetchStartErrorMessage(String d) throws IOException {
                String formFail = "<div class='formError'>";
                int idx = d.indexOf(formFail);
                assertNotEquals(-1, idx);
@@ -185,7 +186,7 @@ public class ManagedTest {
                                        + "&pword1=" + URLEncoder.encode(password, "UTF-8")
                                        + "&pword2=" + URLEncoder.encode(password, "UTF-8")
                                        + "&day=1&month=1&year=1910&cca_agree=1";
-                       String data = fetchStartErrorMessage(query);
+                       String data = fetchStartErrorMessage(runRegister(query));
                        assertTrue(data, data.startsWith("</div>"));
                } catch (UnsupportedEncodingException e) {
                        throw new Error(e);
@@ -193,7 +194,7 @@ public class ManagedTest {
                        throw new Error(e);
                }
        }
-       public void createVerifiedUser(String firstName, String lastName,
+       public int createVerifiedUser(String firstName, String lastName,
                        String email, String password) {
                registerUser(firstName, lastName, email, password);
                try {
@@ -203,10 +204,59 @@ public class ManagedTest {
                        URL u = new URL("https://" + getServerName() + "/verify?"
                                        + parts[1]);
                        u.openStream().close();;
+                       PreparedStatement ps = DatabaseConnection.getInstance().prepare(
+                                       "SELECT id FROM users where email=?");
+                       ps.setString(1, email);
+                       ResultSet rs = ps.executeQuery();
+                       if (rs.next()) {
+                               return rs.getInt(1);
+                       }
+                       throw new Error();
                } catch (InterruptedException e) {
                        throw new Error(e);
                } catch (IOException e) {
                        throw new Error(e);
+               } catch (SQLException e) {
+                       throw new Error(e);
+               }
+       }
+       public int createAssuranceUser(String firstName, String lastName,
+                       String email, String password) {
+               int uid = createVerifiedUser(firstName, lastName, email, password);
+               try {
+                       PreparedStatement ps = DatabaseConnection
+                                       .getInstance()
+                                       .prepare(
+                                                       "INSERT INTO `cats_passed` SET `user_id`=?, `variant_id`=?");
+                       ps.setInt(1, uid);
+                       ps.setInt(2, 0);
+                       ps.execute();
+                       ps = DatabaseConnection.getInstance().prepare(
+                                       "INSERT INTO `notary` SET `from`=?, `to`=?, points='100'");
+                       ps.setInt(1, uid);
+                       ps.setInt(2, uid);
+                       ps.execute();
+
+               } catch (SQLException e) {
+                       throw new Error(e);
                }
+               return uid;
+       }
+       static int count = 0;
+       public String createUniqueName() {
+               return "test" + System.currentTimeMillis() + "a" + (count++);
+       }
+       public String login(String email, String pw) throws IOException {
+               URL u = new URL("https://" + getServerName() + "/login");
+               HttpURLConnection huc = (HttpURLConnection) u.openConnection();
+               huc.setDoOutput(true);
+               OutputStream os = huc.getOutputStream();
+               String data = "username=" + URLEncoder.encode(email, "UTF-8")
+                               + "&password=" + URLEncoder.encode(pw, "UTF-8");
+               os.write(data.getBytes());
+               os.flush();
+               String headerField = huc.getHeaderField("Set-Cookie");
+               headerField = headerField.substring(0, headerField.indexOf(';'));
+               return headerField;
        }
 }