Add Mtom Upload Stream test code (#5034)

Author: afifi-ins

global.json

@@ -2,7 +2,7 @@
   "tools": {
     "dotnet": "8.0.100-preview.3.23178.7",
     "runtimes": {
-      "dotnet": [
+      "aspnetcore": [
         "3.1.5",
         "6.0.15",
         "7.0.4"

src/System.Private.ServiceModel/tests/Common/Scenarios/ScenarioTestTypes.cs

@@ -947,6 +947,64 @@ public void OnDataContractPingCallback(ComplexCompositeTypeDuplexCallbackOnly da
     }
 }
 
+public class BloatedStream : Stream
+{
+    private readonly long _length;
+    private long _position;
+
+    public BloatedStream(long length)
+    {
+        _length = length;
+        _position = 0;
+    }
+
+    public override bool CanRead => true;
+
+    public override bool CanSeek => false;
+
+    public override bool CanWrite => false;
+
+    public override long Length => throw new NotSupportedException();
+
+    public override long Position
+    {
+        get => throw new NotSupportedException();
+        set => throw new NotSupportedException();
+    }
+
+    public override void Flush()
+    {
+        throw new NotSupportedException();
+    }
+
+    public override int Read(byte[] buffer, int offset, int count)
+    {
+        if (_position < _length)
+        {
+            var numberOfBytesInThisRead = (int)Math.Min(count, _length - _position);
+            _position += numberOfBytesInThisRead;
+            return numberOfBytesInThisRead;
+        }
+
+        return 0;
+    }
+
+    public override long Seek(long offset, SeekOrigin origin)
+    {
+        throw new NotSupportedException();
+    }
+
+    public override void SetLength(long value)
+    {
+        throw new NotSupportedException();
+    }
+
+    public override void Write(byte[] buffer, int offset, int count)
+    {
+        throw new NotSupportedException();
+    }
+}
+
 public class FlowControlledStream : Stream
 {
     private ManualResetEvent _waitEvent = new ManualResetEvent(false);

src/System.Private.ServiceModel/tests/Scenarios/Binding/Http/BasicHttpBindingTests.4.0.0.cs

@@ -4,11 +4,14 @@
 
 using System;
 using System.Collections.Generic;
+using System.Linq;
 using System.Net.Http;
 using System.ServiceModel;
 using System.ServiceModel.Channels;
 using System.Threading.Tasks;
 using Infrastructure.Common;
+using Microsoft.AspNetCore.Builder;
+using Microsoft.Extensions.Hosting;
 using Xunit;
 
 public static class Binding_Http_BasicHttpBindingTests
@@ -359,4 +362,44 @@ public static void DecompressionEnabled_Echo_RoundTrips_String(bool decompressio
             ScenarioTestHelpers.CloseCommunicationObjects((ICommunicationObject)serviceProxy, factory);
         }
     }
+
+    [WcfTheory]
+    [InlineData(65536)]      // 64KB
+    [InlineData(1048576)]    // 1MB
+    [InlineData(67108864)]   // 64MB
+    [InlineData(4294967296)] // 4GB
+    [OuterLoop]
+    public static async void DefaultSettings_Http_Mtom_Stream_Upload(long uploadBytes)
+    {
+        ChannelFactory<MtomBindingTestHelper.IMtomStreamingService> factory = null;
+        MtomBindingTestHelper.IMtomStreamingService serviceProxy = null;
+
+        try
+        {
+            // *** SETUP *** \\
+            // WCF Service
+            await using (WebApplication app = MtomBindingTestHelper.BuildWCFService())
+            {
+                app.Start();
+
+                // WCF Client
+                factory = new ChannelFactory<MtomBindingTestHelper.IMtomStreamingService>(
+                    MtomBindingTestHelper.CreateMtomClientBinding(),
+                    new EndpointAddress(app.Urls.First(u => u.StartsWith("http:"))));
+                serviceProxy = factory.CreateChannel();
+
+                // *** EXECUTE *** \\
+                long result = serviceProxy.UploadStream(new BloatedStream(uploadBytes));
+            }
+
+            // *** CLEANUP *** \\
+            factory.Close();
+            ((ICommunicationObject)serviceProxy).Close();
+        }
+        finally
+        {
+            // *** ENSURE CLEANUP *** \\
+            ScenarioTestHelpers.CloseCommunicationObjects((ICommunicationObject)serviceProxy, factory);
+        }
+    }
 }

src/System.Private.ServiceModel/tests/Scenarios/Binding/Http/Binding.Http.IntegrationTests.csproj

@@ -13,4 +13,8 @@
     <ProjectReference Include='$(WcfScenarioTestCommonProj)' />
     <ProjectReference Include="$(WcfInfrastructureCommonProj)" />
   </ItemGroup>
+  
+  <ItemGroup>
+    <FrameworkReference Include="Microsoft.AspNetCore.App" />
+  </ItemGroup>
 </Project>

src/System.Private.ServiceModel/tests/Scenarios/Binding/Http/MtomBindingTestHelper.cs

@@ -0,0 +1,82 @@
+// Licensed to the .NET Foundation under one or more agreements.
+// The .NET Foundation licenses this file to you under the MIT license.
+
+using System;
+using System.IO;
+using System.ServiceModel;
+using System.ServiceModel.Channels;
+using System.Text;
+using System.Xml;
+using Microsoft.AspNetCore.Builder;
+using Microsoft.AspNetCore.Hosting;
+using Microsoft.AspNetCore.Http;
+using Microsoft.AspNetCore.Http.Features;
+
+public class MtomBindingTestHelper
+{
+    [ServiceContract]
+    public interface IMtomStreamingService
+    {
+        [OperationContract]
+        long UploadStream(Stream stream);
+    }
+
+    public class MtomStreamingService : IMtomStreamingService
+    {
+        public long UploadStream(Stream stream)
+        {
+            var buffer = new byte[65536];
+            long totalBytes = 0;
+            int bytesRead;
+
+            while ((bytesRead = stream.Read(buffer, 0, buffer.Length)) > 0)
+            {
+                totalBytes += bytesRead;
+            }
+
+            return totalBytes;
+        }
+    }
+
+    public static WebApplication BuildWCFService()
+    {
+        var builder = WebApplication.CreateBuilder();
+        builder.WebHost.ConfigureKestrel(serverOptions =>
+        {
+            serverOptions.AllowSynchronousIO = true;
+            serverOptions.Limits.MaxRequestBodySize = 5_368_709_120;
+        });
+
+        var app = builder.Build();
+
+        app.MapPost("/", async (HttpContext context) =>
+        {
+            context.Features.Get<IHttpMaxRequestBodySizeFeature>().MaxRequestBodySize = 5_368_709_120;
+            var buffer = new byte[8192];
+            while (await context.Request.Body.ReadAsync(buffer, 0, 8192) != 0) { }
+
+            context.Response.Headers.ContentType = "multipart/related; type=\"application/xop+xml\";start=\"<http://tempuri.org/0>\";boundary=\"uuid:fca834ef-6b4a-43c0-a7d0-09064d2827e8+id=1\";start-info=\"text/xml\"";
+            await context.Response.WriteAsync("--uuid:fca834ef-6b4a-43c0-a7d0-09064d2827e8+id=1\r\nContent-ID: <http://tempuri.org/0>\r\nContent-Transfer-Encoding: 8bit\r\nContent-Type: application/xop+xml;charset=utf-8;type=\"text/xml\"\r\n\r\n<s:Envelope xmlns:s=\"http://schemas.xmlsoap.org/soap/envelope/\"><s:Body><UploadStreamResponse xmlns=\"http://tempuri.org/\"></UploadStreamResponse></s:Body></s:Envelope>");
+        });
+        
+        return app;
+    }
+
+    public static Binding CreateMtomClientBinding()
+    {
+        var binding = new CustomBinding();
+        var mtomElement = new MtomMessageEncodingBindingElement(MessageVersion.Soap11, Encoding.UTF8);
+        XmlDictionaryReaderQuotas.Max.CopyTo(mtomElement.ReaderQuotas);
+        binding.Elements.Add(mtomElement);
+        binding.Elements.Add(new HttpTransportBindingElement
+        {
+            TransferMode = TransferMode.Streamed,
+            MaxBufferSize = 1024 * 64,
+            MaxBufferPoolSize = 1,
+            MaxReceivedMessageSize = 5_368_709_120
+        });
+        binding.SendTimeout = TimeSpan.FromMinutes(5);
+        binding.ReceiveTimeout = TimeSpan.FromMinutes(5);
+        return binding;
+    }
+}