Captured max output sizes in tests

PawelGerr · PawelGerr · commit 2758f861aedd · 2025-05-11T12:02:18.000+02:00
diff --git a/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/AdHocUnionSourceGeneratorTests.cs b/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/AdHocUnionSourceGeneratorTests.cs
@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;
 public class AdHocUnionSourceGeneratorTests : SourceGeneratorTestsBase
 {
    public AdHocUnionSourceGeneratorTests(ITestOutputHelper output)
-      : base(output)
+      : base(output, 23_000)
    {
    }
 
diff --git a/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/JsonSmartEnumSourceGeneratorTests.cs b/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/JsonSmartEnumSourceGeneratorTests.cs
@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;
 public class JsonSmartEnumSourceGeneratorTests : SourceGeneratorTestsBase
 {
    public JsonSmartEnumSourceGeneratorTests(ITestOutputHelper output)
-      : base(output)
+      : base(output, 1_000)
    {
    }
 
diff --git a/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/JsonValueObjectSourceGeneratorTests.cs b/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/JsonValueObjectSourceGeneratorTests.cs
@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;
 public class JsonValueObjectSourceGeneratorTests : SourceGeneratorTestsBase
 {
    public JsonValueObjectSourceGeneratorTests(ITestOutputHelper output)
-      : base(output)
+      : base(output, 8_000)
    {
    }
 
diff --git a/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/MessagePackSmartEnumSourceGeneratorTests.cs b/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/MessagePackSmartEnumSourceGeneratorTests.cs
@@ -9,7 +9,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;
 public class MessagePackSmartEnumSourceGeneratorTests : SourceGeneratorTestsBase
 {
    public MessagePackSmartEnumSourceGeneratorTests(ITestOutputHelper output)
-      : base(output)
+      : base(output, 1_000)
    {
    }
 
diff --git a/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/MessagePackValueObjectSourceGeneratorTests.cs b/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/MessagePackValueObjectSourceGeneratorTests.cs
@@ -9,7 +9,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;
 public class MessagePackValueObjectSourceGeneratorTests : SourceGeneratorTestsBase
 {
    public MessagePackValueObjectSourceGeneratorTests(ITestOutputHelper output)
-      : base(output)
+      : base(output, 4_000)
    {
    }
 
diff --git a/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/NewtonsoftJson/ReadJson.cs b/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/NewtonsoftJson/ReadJson.cs
diff --git a/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/NewtonsoftJsonSmartEnumSourceGeneratorTests.cs b/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/NewtonsoftJsonSmartEnumSourceGeneratorTests.cs
@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;
 public class NewtonsoftJsonSmartEnumSourceGeneratorTests : SourceGeneratorTestsBase
 {
    public NewtonsoftJsonSmartEnumSourceGeneratorTests(ITestOutputHelper output)
-      : base(output)
+      : base(output, 1_000)
    {
    }
 
diff --git a/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/NewtonsoftJsonValueObjectSourceGeneratorTests.cs b/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/NewtonsoftJsonValueObjectSourceGeneratorTests.cs
@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;
 public class NewtonsoftJsonValueObjectSourceGeneratorTests : SourceGeneratorTestsBase
 {
    public NewtonsoftJsonValueObjectSourceGeneratorTests(ITestOutputHelper output)
-      : base(output)
+      : base(output, 8_000)
    {
    }
 
diff --git a/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/ObjectFactorySourceGeneratorTests.cs b/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/ObjectFactorySourceGeneratorTests.cs
@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;
 public class ObjectFactorySourceGeneratorTests : SourceGeneratorTestsBase
 {
    public ObjectFactorySourceGeneratorTests(ITestOutputHelper output)
-      : base(output)
+      : base(output, 2_000)
    {
    }
 
diff --git a/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/RegularUnionSourceGeneratorTests.cs b/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/RegularUnionSourceGeneratorTests.cs
@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;
 public class RegularUnionSourceGeneratorTests : SourceGeneratorTestsBase
 {
    public RegularUnionSourceGeneratorTests(ITestOutputHelper output)
-      : base(output)
+      : base(output, 8_000)
    {
    }
 
diff --git a/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/SmartEnumSourceGeneratorTests.cs b/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/SmartEnumSourceGeneratorTests.cs
@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;
 public class SmartEnumSourceGeneratorTests : SourceGeneratorTestsBase
 {
    public SmartEnumSourceGeneratorTests(ITestOutputHelper output)
-      : base(output)
+      : base(output, 40_000)
    {
    }
 
diff --git a/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/SourceGeneratorTestsBase.cs b/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/SourceGeneratorTestsBase.cs
@@ -15,10 +15,14 @@ public abstract class SourceGeneratorTestsBase
    private const string _GENERATION_ERROR = "CS8785";
 
    private readonly ITestOutputHelper _output;
+   private readonly int _maxOutputSize;
 
-   protected SourceGeneratorTestsBase(ITestOutputHelper output)
+   protected SourceGeneratorTestsBase(
+      ITestOutputHelper output,
+      int maxOutputSize)
    {
       _output = output ?? throw new ArgumentNullException(nameof(output));
+      _maxOutputSize = maxOutputSize;
    }
 
    protected Task VerifyAsync(
@@ -48,6 +52,9 @@ protected async Task VerifyAsync(
                                        throw new Exception($"Output file '{fileName}' not found. Available files: {String.Join(", ", outputs.Keys)}", ex);
                                     }
 
+                                    if (content.Length > _maxOutputSize)
+                                       throw new Exception($"Output file '{fileName}' is too big. Actual size: {content.Length}. Max size: {_maxOutputSize}.");
+
                                     var verify = Verifier.Verify(content);
                                     var paramText = parameterText;
 
@@ -68,13 +75,19 @@ protected async Task VerifyAsync(
       string parameterText,
       string output)
    {
+      if (output.Length > _maxOutputSize)
+         throw new Exception($"Output file is too big. Actual size: {output.Length}. Max size: {_maxOutputSize}.");
+
       await Verifier.Verify(output)
                     .UseTextForParameters(parameterText);
    }
 
    protected async Task VerifyAsync(
       string output)
    {
+      if (output.Length > _maxOutputSize)
+         throw new Exception($"Output file is too big. Actual size: {output.Length}. Max size: {_maxOutputSize}.");
+
       await Verifier.Verify(output);
    }
 
@@ -101,7 +114,7 @@ protected string GetGeneratedOutput<T>(
       return output;
    }
 
-   public static Dictionary<string, string> GetGeneratedOutputs<T>(string source, params Assembly[] furtherAssemblies)
+   protected static Dictionary<string, string> GetGeneratedOutputs<T>(string source, params Assembly[] furtherAssemblies)
       where T : IIncrementalGenerator, new()
    {
       var syntaxTree = CSharpSyntaxTree.ParseText(source);
diff --git a/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/SystemTextJson/ReadJson.cs b/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/SystemTextJson/ReadJson.cs
diff --git a/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/ValueObjectSourceGeneratorTests.cs b/test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/ValueObjectSourceGeneratorTests.cs
@@ -10,7 +10,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;
 public class ValueObjectSourceGeneratorTests : SourceGeneratorTestsBase
 {
    public ValueObjectSourceGeneratorTests(ITestOutputHelper output)
-      : base(output)
+      : base(output, 13_000)
    {
    }
 

Original file line number	Diff line number	Diff line change
`@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;`
`7`	`7`	`public class AdHocUnionSourceGeneratorTests : SourceGeneratorTestsBase`
`8`	`8`	`{`
`9`	`9`	`public AdHocUnionSourceGeneratorTests(ITestOutputHelper output)`
`10`		`- : base(output)`
	`10`	`+ : base(output, 23_000)`
`11`	`11`	`{`
`12`	`12`	`}`
`13`	`13`
Original file line number	Diff line number	Diff line change
`@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;`
`7`	`7`	`public class JsonSmartEnumSourceGeneratorTests : SourceGeneratorTestsBase`
`8`	`8`	`{`
`9`	`9`	`public JsonSmartEnumSourceGeneratorTests(ITestOutputHelper output)`
`10`		`- : base(output)`
	`10`	`+ : base(output, 1_000)`
`11`	`11`	`{`
`12`	`12`	`}`
`13`	`13`
Original file line number	Diff line number	Diff line change
`@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;`
`7`	`7`	`public class JsonValueObjectSourceGeneratorTests : SourceGeneratorTestsBase`
`8`	`8`	`{`
`9`	`9`	`public JsonValueObjectSourceGeneratorTests(ITestOutputHelper output)`
`10`		`- : base(output)`
	`10`	`+ : base(output, 8_000)`
`11`	`11`	`{`
`12`	`12`	`}`
`13`	`13`
Original file line number	Diff line number	Diff line change
`@@ -9,7 +9,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;`
`9`	`9`	`public class MessagePackSmartEnumSourceGeneratorTests : SourceGeneratorTestsBase`
`10`	`10`	`{`
`11`	`11`	`public MessagePackSmartEnumSourceGeneratorTests(ITestOutputHelper output)`
`12`		`- : base(output)`
	`12`	`+ : base(output, 1_000)`
`13`	`13`	`{`
`14`	`14`	`}`
`15`	`15`
Original file line number	Diff line number	Diff line change
`@@ -9,7 +9,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;`
`9`	`9`	`public class MessagePackValueObjectSourceGeneratorTests : SourceGeneratorTestsBase`
`10`	`10`	`{`
`11`	`11`	`public MessagePackValueObjectSourceGeneratorTests(ITestOutputHelper output)`
`12`		`- : base(output)`
	`12`	`+ : base(output, 4_000)`
`13`	`13`	`{`
`14`	`14`	`}`
`15`	`15`
Original file line number	Diff line number	Diff line change
`@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;`
`7`	`7`	`public class NewtonsoftJsonSmartEnumSourceGeneratorTests : SourceGeneratorTestsBase`
`8`	`8`	`{`
`9`	`9`	`public NewtonsoftJsonSmartEnumSourceGeneratorTests(ITestOutputHelper output)`
`10`		`- : base(output)`
	`10`	`+ : base(output, 1_000)`
`11`	`11`	`{`
`12`	`12`	`}`
`13`	`13`
Original file line number	Diff line number	Diff line change
`@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;`
`7`	`7`	`public class NewtonsoftJsonValueObjectSourceGeneratorTests : SourceGeneratorTestsBase`
`8`	`8`	`{`
`9`	`9`	`public NewtonsoftJsonValueObjectSourceGeneratorTests(ITestOutputHelper output)`
`10`		`- : base(output)`
	`10`	`+ : base(output, 8_000)`
`11`	`11`	`{`
`12`	`12`	`}`
`13`	`13`
Original file line number	Diff line number	Diff line change
`@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;`
`7`	`7`	`public class ObjectFactorySourceGeneratorTests : SourceGeneratorTestsBase`
`8`	`8`	`{`
`9`	`9`	`public ObjectFactorySourceGeneratorTests(ITestOutputHelper output)`
`10`		`- : base(output)`
	`10`	`+ : base(output, 2_000)`
`11`	`11`	`{`
`12`	`12`	`}`
`13`	`13`
Original file line number	Diff line number	Diff line change
`@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;`
`7`	`7`	`public class RegularUnionSourceGeneratorTests : SourceGeneratorTestsBase`
`8`	`8`	`{`
`9`	`9`	`public RegularUnionSourceGeneratorTests(ITestOutputHelper output)`
`10`		`- : base(output)`
	`10`	`+ : base(output, 8_000)`
`11`	`11`	`{`
`12`	`12`	`}`
`13`	`13`