CE-881 - Updated to actually run in CI (e.g., w/ localstack)

This commit is contained in:
2024-04-01 15:54:09 -05:00
parent d56637b9dd
commit fb8257d34a

View File

@ -26,19 +26,14 @@ import java.io.ByteArrayOutputStream;
import java.io.IOException; import java.io.IOException;
import java.nio.charset.StandardCharsets; import java.nio.charset.StandardCharsets;
import java.time.Instant; import java.time.Instant;
import com.amazonaws.auth.AWSStaticCredentialsProvider; import com.kingsrook.qqq.backend.module.filesystem.s3.BaseS3Test;
import com.amazonaws.auth.BasicAWSCredentials;
import com.amazonaws.services.s3.AmazonS3;
import com.amazonaws.services.s3.AmazonS3ClientBuilder;
import com.kingsrook.qqq.backend.module.filesystem.BaseTest;
import io.github.cdimascio.dotenv.Dotenv;
import org.junit.jupiter.api.Test; import org.junit.jupiter.api.Test;
/******************************************************************************* /*******************************************************************************
** Unit test for S3UploadOutputStream ** Unit test for S3UploadOutputStream
*******************************************************************************/ *******************************************************************************/
class S3UploadOutputStreamTest extends BaseTest class S3UploadOutputStreamTest extends BaseS3Test
{ {
/******************************************************************************* /*******************************************************************************
@ -47,15 +42,7 @@ class S3UploadOutputStreamTest extends BaseTest
@Test @Test
void test() throws IOException void test() throws IOException
{ {
Dotenv dotenv = Dotenv.load(); String bucketName = BaseS3Test.BUCKET_NAME;
BasicAWSCredentials credentials = new BasicAWSCredentials(dotenv.get("NFONE_S3_ACCESS_KEY"), dotenv.get("NFONE_S3_SECRET_KEY"));
AmazonS3 amazonS3 = AmazonS3ClientBuilder.standard()
.withCredentials(new AWSStaticCredentialsProvider(credentials))
.withRegion(dotenv.get("NFONE_S3_REGION"))
.build();
String bucketName = "bucket-upload-archive-nf-one-dev";
String key = "uploader-tests/" + Instant.now().toString() + ".txt"; String key = "uploader-tests/" + Instant.now().toString() + ".txt";
// S3UploadOutputStream outputStream = new S3UploadOutputStream(amazonS3, bucketName, key); // S3UploadOutputStream outputStream = new S3UploadOutputStream(amazonS3, bucketName, key);
@ -70,7 +57,7 @@ class S3UploadOutputStreamTest extends BaseTest
outputStream.write("\n]\n".getBytes(StandardCharsets.UTF_8)); outputStream.write("\n]\n".getBytes(StandardCharsets.UTF_8));
outputStream.close(); outputStream.close();
S3UploadOutputStream s3UploadOutputStream = new S3UploadOutputStream(amazonS3, bucketName, key); S3UploadOutputStream s3UploadOutputStream = new S3UploadOutputStream(getS3Utils().getAmazonS3(), bucketName, key);
s3UploadOutputStream.write(outputStream.toByteArray(), 0, 5 * 1024 * 1024); s3UploadOutputStream.write(outputStream.toByteArray(), 0, 5 * 1024 * 1024);
s3UploadOutputStream.write(outputStream.toByteArray(), 0, 3 * 1024 * 1024); s3UploadOutputStream.write(outputStream.toByteArray(), 0, 3 * 1024 * 1024);
s3UploadOutputStream.write(outputStream.toByteArray(), 0, 3 * 1024 * 1024); s3UploadOutputStream.write(outputStream.toByteArray(), 0, 3 * 1024 * 1024);