Refactor "Email checker" (for testing purposes)

[gigi.git] / src / org / cacert / gigi / email / TestEmailProvider.java

diff --git a/src/org/cacert/gigi/email/TestEmailProvider.java b/src/org/cacert/gigi/email/TestEmailProvider.java
index e37e802a079e9f69f22f0aaa5c9ea191ffb6208f..9f37be97f3966bcb7091c1fa328eae5446de830f 100644 (file)
--- a/src/org/cacert/gigi/email/TestEmailProvider.java
+++ b/src/org/cacert/gigi/email/TestEmailProvider.java
@@ -1,5 +1,6 @@
 package org.cacert.gigi.email;
 
+import java.io.DataInputStream;
 import java.io.DataOutputStream;
 import java.io.IOException;
 import java.net.InetAddress;
@@ -11,6 +12,7 @@ class TestEmailProvider extends EmailProvider {
        ServerSocket servs;
        Socket client;
        DataOutputStream out;
+       DataInputStream in;
        protected TestEmailProvider(Properties props) {
                try {
                        servs = new ServerSocket(Integer.parseInt(props
@@ -29,8 +31,10 @@ class TestEmailProvider extends EmailProvider {
                        if (client == null || client.isClosed()) {
                                client = servs.accept();
                                out = new DataOutputStream(client.getOutputStream());
+                               in = new DataInputStream(client.getInputStream());
                        }
                        try {
+                               out.writeUTF("mail");
                                write(to);
                                write(subject);
                                write(message);
@@ -43,6 +47,14 @@ class TestEmailProvider extends EmailProvider {
                        }
                }
        }
+       @Override
+       public String checkEmailServer(int forUid, String address)
+                       throws IOException {
+               out.writeUTF("challengeAddrBox");
+               out.writeUTF(address);
+               return in.readUTF();
+       }
+
        private void write(String to) throws IOException {
                if (to == null) {
                        out.writeUTF("<null>");