Add short format ipv6 test

Signed-off-by: Derek Ho <dxho@amazon.com>
derek-ho · Jun 5, 2024 · a680e92 · a680e92
1 parent 9454232
commit a680e92
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/src/test/java/org/opensearch/security/auth/UserInjectorTest.java b/src/test/java/org/opensearch/security/auth/UserInjectorTest.java
@@ -78,6 +78,17 @@ public void testValidInjectUserIpV6() {
         assertEquals("2001:db8:3333:4444:5555:6666:7777:8888", injectedUser.getTransportAddress().getAddress());
     }
 
+    @Test
+    public void testValidInjectUserIpV6ShortFormat() {
+        HashSet<String> roles = new HashSet<>();
+        roles.addAll(Arrays.asList("role1", "role2"));
+        threadContext.putTransient(ConfigConstants.OPENDISTRO_SECURITY_INJECTED_USER, "user|role1,role2|2001:db8::1:9200");
+        UserInjector.InjectedUser injectedUser = userInjector.getInjectedUser();
+        assertEquals("user", injectedUser.getName());
+        assertEquals(9200, injectedUser.getTransportAddress().getPort());
+        assertEquals("2001:db8::1", injectedUser.getTransportAddress().getAddress());
+    }
+
     @Test
     public void testInvalidInjectUserIpV6() {
         HashSet<String> roles = new HashSet<>();