Fix shiro test

Signed-off-by: Craig Perkins <[email protected]>
cwperks · Jan 7, 2025 · be4b7a5 · be4b7a5
1 parent 2765e88
commit be4b7a5
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/.../identity-shiro/src/test/java/org/opensearch/identity/shiro/ShiroIdentityPluginTests.java b/.../identity-shiro/src/test/java/org/opensearch/identity/shiro/ShiroIdentityPluginTests.java
@@ -9,11 +9,11 @@
 package org.opensearch.identity.shiro;
 
 import org.opensearch.OpenSearchException;
+import org.opensearch.client.Client;
 import org.opensearch.common.settings.Settings;
 import org.opensearch.identity.IdentityService;
 import org.opensearch.plugins.IdentityPlugin;
 import org.opensearch.test.OpenSearchTestCase;
-import org.opensearch.threadpool.ThreadPool;
 
 import java.util.List;
 
@@ -28,7 +28,7 @@ public class ShiroIdentityPluginTests extends OpenSearchTestCase {
     public void testSingleIdentityPluginSucceeds() {
         IdentityPlugin identityPlugin1 = new ShiroIdentityPlugin(Settings.EMPTY);
         List<IdentityPlugin> pluginList1 = List.of(identityPlugin1);
-        IdentityService identityService1 = new IdentityService(Settings.EMPTY, mock(ThreadPool.class), pluginList1);
+        IdentityService identityService1 = new IdentityService(Settings.EMPTY, mock(Client.class), pluginList1);
         assertThat(identityService1.getTokenManager(), is(instanceOf(ShiroTokenManager.class)));
     }
 
@@ -37,10 +37,7 @@ public void testMultipleIdentityPluginsFail() {
         IdentityPlugin identityPlugin2 = new ShiroIdentityPlugin(Settings.EMPTY);
         IdentityPlugin identityPlugin3 = new ShiroIdentityPlugin(Settings.EMPTY);
         List<IdentityPlugin> pluginList = List.of(identityPlugin1, identityPlugin2, identityPlugin3);
-        Exception ex = assertThrows(
-            OpenSearchException.class,
-            () -> new IdentityService(Settings.EMPTY, mock(ThreadPool.class), pluginList)
-        );
+        Exception ex = assertThrows(OpenSearchException.class, () -> new IdentityService(Settings.EMPTY, mock(Client.class), pluginList));
         assert (ex.getMessage().contains("Multiple identity plugins are not supported,"));
     }