Skip to content

Commit 1e2b21a

Browse files
committed
add test for queryInline
1 parent b684c01 commit 1e2b21a

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

plugins/amazonq/chat/jetbrains-community/tst/software/aws/toolkits/jetbrains/services/amazonq/workspace/context/ProjectContextProviderTest.kt

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@ import software.aws.toolkits.jetbrains.services.amazonq.project.InlineBm25Chunk
3333
import software.aws.toolkits.jetbrains.services.amazonq.project.LspMessage
3434
import software.aws.toolkits.jetbrains.services.amazonq.project.ProjectContextProvider
3535
import software.aws.toolkits.jetbrains.services.amazonq.project.QueryChatRequest
36+
import software.aws.toolkits.jetbrains.services.amazonq.project.QueryInlineCompletionRequest
3637
import software.aws.toolkits.jetbrains.services.amazonq.project.RelevantDocument
3738
import software.aws.toolkits.jetbrains.services.amazonq.project.UpdateIndexRequest
3839
import software.aws.toolkits.jetbrains.utils.rules.CodeInsightTestFixtureRule
@@ -193,6 +194,25 @@ class ProjectContextProviderTest {
193194
)
194195
}
195196

197+
@Test
198+
fun `queryInline should send correct encrypted request to lsp`() {
199+
sut.queryInline("foo", "Foo.java")
200+
201+
val request = QueryInlineCompletionRequest("foo", "Foo.java")
202+
val requestJson = mapper.writeValueAsString(request)
203+
204+
assertThat(mapper.readTree(requestJson)).isEqualTo(mapper.readTree("""{ "query": "foo", "filePath": "Foo.java" }"""))
205+
206+
val encryptedRequest = encoderServer.encrypt(requestJson)
207+
208+
wireMock.verify(
209+
1,
210+
postRequestedFor(urlPathEqualTo("/queryInlineProjectContext"))
211+
.withHeader("Content-Type", equalTo("text/plain"))
212+
.withRequestBody(equalTo(encryptedRequest))
213+
)
214+
}
215+
196216
@Test
197217
fun `query chat should return empty if result set non deserializable`() = runTest {
198218
stubFor(

0 commit comments

Comments
 (0)