Skip to content

Commit e3fd998

Browse files
committed
fix ut
1 parent 0fb13ef commit e3fd998

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

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

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ import software.aws.toolkits.jetbrains.core.region.MockRegionProviderRule
4343
import software.aws.toolkits.jetbrains.services.amazonq.CodeWhispererFeatureConfigService
4444
import software.aws.toolkits.jetbrains.services.amazonq.FeatureContext
4545
import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfile
46+
import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfileManager
4647
import software.aws.toolkits.jetbrains.services.amazonq.profile.QRegionProfileSelectedListener
4748
import software.aws.toolkits.jetbrains.services.codewhisperer.credentials.CodeWhispererClientAdaptor
4849
import software.aws.toolkits.jetbrains.services.codewhisperer.customization.CodeWhispererCustomization
@@ -83,6 +84,7 @@ class CodeWhispererModelConfiguratorTest {
8384
private lateinit var mockClient: CodeWhispererRuntimeClient
8485
private lateinit var abManager: CodeWhispererFeatureConfigService
8586
private lateinit var mockClintAdaptor: CodeWhispererClientAdaptor
87+
private lateinit var mockQRegionProfileManager: QRegionProfileManager
8688

8789
@Before
8890
fun setup() {
@@ -125,6 +127,11 @@ class CodeWhispererModelConfiguratorTest {
125127

126128
mockClintAdaptor = mock()
127129
projectRule.project.registerServiceInstance(CodeWhispererClientAdaptor::class.java, mockClintAdaptor)
130+
131+
mockQRegionProfileManager = mock {
132+
on { listRegionProfiles(any()) }.thenReturn(listOf(QRegionProfile("fake_name", "fake_arn")))
133+
}
134+
ApplicationManager.getApplication().replaceService(QRegionProfileManager::class.java, mockQRegionProfileManager, disposableRule.disposable)
128135
}
129136

130137
@Test

0 commit comments

Comments
 (0)