Skip to content

Commit 8504a13

Browse files
authored
add simple test for Kinesis PutRecords (#21)
1 parent 364a97c commit 8504a13

File tree

1 file changed

+23
-4
lines changed

1 file changed

+23
-4
lines changed

src/test/java/cloud/localstack/awssdkv2/BasicFeaturesSDKV2Test.java

Lines changed: 23 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,8 @@
2323

2424
import java.util.*;
2525
import java.net.*;
26+
import java.nio.ByteBuffer;
27+
import software.amazon.awssdk.core.SdkBytes;
2628

2729
@RunWith(LocalstackTestRunner.class)
2830
public class BasicFeaturesSDKV2Test {
@@ -33,7 +35,7 @@ public class BasicFeaturesSDKV2Test {
3335

3436
@Test
3537
public void testCreateSqsQueueV2() throws Exception {
36-
String queueName = "test-q-2159";
38+
String queueName = "test-q-"+ UUID.randomUUID().toString();
3739
CreateQueueRequest request = CreateQueueRequest.builder().queueName(queueName).build();
3840
SqsAsyncClient sqsClient = TestUtils.getClientSQSAsyncV2();
3941
CreateQueueResponse queue = sqsClient.createQueue(request).get();
@@ -42,17 +44,34 @@ public void testCreateSqsQueueV2() throws Exception {
4244

4345
@Test
4446
public void testCreateKinesisStreamV2() throws Exception {
45-
String streamName = "test-s-3198";
47+
String streamName = "test-s-"+ UUID.randomUUID().toString();
4648
KinesisAsyncClient kinesisClient = TestUtils.getClientKinesisAsyncV2();
4749
CreateStreamRequest request = CreateStreamRequest.builder()
4850
.streamName(streamName).shardCount(1).build();
4951
CreateStreamResponse response = kinesisClient.createStream(request).get();
5052
Assert.assertNotNull(response);
5153
}
54+
55+
@Test
56+
public void testCreateKinesisRecordV2() throws Exception {
57+
String streamName = "test-s-"+UUID.randomUUID().toString();
58+
KinesisAsyncClient kinesisClient = TestUtils.getClientKinesisAsyncV2();
59+
CreateStreamRequest request = CreateStreamRequest.builder()
60+
.streamName(streamName).shardCount(1).build();
61+
CreateStreamResponse response = kinesisClient.createStream(request).get();
62+
Assert.assertNotNull(response);
63+
64+
SdkBytes payload = SdkBytes.fromByteBuffer(ByteBuffer.wrap(String.format("testData-%d", 1).getBytes()));
65+
PutRecordRequest.Builder putRecordRequest = PutRecordRequest.builder();
66+
putRecordRequest.streamName(streamName);
67+
putRecordRequest.data(payload);
68+
putRecordRequest.partitionKey(String.format("partitionKey-%d", 1));
69+
Assert.assertNotNull(kinesisClient.putRecord(putRecordRequest.build()));
70+
}
5271

5372
@Test
5473
public void testS3CreateListBuckets() throws Exception {
55-
String bucketName = "test-b-9716";
74+
String bucketName = "test-b-"+UUID.randomUUID().toString();
5675
S3AsyncClient s3Client = TestUtils.getClientS3AsyncV2();
5776
CreateBucketRequest request = CreateBucketRequest.builder().bucket(bucketName).build();
5877
CreateBucketResponse response = s3Client.createBucket(request).get();
@@ -67,7 +86,7 @@ public void testS3CreateListBuckets() throws Exception {
6786
public void testSendSNSMessage() throws Exception {
6887
// Test integration of SNS messaging with LocalStack using SDK v2
6988

70-
final String topicName = "test-t-6210";
89+
final String topicName = "test-t-"+UUID.randomUUID().toString();
7190
final SnsAsyncClient clientSNS = TestUtils.getClientSNSAsyncV2();
7291
CreateTopicResponse createTopicResponse = clientSNS.createTopic(
7392
CreateTopicRequest.builder().name(topicName).build()).get();

0 commit comments

Comments
 (0)