fix: BunitBrowserFile throws when filesize too big

Author: linkdotnet

src/bunit.web/Extensions/InputFile/BUnitBrowserFile.cs

@@ -27,7 +27,12 @@ public BUnitBrowserFile(
 
 	public Stream OpenReadStream(long maxAllowedSize = 512000, CancellationToken cancellationToken = default)
 	{
+		if (Size > maxAllowedSize)
+		{
+			throw new IOException($"Supplied file with size {Size} bytes exceeds the maximum of {maxAllowedSize} bytes.");
+		}
+
 		return new MemoryStream(Content);
 	}
 }
-#endif
+#endif

src/bunit.web/Extensions/InputFile/InputFileExtensions.cs

@@ -1,4 +1,5 @@
 #if NET5_0_OR_GREATER
+using System.Runtime.ExceptionServices;
 using Microsoft.AspNetCore.Components.Forms;
 
 namespace Bunit;
@@ -36,6 +37,11 @@ public static void UploadFiles(
 		{
 			uploadTask.GetAwaiter().GetResult();
 		}
+
+		if (uploadTask.Exception is { InnerException: not null} e)
+		{
+			ExceptionDispatchInfo.Capture(e.InnerException).Throw();
+		}
 	}
 }
-#endif
+#endif

tests/bunit.web.tests/Extensions/InputFile/InputFileTests.cs

@@ -102,9 +102,23 @@ public void Test008()
 
 		act.ShouldNotThrow();
 	}
+	
+	[Fact(DisplayName = "Uploading file exceeding the maximum file size will throw an exception")]
+	public void Test009()
+	{
+		var cut = RenderComponent<InputFileComponent>(ps => ps.Add(p => p.MaxFileSize, 512));
+		var file = InputFileContent.CreateFromText(new string('a', 513));
+
+		Action act = () => cut.FindComponent<InputFile>().UploadFiles(file);
+
+		act.ShouldThrow<IOException>();
+	}
 
 	private sealed class InputFileComponent : ComponentBase
 	{
+		[Parameter]
+		public long MaxFileSize { get; set; } = 512000;
+		
 		public string? Filename { get; private set; }
 		public string? Content { get; private set; }
 		public DateTimeOffset? LastChanged { get; private set; }
@@ -127,7 +141,7 @@ private void OnChange(InputFileChangeEventArgs args)
 			Filename = file.Name;
 			LastChanged = file.LastModified;
 			Size = file.Size;
-			using var stream = new StreamReader(file.OpenReadStream());
+			using var stream = new StreamReader(file.OpenReadStream(MaxFileSize));
 			Content = stream.ReadToEnd();
 		}
 	}