Fix: consolidate TestEmailReveiver interface

[gigi.git] / tests / org / cacert / gigi / pages / main / RegisterPageTest.java

diff --git a/tests/org/cacert/gigi/pages/main/RegisterPageTest.java b/tests/org/cacert/gigi/pages/main/RegisterPageTest.java
index c6b2a67b4520c66da6c81ab1c99ee4c38fc42427..695598ae7d0225d03473570bb65acd3a1d58c034 100644 (file)
--- a/tests/org/cacert/gigi/pages/main/RegisterPageTest.java
+++ b/tests/org/cacert/gigi/pages/main/RegisterPageTest.java
@@ -10,8 +10,6 @@ import java.util.regex.Pattern;
 
 import org.cacert.gigi.testUtils.InitTruststore;
 import org.cacert.gigi.testUtils.ManagedTest;
-import org.cacert.gigi.testUtils.TestEmailReciever.TestMail;
-import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
 
@@ -44,14 +42,12 @@ public class RegisterPageTest extends ManagedTest {
         data = fetchStartErrorMessage(runRegister(query));
         assertNotNull(data);
 
-        assertNull(getMailReciever().recieve());
+        assertNull(getMailReciever().poll());
 
     }
 
     private void assertSuccessfullRegMail() {
-        TestMail tm = waitForMail();
-        Assert.assertNotNull(tm);
-        String link = tm.extractLink();
+        String link = getMailReciever().receive().extractLink();
         assertTrue(link, link.startsWith("https://"));
     }