X-Git-Url: https://code.wpia.club/?a=blobdiff_plain;f=tests%2Forg%2Fcacert%2Fgigi%2FtestUtils%2FManagedTest.java;h=3d6457275d90af32c23b38d05d5be21caa041cda;hb=e155c45f66749b1260fddc4a652ecc9cf40a11a7;hp=f591e77259e0f6a1a6d9fd3cf576f0f92469fa45;hpb=a95fcf745db63e39d16aa3ec34a4d4f00b9b60d5;p=gigi.git

diff --git a/tests/org/cacert/gigi/testUtils/ManagedTest.java b/tests/org/cacert/gigi/testUtils/ManagedTest.java
index f591e772..3d645727 100644
--- 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,11 +16,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;
@@ -35,10 +42,25 @@ public class ManagedTest {
 		return url;
 	}
 	static Properties testProps = new Properties();
+	static {
+		InitTruststore.run();
+		HttpURLConnection.setFollowRedirects(false);
+	}
+
 	@BeforeClass
 	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");
@@ -53,12 +75,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"));
@@ -102,6 +130,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();
 		}
 
 	}
@@ -137,8 +169,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);
@@ -156,7 +187,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);
@@ -164,19 +195,69 @@ 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 {
 			TestMail tm = ter.recieve();
 			String verifyLink = tm.extractLink();
 			String[] parts = verifyLink.split("\\?");
-			URL u = new URL("https://" + getServerName() + "/verify" + parts[1]);
+			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;
 	}
 }