Update test for port and address

Signed-off-by: Derek Ho <[email protected]>
derek-ho · Jun 4, 2024 · 7120a8d · 7120a8d
1 parent 584be35
commit 7120a8d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/src/main/java/org/opensearch/security/auth/UserInjector.java b/src/main/java/org/opensearch/security/auth/UserInjector.java
@@ -67,7 +67,7 @@ public UserInjector(Settings settings, ThreadPool threadPool, AuditLog auditLog,
     }
 
     public static class InjectedUser extends User {
-        private transient TransportAddress transportAddress;
+        transient TransportAddress transportAddress;
 
         public InjectedUser(String name) {
             super(name);

diff --git a/src/test/java/org/opensearch/security/auth/UserInjectorTest.java b/src/test/java/org/opensearch/security/auth/UserInjectorTest.java
@@ -72,9 +72,10 @@ public void testValidInjectUserIpV6() {
             ConfigConstants.OPENDISTRO_SECURITY_INJECTED_USER,
             "user|role1,role2|2001:db8:3333:4444:5555:6666:7777:8888:9200"
         );
-        User injectedUser = userInjector.getInjectedUser();
-        assertEquals(injectedUser.getName(), "user");
-        assertEquals(injectedUser.getRoles(), roles);
+        UserInjector.InjectedUser injectedUser = userInjector.getInjectedUser();
+        assertEquals("user", injectedUser.getName());
+        assertEquals(9200, injectedUser.getTransportAddress().getPort());
+        assertEquals("2001:db8:3333:4444:5555:6666:7777:8888", injectedUser.getTransportAddress().getAddress());
     }
 
     @Test
@@ -97,9 +98,11 @@ public void testValidInjectUserBracketsIpV6() {
             ConfigConstants.OPENDISTRO_SECURITY_INJECTED_USER,
             "user|role1,role2|[2001:db8:3333:4444:5555:6666:7777:8888]:9200"
         );
-        User injectedUser = userInjector.getInjectedUser();
-        assertEquals(injectedUser.getName(), "user");
-        assertEquals(injectedUser.getRoles(), roles);
+        UserInjector.InjectedUser injectedUser = userInjector.getInjectedUser();
+        assertEquals("user", injectedUser.getName());
+        assertEquals(roles, injectedUser.getRoles());
+        assertEquals(9200, injectedUser.getTransportAddress().getPort());
+        assertEquals("2001:db8:3333:4444:5555:6666:7777:8888", injectedUser.getTransportAddress().getAddress());
     }
 
     @Test