From a680e921e3e5f866a5f91a05883bf547309e36ee Mon Sep 17 00:00:00 2001
From: Derek Ho <dxho@amazon.com>
Date: Wed, 5 Jun 2024 10:07:25 -0400
Subject: [PATCH] Add short format ipv6 test

Signed-off-by: Derek Ho <dxho@amazon.com>
---
 .../opensearch/security/auth/UserInjectorTest.java    | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/src/test/java/org/opensearch/security/auth/UserInjectorTest.java b/src/test/java/org/opensearch/security/auth/UserInjectorTest.java
index 2aae2edf37..6338ef68a7 100644
--- 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<>();