Add withListFlag() to InMemoryFlagSource

1 files changed, 5 insertions, 0 deletions

diff --git a/flags/src/main/java/com/yahoo/vespa/flags/InMemoryFlagSource.java b/flags/src/main/java/com/yahoo/vespa/flags/InMemoryFlagSource.java
index 5ed1859b89b..b6e716fc912 100644
--- a/flags/src/main/java/com/yahoo/vespa/flags/InMemoryFlagSource.java
+++ b/flags/src/main/java/com/yahoo/vespa/flags/InMemoryFlagSource.java
@@ -1,6 +1,7 @@
 // Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
 package com.yahoo.vespa.flags;
 
+import java.util.List;
 import java.util.Map;
 import java.util.Optional;
 import java.util.concurrent.ConcurrentHashMap;
@@ -40,6 +41,10 @@ public class InMemoryFlagSource implements FlagSource {
         return withRawFlag(flagId, new UnboundJacksonFlag<>(flagId, value, jacksonClass).serializer().serialize(value));
     }
 
+    public <T> InMemoryFlagSource withListFlag(FlagId flagId, List<T> value) {
+        return withRawFlag(flagId, new UnboundListFlag<T>(flagId, value).serializer().serialize(value));
+    }
+
     public InMemoryFlagSource removeFlag(FlagId flagId) {
         rawFlagsById.remove(flagId);
         return this;