@@ -6,6 +6,7 @@ package software.aws.toolkits.jetbrains.services.codemodernizer
66import org.junit.Assert.assertEquals
77import org.junit.Assert.assertNotNull
88import org.junit.Assert.assertNull
9+ import org.junit.Assert.fail
910import org.junit.Test
1011import software.aws.toolkits.jetbrains.services.codemodernizer.constants.buildTransformResultChatContent
1112import software.aws.toolkits.jetbrains.services.codemodernizer.messages.CodeTransformButtonId
@@ -21,8 +22,8 @@ class CodeTransformChatTest {
2122 val chatItem = buildTransformResultChatContent(result)
2223 assertEquals(chatItem.message, message(" codemodernizer.chat.message.result.fail_initial_build" ))
2324 assertNotNull(chatItem.buttons)
24- assertEquals(chatItem.buttons!! .size, 1 )
25- assertEquals(chatItem.buttons!! [ 0 ] .id, CodeTransformButtonId .ViewBuildLog .id)
25+ assertEquals(chatItem.buttons? .size ? : fail( " buttons is null " ) , 1 )
26+ assertEquals(chatItem.buttons?.get( 0 )? .id ? : fail( " buttons is null " ) , CodeTransformButtonId .ViewBuildLog .id)
2627 }
2728
2829 @Test
@@ -31,10 +32,10 @@ class CodeTransformChatTest {
3132 val chatItem = buildTransformResultChatContent(result, 5 )
3233 assertEquals(chatItem.message, message(" codemodernizer.chat.message.result.success.multiple_diffs" ))
3334 assertNotNull(chatItem.buttons)
34- assertEquals(chatItem.buttons!! .size, 2 )
35- assertEquals(chatItem.buttons!! [ 0 ] .id, CodeTransformButtonId .ViewDiff .id)
36- assertEquals(chatItem.buttons!! [ 0 ] .text, " View diff 1/5" )
37- assertEquals(chatItem.buttons!! [ 1 ] .id, CodeTransformButtonId .ViewSummary .id)
35+ assertEquals(chatItem.buttons? .size ? : fail( " buttons is null " ) , 2 )
36+ assertEquals(chatItem.buttons?.get( 0 )? .id ? : fail( " buttons is null " ) , CodeTransformButtonId .ViewDiff .id)
37+ assertEquals(chatItem.buttons?.get( 0 )? .text ? : fail( " buttons is null " ) , " View diff 1/5" )
38+ assertEquals(chatItem.buttons?.get( 1 )? .id ? : fail( " buttons is null " ) , CodeTransformButtonId .ViewSummary .id)
3839 }
3940
4041 @Test
@@ -43,10 +44,10 @@ class CodeTransformChatTest {
4344 val chatItem = buildTransformResultChatContent(result, 1 )
4445 assertEquals(chatItem.message, message(" codemodernizer.chat.message.result.partially_success" ))
4546 assertNotNull(chatItem.buttons)
46- assertEquals(chatItem.buttons!! .size, 2 )
47- assertEquals(chatItem.buttons!! [ 0 ] .id, CodeTransformButtonId .ViewDiff .id)
48- assertEquals(chatItem.buttons!! [ 0 ] .text, " View diff" )
49- assertEquals(chatItem.buttons!! [ 1 ] .id, CodeTransformButtonId .ViewSummary .id)
47+ assertEquals(chatItem.buttons? .size ? : fail( " buttons is null " ) , 2 )
48+ assertEquals(chatItem.buttons?.get( 0 )? .id ? : fail( " buttons is null " ) , CodeTransformButtonId .ViewDiff .id)
49+ assertEquals(chatItem.buttons?.get( 0 )? .text ? : fail( " buttons is null " ) , " View diff" )
50+ assertEquals(chatItem.buttons?.get( 1 )? .id ? : fail( " buttons is null " ) , CodeTransformButtonId .ViewSummary .id)
5051 }
5152
5253 @Test
0 commit comments