Merge pull request #1844 from Netflix/add_more_sessioncontext_methods

Add Key<T> specialization to SessionContext.containsKey

zuul-core/src/main/java/com/netflix/zuul/context/SessionContext.java

@@ -151,6 +151,23 @@ public <T> T getOrDefault(Key<T> key, T defaultValue) {
         return defaultValue;
     }
 
+    /**
+     * {@inheritDoc}
+     *
+     * <p>This method exists for static analysis.
+     */
+    @Override
+    public boolean containsKey(Object key) {
+        return super.containsKey(key);
+    }
+
+    /**
+     * Checks for the existence of the key in the context.
+     */
+    public <T> boolean containsKey(Key<T> key) {
+        return typedMap.containsKey(Objects.requireNonNull(key, "key"));
+    }
+
     /**
      * {@inheritDoc}
      *

zuul-core/src/test/java/com/netflix/zuul/context/SessionContextTest.java

@@ -15,16 +15,14 @@
  */
 package com.netflix.zuul.context;
 
-import static org.junit.jupiter.api.Assertions.assertEquals;
-import static org.junit.jupiter.api.Assertions.assertNull;
-import static org.junit.jupiter.api.Assertions.assertThrows;
-
 import com.google.common.truth.Truth;
 import com.netflix.zuul.context.SessionContext.Key;
 import org.junit.jupiter.api.Test;
 import org.junit.jupiter.api.extension.ExtendWith;
 import org.mockito.junit.jupiter.MockitoExtension;
 
+import static org.junit.jupiter.api.Assertions.*;
+
 @ExtendWith(MockitoExtension.class)
 class SessionContextTest {
 
@@ -107,4 +105,18 @@ void remove() {
         Truth.assertThat(context.get(key)).isNull();
         Truth.assertThat(val).isEqualTo("bar");
     }
+
+    @Test
+    void containsKey() {
+        SessionContext context = new SessionContext();
+        Key<String> key = SessionContext.newKey("foo");
+        context.put(key, "bar");
+
+        Truth.assertThat(context.containsKey(key)).isTrue();
+
+        String val = context.remove(key);
+        Truth.assertThat(val).isEqualTo("bar");
+
+        Truth.assertThat(context.containsKey(key)).isFalse();
+    }
 }