1515
1616package com .amazon .opendistroforelasticsearch .commons .authuser ;
1717
18- import static com .amazon .opendistroforelasticsearch .commons .ConfigConstants .OPENDISTRO_SECURITY_USER_AND_ROLES ;
18+ import static com .amazon .opendistroforelasticsearch .commons .ConfigConstants .OPENDISTRO_SECURITY_USER_INFO_THREAD_CONTEXT ;
1919import static org .junit .Assert .assertEquals ;
2020import static org .junit .Assert .assertFalse ;
2121import static org .junit .Assert .assertNull ;
@@ -122,8 +122,8 @@ public void testStreamConstForTenantUser() throws IOException {
122122 @ Test
123123 public void testParseUserString () {
124124 ThreadContext tc = new ThreadContext (Settings .EMPTY );
125- tc .putTransient ("user_roles_string" , "myuser|bckrole1,bckrol2|role1,role2|myTenant" );
126- String str = tc .getTransient ("user_roles_string" );
125+ tc .putTransient (OPENDISTRO_SECURITY_USER_INFO_THREAD_CONTEXT , "myuser|bckrole1,bckrol2|role1,role2|myTenant" );
126+ String str = tc .getTransient (OPENDISTRO_SECURITY_USER_INFO_THREAD_CONTEXT );
127127 User user = User .parse (str );
128128
129129 assertEquals ("myuser" , user .getName ());
@@ -137,16 +137,16 @@ public void testParseUserString() {
137137 @ Test
138138 public void testParseUserStringEmpty () {
139139 ThreadContext tc = new ThreadContext (Settings .EMPTY );
140- String str = tc .getTransient (OPENDISTRO_SECURITY_USER_AND_ROLES );
140+ String str = tc .getTransient (OPENDISTRO_SECURITY_USER_INFO_THREAD_CONTEXT );
141141 User user = User .parse (str );
142142 assertEquals (null , user );
143143 }
144144
145145 @ Test
146146 public void testParseUserStringName () {
147147 ThreadContext tc = new ThreadContext (Settings .EMPTY );
148- tc .putTransient (OPENDISTRO_SECURITY_USER_AND_ROLES , "myuser||" );
149- String str = tc .getTransient (OPENDISTRO_SECURITY_USER_AND_ROLES );
148+ tc .putTransient (OPENDISTRO_SECURITY_USER_INFO_THREAD_CONTEXT , "myuser||" );
149+ String str = tc .getTransient (OPENDISTRO_SECURITY_USER_INFO_THREAD_CONTEXT );
150150 User user = User .parse (str );
151151
152152 assertEquals ("myuser" , user .getName ());
@@ -157,8 +157,8 @@ public void testParseUserStringName() {
157157 @ Test
158158 public void testParseUserStringNameWithTenant () {
159159 ThreadContext tc = new ThreadContext (Settings .EMPTY );
160- tc .putTransient (OPENDISTRO_SECURITY_USER_AND_ROLES , "myuser|||myTenant" );
161- String str = tc .getTransient (OPENDISTRO_SECURITY_USER_AND_ROLES );
160+ tc .putTransient (OPENDISTRO_SECURITY_USER_INFO_THREAD_CONTEXT , "myuser|||myTenant" );
161+ String str = tc .getTransient (OPENDISTRO_SECURITY_USER_INFO_THREAD_CONTEXT );
162162 User user = User .parse (str );
163163
164164 assertEquals ("myuser" , user .getName ());
@@ -170,8 +170,8 @@ public void testParseUserStringNameWithTenant() {
170170 @ Test
171171 public void testParseUserStringNobackendRoles () {
172172 ThreadContext tc = new ThreadContext (Settings .EMPTY );
173- tc .putTransient (OPENDISTRO_SECURITY_USER_AND_ROLES , "myuser||role1,role2" );
174- String str = tc .getTransient (OPENDISTRO_SECURITY_USER_AND_ROLES );
173+ tc .putTransient (OPENDISTRO_SECURITY_USER_INFO_THREAD_CONTEXT , "myuser||role1,role2" );
174+ String str = tc .getTransient (OPENDISTRO_SECURITY_USER_INFO_THREAD_CONTEXT );
175175 User user = User .parse (str );
176176
177177 assertEquals ("myuser" , user .getName ());
@@ -182,8 +182,8 @@ public void testParseUserStringNobackendRoles() {
182182 @ Test
183183 public void testParseUserStringNoRoles () {
184184 ThreadContext tc = new ThreadContext (Settings .EMPTY );
185- tc .putTransient (OPENDISTRO_SECURITY_USER_AND_ROLES , "myuser|brole1,brole2|" );
186- String str = tc .getTransient (OPENDISTRO_SECURITY_USER_AND_ROLES );
185+ tc .putTransient (OPENDISTRO_SECURITY_USER_INFO_THREAD_CONTEXT , "myuser|brole1,brole2|" );
186+ String str = tc .getTransient (OPENDISTRO_SECURITY_USER_INFO_THREAD_CONTEXT );
187187 User user = User .parse (str );
188188
189189 assertEquals ("myuser" , user .getName ());
@@ -194,8 +194,8 @@ public void testParseUserStringNoRoles() {
194194 @ Test
195195 public void testParseUserStringNoRolesWithTenant () {
196196 ThreadContext tc = new ThreadContext (Settings .EMPTY );
197- tc .putTransient (OPENDISTRO_SECURITY_USER_AND_ROLES , "myuser|brole1,brole2||myTenant" );
198- String str = tc .getTransient (OPENDISTRO_SECURITY_USER_AND_ROLES );
197+ tc .putTransient (OPENDISTRO_SECURITY_USER_INFO_THREAD_CONTEXT , "myuser|brole1,brole2||myTenant" );
198+ String str = tc .getTransient (OPENDISTRO_SECURITY_USER_INFO_THREAD_CONTEXT );
199199 User user = User .parse (str );
200200
201201 assertEquals ("myuser" , user .getName ());
@@ -207,8 +207,8 @@ public void testParseUserStringNoRolesWithTenant() {
207207 @ Test
208208 public void testParseUserStringMalformed () {
209209 ThreadContext tc = new ThreadContext (Settings .EMPTY );
210- tc .putTransient (OPENDISTRO_SECURITY_USER_AND_ROLES , "|backendrole1,backendrole2|role1,role2" );
211- String str = tc .getTransient (OPENDISTRO_SECURITY_USER_AND_ROLES );
210+ tc .putTransient (OPENDISTRO_SECURITY_USER_INFO_THREAD_CONTEXT , "|backendrole1,backendrole2|role1,role2" );
211+ String str = tc .getTransient (OPENDISTRO_SECURITY_USER_INFO_THREAD_CONTEXT );
212212 User user = User .parse (str );
213213 assertEquals (null , user );
214214 }
0 commit comments