Add blackhole support to MockMailer

This will ignore all sent mail to avoid storing it in memory where that is not required.

1 files changed, 16 insertions, 0 deletions

diff --git a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/stubs/MockMailer.java b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/stubs/MockMailer.java
index 6050f238da7..9e6653e70ff 100644
--- a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/stubs/MockMailer.java
+++ b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/stubs/MockMailer.java
@@ -12,9 +12,25 @@ import java.util.Map;
 public class MockMailer implements Mailer {
 
     public final Map<String, List<Mail>> mails = new HashMap<>();
+    public final boolean blackhole;
+
+    public MockMailer() {
+        this(false);
+    }
+
+    MockMailer(boolean blackhole) {
+        this.blackhole = blackhole;
+    }
+
+    public static MockMailer blackhole() {
+        return new MockMailer(true);
+    }
 
     @Override
     public void send(Mail mail) {
+        if (blackhole) {
+            return;
+        }
         for (String recipient : mail.recipients()) {
             mails.putIfAbsent(recipient, new ArrayList<>());
             mails.get(recipient).add(mail);