Skip to content

Commit 7a9c239

Browse files
Fix CloudWatch TestChecksumInvoke and TestCompressionCall unit tests
1 parent 0338c62 commit 7a9c239

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

sdk/test/UnitTests/Custom/Runtime/ChecksumHandlerTests.cs

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@
2323
using Amazon.Runtime.Internal.Auth;
2424
using Amazon.Util;
2525
using Amazon.Runtime.Internal.Util;
26+
using System.Linq;
2627

2728
namespace AWSSDK.UnitTests
2829
{
@@ -93,7 +94,7 @@ RequestChecksumCalculation requestChecksumCalculation
9394
{
9495
// Make sure that we don't set checksum header twice when it already exists
9596
Assert.IsTrue(request.Headers[headerKey].Equals(JunkChecksumHeaderValue));
96-
Assert.IsTrue(request.Headers.Count == 1);
97+
Assert.IsTrue(request.Headers.Count(h => h.Key.StartsWith("x-amz-checksum-")) == 1);
9798
}
9899
else if (headerKey != null)
99100
{
@@ -109,7 +110,7 @@ RequestChecksumCalculation requestChecksumCalculation
109110
}
110111
else
111112
{
112-
Assert.IsTrue(request.Headers.Count == 0);
113+
Assert.IsTrue(request.Headers.Count(h => h.Key.StartsWith("x-amz-checksum-")) == 0);
113114
}
114115
}
115116
#endif

sdk/test/UnitTests/Custom/Runtime/CompressionHandlerTests.cs

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,6 @@ public async Task TestCompressionCall(bool? disableCompression, long? minCompres
7676

7777
if (notCompressed)
7878
{
79-
Assert.IsNull(request.Content);
8079
CollectionAssert.AreEqual(originalInput, newInput);
8180
}
8281
else
@@ -121,7 +120,6 @@ public async Task TestStreamCompressionCall(bool? disableCompression, long? minC
121120

122121
await handler.InvokeAsync<AmazonWebServiceResponse>(executionContext);
123122

124-
Assert.IsNull(request.Content);
125123
Assert.IsNotNull(request.ContentStream);
126124
Assert.AreEqual(originalInputLen, request.ContentStream.Length);
127125

0 commit comments

Comments
 (0)