feat(studio): 补齐剩余工作台聚合接口与真实对接

common-agent-agent/src/test/java/com/bruce/agent/session/AgentSessionServiceTests.java

@@ -16,6 +16,7 @@ import org.mockito.InjectMocks;
 import org.mockito.Mock;
 import org.mockito.Spy;
 import org.mockito.junit.jupiter.MockitoExtension;
+import org.springframework.beans.factory.ObjectProvider;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertThrows;
@@ -32,6 +33,9 @@ class AgentSessionServiceTests {
     @Mock
     private IAgentDefinitionService agentDefinitionService;
 
+    @Mock
+    private ObjectProvider<IAgentDefinitionService> agentDefinitionServiceProvider;
+
     @Spy
     @InjectMocks
     private AgentSessionServiceImpl agentSessionService;
@@ -42,6 +46,7 @@ class AgentSessionServiceTests {
 
     @Test
     void createSessionShouldRejectDisabledAgent() {
+        when(agentDefinitionServiceProvider.getIfAvailable()).thenReturn(agentDefinitionService);
         AgentDefinition agent = new AgentDefinition();
         agent.setId(1L);
         agent.setStatus("DISABLED");
@@ -56,6 +61,7 @@ class AgentSessionServiceTests {
 
     @Test
     void createSessionShouldPersistActiveSession() {
+        when(agentDefinitionServiceProvider.getIfAvailable()).thenReturn(agentDefinitionService);
         AgentDefinition agent = new AgentDefinition();
         agent.setId(1L);
         agent.setStatus(EnableStatusEnum.ENABLED.name());