fix(velocity): mock executor service maybe

xoogware · Sep 24, 2024 · f0f6c7f · f0f6c7f
1 parent fce5f50
commit f0f6c7f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mccoroutine-velocity-core/src/test/java/helper/MockedVelocityServer.kt b/mccoroutine-velocity-core/src/test/java/helper/MockedVelocityServer.kt
@@ -68,7 +68,6 @@ class MockedVelocityServer {
         }
         Mockito.`when`(server.scheduler).thenReturn(scheduler)
         Mockito.`when`(server.pluginManager).thenReturn(pluginManager)
-        Mockito.`when`(pluginManager.isLoaded(Mockito.anyString())).thenReturn(true)
 
         val commandManager = VelocityCommandManager(eventManager, pluginManager)
         Mockito.`when`(server.commandManager).thenReturn(commandManager)
@@ -79,6 +78,9 @@ class MockedVelocityServer {
         val suspendingPluginContainer = SuspendingPluginContainer(plugin, server, logger)
         suspendingPluginContainer.initialize(plugin.instance.get())
 
+        Mockito.`when`(pluginManager.isLoaded(Mockito.anyString())).thenReturn(true)
+        Mockito.`when`(pluginManager.ensurePluginContainer(Mockito.any())).thenReturn(plugin)
+
         return plugin
     }