diff options
Diffstat (limited to 'container-core/src/test/java/com/yahoo/container/di/ConfigRetrieverTest.java')
| -rw-r--r-- | container-core/src/test/java/com/yahoo/container/di/ConfigRetrieverTest.java | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/container-core/src/test/java/com/yahoo/container/di/ConfigRetrieverTest.java b/container-core/src/test/java/com/yahoo/container/di/ConfigRetrieverTest.java index 99ea6a4ee58..bba5675e458 100644 --- a/container-core/src/test/java/com/yahoo/container/di/ConfigRetrieverTest.java +++ b/container-core/src/test/java/com/yahoo/container/di/ConfigRetrieverTest.java @@ -19,6 +19,7 @@ import org.junit.rules.ExpectedException; import java.util.Collections; import java.util.HashSet; +import java.util.Optional; import java.util.Set; import static org.hamcrest.CoreMatchers.instanceOf; @@ -107,11 +108,11 @@ public class ConfigRetrieverTest { private ConfigRetriever createConfigRetriever() { String configId = dirConfigSource.configId(); - CloudSubscriberFactory subscriberFactory = new CloudSubscriberFactory(dirConfigSource.configSource()); + CloudSubscriberFactory subscriber = new CloudSubscriberFactory(dirConfigSource.configSource()); Set<ConfigKey<? extends ConfigInstance>> keys = new HashSet<>(); keys.add(new ConfigKey<>(Bootstrap1Config.class, configId)); keys.add(new ConfigKey<>(Bootstrap2Config.class, configId)); - return new ConfigRetriever(keys, subscriberFactory); + return new ConfigRetriever(keys, keySet -> subscriber.getSubscriber(keySet)); } private void writeConfig(String name, String contents) { |