Skip to content

Commit 285bf98

Browse files
committed
fix: fix code format
Signed-off-by: Sun Yuhan <[email protected]>
1 parent 36c654a commit 285bf98

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

spring-ai-client-chat/src/test/java/org/springframework/ai/chat/client/advisor/SafeGuardAdvisorTests.java

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -59,8 +59,8 @@ class SafeGuardAdvisorTests {
5959

6060
@BeforeEach
6161
void setUp() {
62-
safeRequest = new ChatClientRequest(Prompt.builder().content("hello world").build(), Map.of());
63-
unsafeRequest = new ChatClientRequest(Prompt.builder().content("this contains secret").build(), Map.of());
62+
this.safeRequest = new ChatClientRequest(Prompt.builder().content("hello world").build(), Map.of());
63+
this.unsafeRequest = new ChatClientRequest(Prompt.builder().content("this contains secret").build(), Map.of());
6464
}
6565

6666
@Test
@@ -82,7 +82,7 @@ void adviseCallInterceptsWhenContainsSensitiveWord() {
8282

8383
CallAdvisorChain mockChain = mock(CallAdvisorChain.class);
8484

85-
ChatClientResponse response = advisor.adviseCall(unsafeRequest, mockChain);
85+
ChatClientResponse response = advisor.adviseCall(this.unsafeRequest, mockChain);
8686

8787
assertThat(response.chatResponse().getResult().getOutput().getText()).contains("I'm unable to respond");
8888
verify(mockChain, never()).nextCall(any());
@@ -98,12 +98,12 @@ void adviseCallPassesThroughWhenNoSensitiveWord() {
9898
.context(Map.of())
9999
.build();
100100

101-
when(mockChain.nextCall(safeRequest)).thenReturn(expected);
101+
when(mockChain.nextCall(this.safeRequest)).thenReturn(expected);
102102

103-
ChatClientResponse response = advisor.adviseCall(safeRequest, mockChain);
103+
ChatClientResponse response = advisor.adviseCall(this.safeRequest, mockChain);
104104

105105
assertThat(response).isSameAs(expected);
106-
verify(mockChain).nextCall(safeRequest);
106+
verify(mockChain).nextCall(this.safeRequest);
107107
}
108108

109109
@Test
@@ -112,7 +112,7 @@ void adviseStreamInterceptsWhenContainsSensitiveWord() {
112112

113113
StreamAdvisorChain mockChain = mock(StreamAdvisorChain.class);
114114

115-
Flux<ChatClientResponse> flux = advisor.adviseStream(unsafeRequest, mockChain);
115+
Flux<ChatClientResponse> flux = advisor.adviseStream(this.unsafeRequest, mockChain);
116116

117117
StepVerifier.create(flux)
118118
.assertNext(r -> assertThat(r.chatResponse().getResult().getOutput().getText())
@@ -132,13 +132,13 @@ void adviseStreamPassesThroughWhenNoSensitiveWord() {
132132
.context(Map.of())
133133
.build();
134134

135-
when(mockChain.nextStream(safeRequest)).thenReturn(Flux.just(expected));
135+
when(mockChain.nextStream(this.safeRequest)).thenReturn(Flux.just(expected));
136136

137-
Flux<ChatClientResponse> flux = advisor.adviseStream(safeRequest, mockChain);
137+
Flux<ChatClientResponse> flux = advisor.adviseStream(this.safeRequest, mockChain);
138138

139139
StepVerifier.create(flux).expectNext(expected).verifyComplete();
140140

141-
verify(mockChain).nextStream(safeRequest);
141+
verify(mockChain).nextStream(this.safeRequest);
142142
}
143143

144144
@Test

0 commit comments

Comments
 (0)