Create test method to create assurer (stub) + add id return.

[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 c9a44b8a158046c778ac6291f53faee948e078d9..25ae8bb6ed503aa02a77a82ab8f463618fc73183 100644 (file)
--- a/tests/org/cacert/gigi/testUtils/ManagedTest.java
+++ b/tests/org/cacert/gigi/testUtils/ManagedTest.java
@@ -15,12 +15,17 @@ 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;
 import org.junit.After;
 import org.junit.AfterClass;
 import org.junit.BeforeClass;
@@ -45,6 +50,16 @@ public class ManagedTest {
        public static void connectToServer() {
                try {
                        testProps.load(new FileInputStream("config/test.properties"));
+                       if (!DatabaseConnection.isInited()) {
+                               DatabaseConnection.init(testProps);
+                       }
+                       System.out.println("... purging Database");
+                       DatabaseManager.run(new String[]{
+                                       testProps.getProperty("sql.driver"),
+                                       testProps.getProperty("sql.url"),
+                                       testProps.getProperty("sql.user"),
+                                       testProps.getProperty("sql.password")});
+
                        String type = testProps.getProperty("type");
                        if (type.equals("local")) {
                                url = testProps.getProperty("server");
@@ -59,12 +74,18 @@ public class ManagedTest {
                                        gigi.getOutputStream());
                        System.out.println("... starting server");
                        Properties mainProps = new Properties();
-                       mainProps.load(new FileInputStream("config/gigi.properties"));
                        mainProps.setProperty("host", "127.0.0.1");
                        mainProps.setProperty("port", testProps.getProperty("serverPort"));
                        mainProps.setProperty("emailProvider",
                                        "org.cacert.gigi.email.TestEmailProvider");
                        mainProps.setProperty("emailProvider.port", "8473");
+                       mainProps.setProperty("sql.driver",
+                                       testProps.getProperty("sql.driver"));
+                       mainProps.setProperty("sql.url", testProps.getProperty("sql.url"));
+                       mainProps
+                                       .setProperty("sql.user", testProps.getProperty("sql.user"));
+                       mainProps.setProperty("sql.password",
+                                       testProps.getProperty("sql.password"));
 
                        byte[] cacerts = Files
                                        .readAllBytes(Paths.get("config/cacerts.jks"));
@@ -108,6 +129,10 @@ public class ManagedTest {
                                        new InetSocketAddress("localhost", 8473));
                } catch (IOException e) {
                        throw new Error(e);
+               } catch (ClassNotFoundException e1) {
+                       e1.printStackTrace();
+               } catch (SQLException e1) {
+                       e1.printStackTrace();
                }
 
        }
@@ -170,7 +195,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 {
@@ -180,10 +205,30 @@ 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);
+               // TODO make him pass CATS and be assured for 100 points
+               return uid;
+       }
+       static int count = 0;
+       public String createUniqueName() {
+               return "test" + System.currentTimeMillis() + "a" + (count++);
+       }
 }