Extend test to mbox verifications

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

diff --git a/tests/org/cacert/gigi/testUtils/TestEmailReciever.java b/tests/org/cacert/gigi/testUtils/TestEmailReciever.java
index 53e461625eea2ccfd9783e32925f9a00731cd7ee..446d2aad769d59fe5efd55b71f3ebde148eaa23a 100644 (file)
--- a/tests/org/cacert/gigi/testUtils/TestEmailReciever.java
+++ b/tests/org/cacert/gigi/testUtils/TestEmailReciever.java
@@ -1,11 +1,14 @@
 package org.cacert.gigi.testUtils;
 
 import java.io.DataInputStream;
+import java.io.DataOutputStream;
 import java.io.IOException;
 import java.net.SocketAddress;
 import java.net.Socket;
 import java.util.concurrent.LinkedBlockingQueue;
 import java.util.concurrent.TimeUnit;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
 
 public class TestEmailReciever implements Runnable {
        public class TestMail {
@@ -37,10 +40,17 @@ public class TestEmailReciever implements Runnable {
                public String getReplyto() {
                        return replyto;
                }
+               public String extractLink() {
+                       Pattern link = Pattern.compile("http://[^\\s]+(?=\\s)");
+                       Matcher m = link.matcher(getMessage());
+                       m.find();
+                       return m.group(0);
+               }
 
        }
        private Socket s;
        private DataInputStream dis;
+       private DataOutputStream dos;
 
        public TestEmailReciever(SocketAddress target) throws IOException {
                s = new Socket();
@@ -48,6 +58,7 @@ public class TestEmailReciever implements Runnable {
                s.setKeepAlive(true);
                s.setSoTimeout(1000 * 60 * 60);
                dis = new DataInputStream(s.getInputStream());
+               dos = new DataOutputStream(s.getOutputStream());
                new Thread(this).start();
        }
        LinkedBlockingQueue<TestMail> mails = new LinkedBlockingQueue<TestEmailReciever.TestMail>();
@@ -59,21 +70,44 @@ public class TestEmailReciever implements Runnable {
        public void run() {
                try {
                        while (true) {
-                               String to = dis.readUTF();
-                               String subject = dis.readUTF();
-                               String message = dis.readUTF();
-                               String from = dis.readUTF();
-                               String replyto = dis.readUTF();
-                               mails.add(new TestMail(to, subject, message, from, replyto));
+                               String type = dis.readUTF();
+                               if (type.equals("mail")) {
+                                       String to = dis.readUTF();
+                                       String subject = dis.readUTF();
+                                       String message = dis.readUTF();
+                                       String from = dis.readUTF();
+                                       String replyto = dis.readUTF();
+                                       mails.add(new TestMail(to, subject, message, from, replyto));
+                               } else if (type.equals("challengeAddrBox")) {
+                                       String email = dis.readUTF();
+                                       if (approveRegex.matcher(email).matches()) {
+                                               System.out.println("approving mbox: " + email);
+                                               dos.writeUTF("OK");
+                                       } else {
+                                               System.out.println("rejecting mbox: " + email);
+                                               dos.writeUTF("FAIL");
+                                       }
+                                       dos.flush();
+                               } else {
+                                       System.err.println("Unknown type: " + type);
+                               }
                        }
                } catch (IOException e) {
                        e.printStackTrace();
                }
 
        }
+       Pattern approveRegex = Pattern.compile(".*");
+       public void setApproveRegex(Pattern approveRegex) {
+               this.approveRegex = approveRegex;
+       }
 
        public void clearMails() {
                mails.clear();
        }
+       public void reset() {
+               clearMails();
+               approveRegex = Pattern.compile(".*");
+       }
 
 }