tst

Author: rli

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

@@ -31,13 +31,10 @@ import software.amazon.awssdk.services.codewhispererruntime.model.CodeAnalysisSt
 import software.amazon.awssdk.services.codewhispererruntime.model.CompletionType
 import software.amazon.awssdk.services.codewhispererruntime.model.CreateUploadUrlRequest
 import software.amazon.awssdk.services.codewhispererruntime.model.CreateUploadUrlResponse
-import software.amazon.awssdk.services.codewhispererruntime.model.Customization
 import software.amazon.awssdk.services.codewhispererruntime.model.GenerateCompletionsRequest
 import software.amazon.awssdk.services.codewhispererruntime.model.GetCodeAnalysisRequest
 import software.amazon.awssdk.services.codewhispererruntime.model.GetCodeAnalysisResponse
 import software.amazon.awssdk.services.codewhispererruntime.model.IdeCategory
-import software.amazon.awssdk.services.codewhispererruntime.model.ListAvailableCustomizationsRequest
-import software.amazon.awssdk.services.codewhispererruntime.model.ListAvailableCustomizationsResponse
 import software.amazon.awssdk.services.codewhispererruntime.model.ListCodeAnalysisFindingsRequest
 import software.amazon.awssdk.services.codewhispererruntime.model.ListCodeAnalysisFindingsResponse
 import software.amazon.awssdk.services.codewhispererruntime.model.ListFeatureEvaluationsRequest
@@ -51,7 +48,6 @@ import software.amazon.awssdk.services.codewhispererruntime.model.StartCodeAnaly
 import software.amazon.awssdk.services.codewhispererruntime.model.StartCodeAnalysisResponse
 import software.amazon.awssdk.services.codewhispererruntime.model.SuggestionState
 import software.amazon.awssdk.services.codewhispererruntime.paginators.GenerateCompletionsIterable
-import software.amazon.awssdk.services.codewhispererruntime.paginators.ListAvailableCustomizationsIterable
 import software.amazon.awssdk.services.ssooidc.SsoOidcClient
 import software.aws.toolkits.core.utils.test.aString
 import software.aws.toolkits.jetbrains.core.MockClientManagerRule
@@ -158,48 +154,6 @@ class CodeWhispererClientAdaptorTest {
             .isInstanceOf(CodeWhispererRuntimeClient::class.java)
     }
 
-    @Test
-    fun `listCustomizations`() {
-        val sdkIterable = ListAvailableCustomizationsIterable(bearerClient, ListAvailableCustomizationsRequest.builder().build())
-        val mockResponse1 = ListAvailableCustomizationsResponse.builder()
-            .customizations(
-                listOf(
-                    Customization.builder().name("custom-1").arn("arn-1").build(),
-                    Customization.builder().name("custom-2").arn("arn-2").build()
-                )
-            )
-            .nextToken("token-1")
-            .responseMetadata(metadata)
-            .sdkHttpResponse(sdkHttpResponse)
-            .build() as ListAvailableCustomizationsResponse
-
-        val mockResponse2 = ListAvailableCustomizationsResponse.builder()
-            .customizations(
-                listOf(
-                    Customization.builder().name("custom-3").arn("arn-3").build(),
-                )
-            )
-            .nextToken("")
-            .responseMetadata(metadata)
-            .sdkHttpResponse(sdkHttpResponse)
-            .build() as ListAvailableCustomizationsResponse
-
-        bearerClient.stub { client ->
-            on { client.listAvailableCustomizations(any<ListAvailableCustomizationsRequest>()) } doReturnConsecutively listOf(mockResponse1, mockResponse2)
-            on { client.listAvailableCustomizationsPaginator(any<ListAvailableCustomizationsRequest>()) } doReturn sdkIterable
-        }
-
-        val actual = sut.listAvailableCustomizations()
-        assertThat(actual).hasSize(3)
-        assertThat(actual).isEqualTo(
-            listOf(
-                CodeWhispererCustomization(name = "custom-1", arn = "arn-1"),
-                CodeWhispererCustomization(name = "custom-2", arn = "arn-2"),
-                CodeWhispererCustomization(name = "custom-3", arn = "arn-3")
-            )
-        )
-    }
-
     @Test
     fun `generateCompletionsPaginator - bearer`() {
         val request = pythonRequest

plugins/amazonq/shared/jetbrains-community/src/software/aws/toolkits/jetbrains/services/codewhisperer/customization/CodeWhispererCustomization.kt

@@ -4,7 +4,12 @@
 package software.aws.toolkits.jetbrains.services.codewhisperer.customization
 
 data class CodeWhispererCustomization(
+    @JvmField
     val arn: String = "",
+
+    @JvmField
     val name: String = "",
+
+    @JvmField
     val description: String? = null,
 )