File tree Expand file tree Collapse file tree 1 file changed +16
-0
lines changed
test/stability-tests/src/it/java/software/amazon/awssdk/stability/tests/kinesis Expand file tree Collapse file tree 1 file changed +16
-0
lines changed Original file line number Diff line number Diff line change 38
38
import software .amazon .awssdk .services .kinesis .model .ConsumerStatus ;
39
39
import software .amazon .awssdk .services .kinesis .model .PutRecordRequest ;
40
40
import software .amazon .awssdk .services .kinesis .model .Record ;
41
+ import software .amazon .awssdk .services .kinesis .model .ResourceInUseException ;
41
42
import software .amazon .awssdk .services .kinesis .model .Shard ;
42
43
import software .amazon .awssdk .services .kinesis .model .ShardIteratorType ;
43
44
import software .amazon .awssdk .services .kinesis .model .StreamStatus ;
@@ -202,6 +203,21 @@ private void waitForStreamToBeActive() {
202
203
.join ())
203
204
.until (b -> b .streamDescription ().streamStatus ().equals (StreamStatus .ACTIVE ))
204
205
.orFailAfter (Duration .ofMinutes (5 ));
206
+
207
+ // Additional verification to ensure stream is fully operational
208
+ Waiter .run (() -> {
209
+ try {
210
+ asyncClient .listShards (r -> r .streamName (streamName )).join ();
211
+ return true ;
212
+ } catch (Exception e ) {
213
+ if (e .getCause () instanceof ResourceInUseException ) {
214
+ return false ;
215
+ }
216
+ throw e ;
217
+ }
218
+ })
219
+ .until (Boolean ::booleanValue )
220
+ .orFailAfter (Duration .ofMinutes (1 ));
205
221
}
206
222
207
223
private void waitForConsumersToBeActive () {
You can’t perform that action at this time.
0 commit comments