Add withRoles

1 files changed, 4 insertions, 0 deletions

diff --git a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/role/SecurityContext.java b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/role/SecurityContext.java
index 85f42eecc58..f681e84a020 100644
--- a/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/role/SecurityContext.java
+++ b/controller-api/src/main/java/com/yahoo/vespa/hosted/controller/api/role/SecurityContext.java
@@ -48,6 +48,10 @@ public class SecurityContext {
     /** @return credential expiration or {@link Instant#MAX} is not available */
     public Instant expiresAt() { return expiresAt; }
 
+    public SecurityContext withRoles(Set<Role> roles) {
+        return new SecurityContext(this.principal, roles, this.issuedAt, this.expiresAt);
+    }
+
     @Override
     public boolean equals(Object o) {
         if (this == o) return true;