upd: enforce that test cases receive all mails explicitly

[gigi.git] / tests / club / wpia / gigi / pages / main / KeyCompromiseTestMessage.java

diff --git a/tests/club/wpia/gigi/pages/main/KeyCompromiseTestMessage.java b/tests/club/wpia/gigi/pages/main/KeyCompromiseTestMessage.java
index 06a5ba5107c6418e7ab40799b2e59ce2c27fcbc2..7dfb4155ffbbdef0fd1683644b94576914c5bac2 100644 (file)
--- a/tests/club/wpia/gigi/pages/main/KeyCompromiseTestMessage.java
+++ b/tests/club/wpia/gigi/pages/main/KeyCompromiseTestMessage.java
@@ -92,7 +92,7 @@ public class KeyCompromiseTestMessage extends ClientTest {
         String data = IOUtils.readURL(rc);
         assertThat(data, hasError());
         assertThat(data, CoreMatchers.containsString(HTMLEncoder.encodeHTML("message may not contain '---'")));
-        assertNull(getMailReceiver().poll());
+        assertNull(getMailReceiver().poll(null));
         assertEquals(CertificateStatus.ISSUED, cert.getStatus());
     }
 
@@ -108,7 +108,7 @@ public class KeyCompromiseTestMessage extends ClientTest {
     private TestMail reportCompromiseAndCheck(String params) throws IOException, UnsupportedEncodingException, CertificateEncodingException, GeneralSecurityException {
         HttpURLConnection huc = reportCompromise(params);
         assertThat(IOUtils.readURL(huc), hasNoError());
-        TestMail rc = getMailReceiver().receive();
+        TestMail rc = getMailReceiver().receive(email);
         assertEquals(u.getEmail(), rc.getTo());
         assertThat(rc.getMessage(), CoreMatchers.containsString(cert.getSerial()));
         assertEquals(CertificateStatus.REVOKED, cert.getStatus());