Migrated Azure.Storage.Blobs to new generator. (Azure#18830)

Author: seanmcc-msft

sdk/storage/Azure.Storage.Blobs.Batch/tests/BatchStorageRecordedTestSanitizer.cs

@@ -0,0 +1,28 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+using System.Text;
+using System.Text.RegularExpressions;
+using Azure.Storage.Test.Shared;
+
+namespace Azure.Storage.Blobs.Batch.Tests
+{
+    public class BatchStorageRecordedTestSanitizer : StorageRecordedTestSanitizer
+    {
+        private static Regex pattern = new Regex(@"sig=\S+\s", RegexOptions.Compiled);
+
+        public override byte[] SanitizeBody(string contentType, byte[] body)
+        {
+            if (contentType != null && contentType.Contains("multipart/mixed"))
+            {
+                string bodyAsString = Encoding.UTF8.GetString(body);
+                bodyAsString = pattern.Replace(bodyAsString, "sig=Sanitized ");
+                return Encoding.UTF8.GetBytes(bodyAsString);
+            }
+            else
+            {
+                return base.SanitizeBody(contentType, body);
+            }
+        }
+    }
+}

sdk/storage/Azure.Storage.Blobs.Batch/tests/BatchStorageRecordedTestSanitizerTests.cs

@@ -0,0 +1,26 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License.
+
+using System.Text;
+using NUnit.Framework;
+
+namespace Azure.Storage.Blobs.Batch.Tests
+{
+    public class BatchStorageRecordedTestSanitizerTests
+    {
+        [Test]
+        public void ShouldStripSignatureFromBody()
+        {
+            // Arrange
+            var sampleBody = "DELETE /test-container/blob3?sv=2020-06-12&ss=b&srt=sco&st=2021-02-18T19%3A39%3A42Z&se=2021-02-18T21%3A39%3A42Z&sp=rwdxlacuptf&sig=abcde HTTP/1.1\r\n"
+                + "DELETE /test-container/blob3?sv=2020-06-12&ss=b&srt=sco&st=2021-02-18T19%3A39%3A42Z&se=2021-02-18T21%3A39%3A42Z&sp=rwdxlacuptf&sig=abcd HTTP/1.1\r\n";
+            var expectedSanitizedBody = "DELETE /test-container/blob3?sv=2020-06-12&ss=b&srt=sco&st=2021-02-18T19%3A39%3A42Z&se=2021-02-18T21%3A39%3A42Z&sp=rwdxlacuptf&sig=Sanitized HTTP/1.1\r\n"
+                + "DELETE /test-container/blob3?sv=2020-06-12&ss=b&srt=sco&st=2021-02-18T19%3A39%3A42Z&se=2021-02-18T21%3A39%3A42Z&sp=rwdxlacuptf&sig=Sanitized HTTP/1.1\r\n";
+            // Act
+            var sanitizedBody = new BatchStorageRecordedTestSanitizer().SanitizeBody("multipart/mixed", Encoding.UTF8.GetBytes(sampleBody));
+
+            // Assert
+            Assert.AreEqual(expectedSanitizedBody, Encoding.UTF8.GetString(sanitizedBody));
+        }
+    }
+}

sdk/storage/Azure.Storage.Blobs.Batch/tests/BlobBatchClientTests.cs

@@ -8,6 +8,7 @@
 using System.Runtime.InteropServices;
 using System.Threading.Tasks;
 using Azure.Core.TestFramework;
+using Azure.Storage.Blobs.Batch.Tests;
 using Azure.Storage.Blobs.Models;
 using Azure.Storage.Blobs.Specialized;
 using Azure.Storage.Sas;
@@ -25,6 +26,7 @@ public BlobBatchClientTests(bool async, BlobClientOptions.ServiceVersion service
         {
             // Batch delimiters are random so disable body comparison
             Matcher = new RecordMatcher(compareBodies: false);
+            Sanitizer = new BatchStorageRecordedTestSanitizer();
         }
 
         [SetUp]