Skip to content

Commit f971b9b

Browse files
Copilotphrocker
andcommitted
Fix MCP proxy tests to work with updated service constructor
Co-authored-by: phrocker <[email protected]>
1 parent 08121fd commit f971b9b

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

llm-proxy/src/test/java/io/sentrius/sso/mcp/MCPProxyServiceTest.java

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,9 @@
77
import io.sentrius.sso.core.services.security.KeycloakService;
88
import io.sentrius.sso.core.services.security.ZeroTrustAccessTokenService;
99
import io.sentrius.sso.core.services.security.ZeroTrustRequestService;
10+
import io.sentrius.sso.core.services.agents.AgentClientService;
11+
import io.sentrius.sso.core.services.agents.AgentExecutionService;
12+
import io.sentrius.sso.core.services.agents.ZeroTrustClientService;
1013
import io.sentrius.sso.provenance.kafka.ProvenanceKafkaProducer;
1114

1215
import com.fasterxml.jackson.databind.ObjectMapper;
@@ -39,6 +42,15 @@ public class MCPProxyServiceTest {
3942
@Mock
4043
private ZeroTrustRequestService ztrService;
4144

45+
@Mock
46+
private AgentClientService agentClientService;
47+
48+
@Mock
49+
private AgentExecutionService agentExecutionService;
50+
51+
@Mock
52+
private ZeroTrustClientService zeroTrustClientService;
53+
4254
@Mock
4355
private ProvenanceKafkaProducer provenanceKafkaProducer;
4456

@@ -55,6 +67,9 @@ void setUp() {
5567
keycloakService,
5668
ztatService,
5769
ztrService,
70+
agentClientService,
71+
agentExecutionService,
72+
zeroTrustClientService,
5873
provenanceKafkaProducer,
5974
restTemplate,
6075
objectMapper

0 commit comments

Comments
 (0)