Skip to content

Commit bfb4ed7

Browse files
committed
rename cwsprEndpointCustomizer -> ClientCustomizer
1 parent d148b66 commit bfb4ed7

File tree

4 files changed

+17
-17
lines changed

4 files changed

+17
-17
lines changed

plugins/amazonq/codewhisperer/jetbrains-community/resources/META-INF/plugin-codewhisperer.xml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@
6666
</extensions>
6767

6868
<extensions defaultExtensionNs="aws.toolkit.core">
69-
<sdk.clientCustomizer implementation="software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererEndpointCustomizer"/>
69+
<sdk.clientCustomizer implementation="software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererClientCustomizer"/>
7070
</extensions>
7171

7272
<extensions defaultExtensionNs="amazon.q.codewhisperer">
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ import java.time.Duration
3434
import javax.net.ssl.TrustManager
3535

3636
// TODO: move this file to package /client
37-
class CodeWhispererEndpointCustomizer : ToolkitClientCustomizer {
37+
class CodeWhispererClientCustomizer : ToolkitClientCustomizer {
3838

3939
override fun customize(
4040
credentialProvider: AwsCredentialsProvider?,
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ import software.aws.toolkits.jetbrains.core.AwsClientManager
3232
import software.aws.toolkits.jetbrains.core.MockClientManager.Companion.useRealImplementations
3333
import java.util.concurrent.CountDownLatch
3434

35-
class CodeWhispererEndpointCustomizerTest {
35+
class CodeWhispererClientCustomizerTest {
3636
@Rule
3737
@JvmField
3838
val application = ApplicationRule()

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

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -31,11 +31,11 @@ import software.aws.toolkits.jetbrains.core.MockClientManager
3131
import software.aws.toolkits.jetbrains.core.MockClientManagerRule
3232
import software.aws.toolkits.jetbrains.core.credentials.MockCredentialManagerRule
3333
import software.aws.toolkits.jetbrains.services.codewhisperer.explorer.CodeWhispererExplorerActionManager
34-
import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererEndpointCustomizer
34+
import software.aws.toolkits.jetbrains.services.codewhisperer.util.CodeWhispererClientCustomizer
3535
import software.aws.toolkits.jetbrains.settings.CodeWhispererSettings
3636

3737
/**
38-
* If this test class failed, highly likely because the endpoint is overwritten by [CodeWhispererEndpointCustomizer]
38+
* If this test class failed, highly likely because the endpoint is overwritten by [CodeWhispererClientCustomizer]
3939
*/
4040
class CodeWhispererClientTest {
4141
val applicationRule = ApplicationRule()
@@ -90,8 +90,8 @@ class CodeWhispererClientTest {
9090
mockClient.getAccessToken {}
9191
verify(
9292
postRequestedFor(urlEqualTo("/"))
93-
.withoutHeader(CodeWhispererEndpointCustomizer.OPTOUT_KEY_NAME)
94-
.withoutHeader(CodeWhispererEndpointCustomizer.OPTOUT_KEY_NAME)
93+
.withoutHeader(CodeWhispererClientCustomizer.OPTOUT_KEY_NAME)
94+
.withoutHeader(CodeWhispererClientCustomizer.OPTOUT_KEY_NAME)
9595
)
9696
}
9797

@@ -101,16 +101,16 @@ class CodeWhispererClientTest {
101101
// default is opt-in (thus opt-out = false)
102102
verify(
103103
postRequestedFor(urlEqualTo("/"))
104-
.withHeader(CodeWhispererEndpointCustomizer.TOKEN_KEY_NAME, matching(CodeWhispererTestUtil.testValidAccessToken))
105-
.withHeader(CodeWhispererEndpointCustomizer.OPTOUT_KEY_NAME, matching("false"))
104+
.withHeader(CodeWhispererClientCustomizer.TOKEN_KEY_NAME, matching(CodeWhispererTestUtil.testValidAccessToken))
105+
.withHeader(CodeWhispererClientCustomizer.OPTOUT_KEY_NAME, matching("false"))
106106
)
107107

108108
CodeWhispererSettings.getInstance().toggleMetricOptIn(false)
109109
mockClient.listRecommendations {}
110110
verify(
111111
postRequestedFor(urlEqualTo("/"))
112-
.withHeader(CodeWhispererEndpointCustomizer.TOKEN_KEY_NAME, matching(CodeWhispererTestUtil.testValidAccessToken))
113-
.withHeader(CodeWhispererEndpointCustomizer.OPTOUT_KEY_NAME, matching("true"))
112+
.withHeader(CodeWhispererClientCustomizer.TOKEN_KEY_NAME, matching(CodeWhispererTestUtil.testValidAccessToken))
113+
.withHeader(CodeWhispererClientCustomizer.OPTOUT_KEY_NAME, matching("true"))
114114
)
115115
}
116116

@@ -119,8 +119,8 @@ class CodeWhispererClientTest {
119119
mockClient.createCodeScan {}
120120
verify(
121121
postRequestedFor(urlEqualTo("/"))
122-
.withHeader(CodeWhispererEndpointCustomizer.TOKEN_KEY_NAME, matching(CodeWhispererTestUtil.testValidAccessToken))
123-
.withoutHeader(CodeWhispererEndpointCustomizer.OPTOUT_KEY_NAME)
122+
.withHeader(CodeWhispererClientCustomizer.TOKEN_KEY_NAME, matching(CodeWhispererTestUtil.testValidAccessToken))
123+
.withoutHeader(CodeWhispererClientCustomizer.OPTOUT_KEY_NAME)
124124

125125
)
126126
}
@@ -130,8 +130,8 @@ class CodeWhispererClientTest {
130130
mockClient.createCodeScanUploadUrl {}
131131
verify(
132132
postRequestedFor(urlEqualTo("/"))
133-
.withHeader(CodeWhispererEndpointCustomizer.TOKEN_KEY_NAME, matching(CodeWhispererTestUtil.testValidAccessToken))
134-
.withoutHeader(CodeWhispererEndpointCustomizer.OPTOUT_KEY_NAME)
133+
.withHeader(CodeWhispererClientCustomizer.TOKEN_KEY_NAME, matching(CodeWhispererTestUtil.testValidAccessToken))
134+
.withoutHeader(CodeWhispererClientCustomizer.OPTOUT_KEY_NAME)
135135
)
136136
}
137137

@@ -140,8 +140,8 @@ class CodeWhispererClientTest {
140140
mockClient.listCodeScanFindings {}
141141
verify(
142142
postRequestedFor(urlEqualTo("/"))
143-
.withHeader(CodeWhispererEndpointCustomizer.TOKEN_KEY_NAME, matching(CodeWhispererTestUtil.testValidAccessToken))
144-
.withoutHeader(CodeWhispererEndpointCustomizer.OPTOUT_KEY_NAME)
143+
.withHeader(CodeWhispererClientCustomizer.TOKEN_KEY_NAME, matching(CodeWhispererTestUtil.testValidAccessToken))
144+
.withoutHeader(CodeWhispererClientCustomizer.OPTOUT_KEY_NAME)
145145
)
146146
}
147147
}

0 commit comments

Comments
 (0)