unit test for chunking stream

Author: kai-ion

tests/aws-cpp-sdk-core-tests/utils/stream/ChunkingInterceptorTest.cpp

@@ -0,0 +1,87 @@
+/**
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ * SPDX-License-Identifier: Apache-2.0.
+ */
+#include <aws/core/http/standard/StandardHttpRequest.h>
+#include <aws/core/utils/crypto/CRC32.h>
+#include <smithy/client/features/ChunkingInterceptor.h>
+#include <aws/testing/AwsCppSdkGTestSuite.h>
+
+using namespace Aws;
+using namespace Aws::Http::Standard;
+using namespace smithy::client::features;
+using namespace Aws::Utils::Crypto;
+
+class ChunkingInterceptorTest : public Aws::Testing::AwsCppSdkGTestSuite {};
+
+const char* CHUNKING_TEST_LOG_TAG = "CHUNKING_INTERCEPTOR_TEST";
+
+// Helper function to read all data from a stream
+std::string ReadAllFromStream(std::shared_ptr<Aws::IOStream> stream) {
+    std::stringstream result;
+    char buffer[1024];
+    while (stream->good() && !stream->eof()) {
+        stream->read(buffer, sizeof(buffer));
+        auto bytesRead = stream->gcount();
+        if (bytesRead > 0) {
+            result.write(buffer, bytesRead);
+        }
+    }
+    return result.str();
+}
+
+TEST_F(ChunkingInterceptorTest, ChunkedStreamShouldWork) {
+    StandardHttpRequest request{"www.elda.com/will", Http::HttpMethod::HTTP_GET};
+    auto requestHash = Aws::MakeShared<CRC32>(CHUNKING_TEST_LOG_TAG);
+    request.SetRequestHash("crc32", requestHash);
+    std::shared_ptr<Aws::IOStream> inputStream = Aws::MakeShared<Aws::StringStream>(CHUNKING_TEST_LOG_TAG, "1234567890123456789012345");
+    
+    AwsChunkedStreamWrapper wrapper{&request, inputStream, 10};
+    auto chunkedStream = wrapper.GetIOStream();
+    
+    std::string output = ReadAllFromStream(std::shared_ptr<Aws::IOStream>(chunkedStream, [](Aws::IOStream*){}));
+    auto expectedStreamWithChecksum = "A\r\n1234567890\r\nA\r\n1234567890\r\n5\r\n12345\r\n0\r\nx-amz-checksum-crc32:78DeVw==\r\n\r\n";
+    EXPECT_EQ(expectedStreamWithChecksum, output);
+}
+
+TEST_F(ChunkingInterceptorTest, ShouldNotRequireTwoReadsOnSmallChunk) {
+    StandardHttpRequest request{"www.clemar.com/strohl", Http::HttpMethod::HTTP_GET};
+    auto requestHash = Aws::MakeShared<CRC32>(CHUNKING_TEST_LOG_TAG);
+    request.SetRequestHash("crc32", requestHash);
+    std::shared_ptr<Aws::IOStream> inputStream = Aws::MakeShared<Aws::StringStream>(CHUNKING_TEST_LOG_TAG, "12345");
+    
+    AwsChunkedStreamWrapper wrapper{&request, inputStream, 100};
+    auto chunkedStream = wrapper.GetIOStream();
+    
+    std::string output = ReadAllFromStream(std::shared_ptr<Aws::IOStream>(chunkedStream, [](Aws::IOStream*){}));
+    auto expectedStreamWithChecksum = "5\r\n12345\r\n0\r\nx-amz-checksum-crc32:y/U6HA==\r\n\r\n";
+    EXPECT_EQ(expectedStreamWithChecksum, output);
+}
+
+TEST_F(ChunkingInterceptorTest, ShouldWorkOnSmallBuffer) {
+    StandardHttpRequest request{"www.eugief.com/hesimay", Http::HttpMethod::HTTP_GET};
+    auto requestHash = Aws::MakeShared<CRC32>(CHUNKING_TEST_LOG_TAG);
+    request.SetRequestHash("crc32", requestHash);
+    std::shared_ptr<Aws::IOStream> inputStream = Aws::MakeShared<Aws::StringStream>(CHUNKING_TEST_LOG_TAG, "1234567890");
+    
+    AwsChunkedStreamWrapper wrapper{&request, inputStream, 5};
+    auto chunkedStream = wrapper.GetIOStream();
+    
+    std::string output = ReadAllFromStream(std::shared_ptr<Aws::IOStream>(chunkedStream, [](Aws::IOStream*){}));
+    auto expectedStreamWithChecksum = "5\r\n12345\r\n5\r\n67890\r\n0\r\nx-amz-checksum-crc32:Jh2u5Q==\r\n\r\n";
+    EXPECT_EQ(expectedStreamWithChecksum, output);
+}
+
+TEST_F(ChunkingInterceptorTest, ShouldWorkOnEmptyStream) {
+    StandardHttpRequest request{"www.nidia.com/juna", Http::HttpMethod::HTTP_GET};
+    auto requestHash = Aws::MakeShared<CRC32>(CHUNKING_TEST_LOG_TAG);
+    request.SetRequestHash("crc32", requestHash);
+    std::shared_ptr<Aws::IOStream> inputStream = Aws::MakeShared<Aws::StringStream>(CHUNKING_TEST_LOG_TAG, "");
+    
+    AwsChunkedStreamWrapper wrapper{&request, inputStream, 5};
+    auto chunkedStream = wrapper.GetIOStream();
+    
+    std::string output = ReadAllFromStream(std::shared_ptr<Aws::IOStream>(chunkedStream, [](Aws::IOStream*){}));
+    auto expectedStreamWithChecksum = "0\r\nx-amz-checksum-crc32:AAAAAA==\r\n\r\n";
+    EXPECT_EQ(expectedStreamWithChecksum, output);
+}