Skip to content

Commit ccf5f0d

Browse files
committed
tst
1 parent 20d5180 commit ccf5f0d

File tree

1 file changed

+18
-0
lines changed

1 file changed

+18
-0
lines changed

plugins/amazonq/codewhisperer/jetbrains-community/tst/software/aws/toolkits/jetbrains/services/codewhisperer/CodeWhispererFileContextProviderTest.kt

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -160,6 +160,24 @@ class CodeWhispererFileContextProviderTest {
160160
assertThat(result.isProcessTimeout).isTrue
161161
}
162162

163+
@Test
164+
fun `should return empty if both project context and opentabs context return empty`() = runTest {
165+
sut = spy(sut)
166+
167+
mockProjectContext.stub { onBlocking { queryInline(any(), any()) }.doReturn(emptyList()) }
168+
val queryPsi = projectRule.fixture.addFileToProject("Foo.java", "public Foo {}")
169+
val mockFileContext = aFileContextInfo(CodeWhispererJava.INSTANCE)
170+
171+
val result = sut.extractSupplementalFileContextForSrc(queryPsi, mockFileContext)
172+
173+
verify(sut, times(1)).fetchProjectContext(any(), any(), any())
174+
verify(sut, times(1)).fetchOpenTabsContext(any(), any(), any())
175+
176+
assertThat(result.isUtg).isFalse
177+
assertThat(result.strategy).isEqualTo(CrossFileStrategy.Empty)
178+
assertThat(result.contents).isEmpty()
179+
}
180+
163181
@Test
164182
fun `should only use openTabsContext if projectContext is empty`() = runTest {
165183
mockProjectContext.stub { onBlocking { queryInline(any(), any()) }.doReturn(emptyList()) }

0 commit comments

Comments
 (0)