Read active token fingerprints from configserver

1 files changed, 10 insertions, 0 deletions

diff --git a/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ConfigServerMock.java b/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ConfigServerMock.java
index ed5226ebc8b..4a1ad86b5a4 100644
--- a/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ConfigServerMock.java
+++ b/controller-server/src/test/java/com/yahoo/vespa/hosted/controller/integration/ConfigServerMock.java
@@ -103,6 +103,7 @@ public class ConfigServerMock extends AbstractComponent implements ConfigServer
     private final Map<DeploymentId, TestReport> testReport = new HashMap<>();
     private final Map<DeploymentId, CloudAccount> cloudAccounts = new HashMap<>();
     private final Map<DeploymentId, List<X509Certificate>> additionalCertificates = new HashMap<>();
+    private final Map<String, List<String>> activeTokenFingerprints = new HashMap<>();
     private List<SearchNodeMetrics> searchNodeMetrics;
 
     private Version lastPrepareVersion = null;
@@ -319,6 +320,10 @@ public class ConfigServerMock extends AbstractComponent implements ConfigServer
         return additionalCertificates.getOrDefault(deployment, List.of());
     }
 
+    public void setActiveTokenFingerprints(String hostname, List<String> fingerprints) {
+        activeTokenFingerprints.put(hostname, List.copyOf(fingerprints));
+    }
+
     @Override
     public NodeRepositoryMock nodeRepository() {
         return nodeRepository;
@@ -585,6 +590,11 @@ public class ConfigServerMock extends AbstractComponent implements ConfigServer
         return "{\"settings\":{\"name\":\"foo\",\"role\":\"vespa-secretstore-access\",\"awsId\":\"892075328880\",\"externalId\":\"*****\",\"region\":\"us-east-1\"},\"status\":\"ok\"}";
     }
 
+    @Override
+    public Map<String, List<String>> activeTokenFingerprints(DeploymentId deploymentId) {
+        return Map.copyOf(activeTokenFingerprints);
+    }
+
     public static class Application {
 
         private final ApplicationId id;