@@ -31,11 +31,11 @@ import software.aws.toolkits.jetbrains.core.MockClientManager
3131import software.aws.toolkits.jetbrains.core.MockClientManagerRule
3232import software.aws.toolkits.jetbrains.core.credentials.MockCredentialManagerRule
3333import 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
3535import 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 */
4040class 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