diff options
author | Jon Marius Venstad <venstad@gmail.com> | 2020-04-17 12:38:33 +0200 |
---|---|---|
committer | Jon Marius Venstad <venstad@gmail.com> | 2020-04-17 12:38:33 +0200 |
commit | 564ea2ed446154746bfb23a29aa8f1a414de3c9b (patch) | |
tree | 00b7d9fcec22654896b4fd5d261dc649bda88046 /controller-api | |
parent | 1b536d61af0be52103fdb83e0fb6f2b6947b57c7 (diff) |
Expand ContactInformationMaintainer unit test to verify exception handling
Diffstat (limited to 'controller-api')
-rw-r--r-- | controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/organization/MockContactRetriever.java | 11 |
1 files changed, 8 insertions, 3 deletions
diff --git a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/organization/MockContactRetriever.java b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/organization/MockContactRetriever.java index dd0c143ae28..ab39d8e5660 100644 --- a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/organization/MockContactRetriever.java +++ b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/integration/organization/MockContactRetriever.java @@ -8,23 +8,28 @@ import java.util.Collections; import java.util.HashMap; import java.util.Map; import java.util.Optional; +import java.util.function.Supplier; /** * @author olaa */ public class MockContactRetriever implements ContactRetriever{ - private final Map<PropertyId, Contact> contacts = new HashMap<>(); + private final Map<PropertyId, Supplier<Contact>> contacts = new HashMap<>(); @Override public Contact getContact(Optional<PropertyId> propertyId) { - return contacts.getOrDefault(propertyId.get(), contact()); + return contacts.getOrDefault(propertyId.get(), this::contact).get(); } - public void addContact(PropertyId propertyId, Contact contact) { + public void addContact(PropertyId propertyId, Supplier<Contact> contact) { contacts.put(propertyId, contact); } + public void addContact(PropertyId propertyId, Contact contact) { + contacts.put(propertyId, () -> contact); + } + public Contact contact() { return new Contact(URI.create("contacts.tld"), URI.create("properties.tld"), URI.create("issues.tld"), Collections.emptyList(), "queue", Optional.of("component")); } |