File tree Expand file tree Collapse file tree 1 file changed +20
-0
lines changed Expand file tree Collapse file tree 1 file changed +20
-0
lines changed Original file line number Diff line number Diff line change @@ -33,6 +33,7 @@ import software.aws.toolkits.jetbrains.services.amazonq.project.InlineBm25Chunk
3333import software.aws.toolkits.jetbrains.services.amazonq.project.LspMessage
3434import software.aws.toolkits.jetbrains.services.amazonq.project.ProjectContextProvider
3535import software.aws.toolkits.jetbrains.services.amazonq.project.QueryChatRequest
36+ import software.aws.toolkits.jetbrains.services.amazonq.project.QueryInlineCompletionRequest
3637import software.aws.toolkits.jetbrains.services.amazonq.project.RelevantDocument
3738import software.aws.toolkits.jetbrains.services.amazonq.project.UpdateIndexRequest
3839import 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(
You can’t perform that action at this time.
0 commit comments