mock ThreadPool in RestSendToExtensionActionTests

Signed-off-by: Craig Perkins <[email protected]>
cwperks · Aug 16, 2024 · 1536380 · 1536380
1 parent efac919
commit 1536380
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/server/src/test/java/org/opensearch/extensions/rest/RestSendToExtensionActionTests.java b/server/src/test/java/org/opensearch/extensions/rest/RestSendToExtensionActionTests.java
@@ -110,7 +110,6 @@ public void setup() throws Exception {
         );
         SettingsModule settingsModule = new SettingsModule(settings);
         UsageService usageService = new UsageService();
-        TestThreadPool threadPool = new TestThreadPool(getTestName());
         actionModule = new ActionModule(
             settingsModule.getSettings(),
             new IndexNameExpressionResolver(new ThreadContext(Settings.EMPTY)),
@@ -123,10 +122,10 @@ public void setup() throws Exception {
             null,
             usageService,
             null,
-            new IdentityService(Settings.EMPTY, threadPool, new ArrayList<>()),
-            new ExtensionsManager(Set.of(), new IdentityService(Settings.EMPTY, threadPool, List.of()))
+            new IdentityService(Settings.EMPTY, mock(ThreadPool.class), new ArrayList<>()),
+            new ExtensionsManager(Set.of(), new IdentityService(Settings.EMPTY, mock(ThreadPool.class), List.of()))
         );
-        identityService = new IdentityService(Settings.EMPTY, threadPool, new ArrayList<>());
+        identityService = new IdentityService(Settings.EMPTY, mock(ThreadPool.class), new ArrayList<>());
         dynamicActionRegistry = actionModule.getDynamicActionRegistry();
     }