@@ -186,8 +186,8 @@ void kinesisMessageDrivenChannelAdapter() {
186186 assertThat (message ).isNotNull ();
187187 assertThat (message .getPayload ()).isInstanceOf (List .class );
188188 List <String > payload = (List <String >) message .getPayload ();
189- assertThat (payload ).size (). isEqualTo (1 );
190- String record = payload .get ( 0 );
189+ assertThat (payload ).hasSize (1 );
190+ String record = payload .getFirst ( );
191191 assertThat (record ).isEqualTo ("bar" );
192192
193193 Object partitionKeyHeader = message .getHeaders ().get (AwsHeaders .RECEIVED_PARTITION_KEY );
@@ -219,7 +219,7 @@ void kinesisMessageDrivenChannelAdapter() {
219219 assertThat (message ).isNotNull ();
220220 assertThat (message .getPayload ()).isInstanceOf (List .class );
221221 List <String > messagePayload = (List <String >) message .getPayload ();
222- assertThat (messagePayload ).size (). isEqualTo (3 );
222+ assertThat (messagePayload ).hasSize (3 );
223223
224224 Object messageSequenceNumberHeader = message .getHeaders ().get (AwsHeaders .RECEIVED_SEQUENCE_NUMBER );
225225 assertThat (messageSequenceNumberHeader ).isInstanceOf (List .class );
@@ -235,10 +235,8 @@ void kinesisMessageDrivenChannelAdapter() {
235235 assertThat (message ).isNotNull ();
236236 assertThat (message .getPayload ()).isInstanceOf (List .class );
237237 messagePayload = (List <String >) message .getPayload ();
238- assertThat (messagePayload ).size ().isEqualTo (2 );
239- assertThat (messagePayload )
240- .contains ("bar" )
241- .contains ("foobar" );
238+ assertThat (messagePayload ).hasSize (2 );
239+ assertThat (messagePayload ).contains ("bar" , "foobar" );
242240
243241 this .kinesisMessageDrivenChannelAdapter .stop ();
244242
0 commit comments