ADD: output pinglog, test DNS and email ping.

[gigi.git] / tests / org / cacert / gigi / ping / TestDNS.java

diff --git a/tests/org/cacert/gigi/ping/TestDNS.java b/tests/org/cacert/gigi/ping/TestDNS.java
new file mode 100644 (file)
index 0000000..0a28491
--- /dev/null
+++ b/tests/org/cacert/gigi/ping/TestDNS.java
@@ -0,0 +1,112 @@
+package org.cacert.gigi.ping;
+
+import static org.junit.Assert.*;
+
+import java.io.IOException;
+import java.net.HttpURLConnection;
+import java.net.MalformedURLException;
+import java.net.URL;
+import java.net.URLConnection;
+import java.net.URLEncoder;
+import java.sql.PreparedStatement;
+import java.sql.ResultSet;
+import java.sql.SQLException;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+import org.cacert.gigi.database.DatabaseConnection;
+import org.cacert.gigi.pages.account.DomainOverview;
+import org.cacert.gigi.testUtils.IOUtils;
+import org.cacert.gigi.testUtils.ManagedTest;
+import org.cacert.gigi.testUtils.TestEmailReciever.TestMail;
+import org.cacert.gigi.util.RandomToken;
+import org.junit.Test;
+
+public class TestDNS extends ManagedTest {
+
+    @Test
+    public void testDNSSanity() throws IOException {
+
+        String token = RandomToken.generateToken(16);
+        String value = RandomToken.generateToken(16);
+
+        Process p = updateDNS(token, value);
+        String reRead = new String(IOUtils.readURL(p.getInputStream()));
+        reRead = reRead.trim();
+        reRead = reRead.substring(1, reRead.length() - 1);
+        assertEquals(value, reRead);
+
+    }
+
+    @Test
+    public void testEmailAndDNS() throws IOException, InterruptedException, SQLException {
+        String email = createUniqueName() + "@example.org";
+        int uid = createVerifiedUser("a", "b", email, TEST_PASSWORD);
+        String cookie = login(email, TEST_PASSWORD);
+
+        String test = getTestProps().getProperty("domain.dnstest");
+        URL u = new URL("https://" + getServerName() + DomainOverview.PATH);
+        URLConnection openConnection = u.openConnection();
+        openConnection.setRequestProperty("Cookie", cookie);
+        String content1 = IOUtils.readURL(openConnection);
+        String csrf = getCSRF(1, content1);
+
+        Pattern p = Pattern.compile("cacert-([A-Za-z0-9]+) IN TXT ([A-Za-z0-9]+)");
+        Matcher m = p.matcher(content1);
+        m.find();
+        updateDNS(m.group(1), m.group(2));
+
+        String content = "newdomain=" + URLEncoder.encode(test, "UTF-8") + //
+                "&emailType=y&email=2&DNSType=y" + //
+                "&ssl-type-0=direct&ssl-port-0=" + //
+                "&ssl-type-1=direct&ssl-port-1=" + //
+                "&ssl-type-2=direct&ssl-port-2=" + //
+                "&ssl-type-3=direct&ssl-port-3=" + //
+                "&adddomain&csrf=" + csrf;
+        openConnection = u.openConnection();
+        openConnection.setRequestProperty("Cookie", cookie);
+        openConnection.setDoOutput(true);
+        openConnection.getOutputStream().write(content.getBytes());
+        openConnection.getHeaderField("Location");
+
+        String newcontent = IOUtils.readURL(cookie(u.openConnection(), cookie));
+        Pattern dlink = Pattern.compile(DomainOverview.PATH + "([0-9]+)'>");
+        Matcher m1 = dlink.matcher(newcontent);
+        m1.find();
+        URL u2 = new URL(u.toString() + m1.group(1));
+
+        TestMail mail = getMailReciever().recieve();
+        String link = mail.extractLink();
+        new URL(link).openConnection().getHeaderField("");
+
+        PreparedStatement ps = DatabaseConnection.getInstance().prepare("SELECT COUNT(*) FROM domainPinglog");
+        while (true) {
+            ResultSet rs = ps.executeQuery();
+            rs.next();
+            if (rs.getInt(1) >= 2) {
+                break;
+            }
+            Thread.sleep(200);
+        }
+
+        newcontent = IOUtils.readURL(cookie(u2.openConnection(), cookie));
+        Pattern pat = Pattern.compile("<td>dns</td>\\s*<td>success</td>");
+        assertTrue(newcontent, pat.matcher(newcontent).find());
+        pat = Pattern.compile("<td>email</td>\\s*<td>success</td>");
+        assertTrue(newcontent, pat.matcher(newcontent).find());
+    }
+
+    private Process updateDNS(String token, String value) throws IOException, MalformedURLException {
+        String test = getTestProps().getProperty("domain.dnstest");
+        String targetDomain = "cacert-" + token + "." + test;
+        String manage = getTestProps().getProperty("domain.dnsmanage");
+        String url = manage + "t1=" + token + "&t2=" + value;
+        assertEquals(200, ((HttpURLConnection) new URL(url).openConnection()).getResponseCode());
+
+        Process p = Runtime.getRuntime().exec(new String[] {
+                "dig", "@" + getTestProps().getProperty("domain.testns"), "+short", "TXT", targetDomain
+        });
+        return p;
+    }
+
+}