Captured max output sizes in tests

Author: PawelGerr

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)
    {
    }
 

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)
    {
    }
 

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)
    {
    }
 

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)
    {
    }
 

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)
    {
    }
 

test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/NewtonsoftJson/ReadJson.cs

@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;
 public class NewtonsoftJsonSmartEnumSourceGeneratorTests : SourceGeneratorTestsBase
 {
    public NewtonsoftJsonSmartEnumSourceGeneratorTests(ITestOutputHelper output)
-      : base(output)
+      : base(output, 1_000)
    {
    }
 

test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/NewtonsoftJsonSmartEnumSourceGeneratorTests.cs

@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;
 public class NewtonsoftJsonValueObjectSourceGeneratorTests : SourceGeneratorTestsBase
 {
    public NewtonsoftJsonValueObjectSourceGeneratorTests(ITestOutputHelper output)
-      : base(output)
+      : base(output, 8_000)
    {
    }
 

test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/NewtonsoftJsonValueObjectSourceGeneratorTests.cs

@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;
 public class ObjectFactorySourceGeneratorTests : SourceGeneratorTestsBase
 {
    public ObjectFactorySourceGeneratorTests(ITestOutputHelper output)
-      : base(output)
+      : base(output, 2_000)
    {
    }
 

test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/ObjectFactorySourceGeneratorTests.cs

@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;
 public class RegularUnionSourceGeneratorTests : SourceGeneratorTestsBase
 {
    public RegularUnionSourceGeneratorTests(ITestOutputHelper output)
-      : base(output)
+      : base(output, 8_000)
    {
    }
 

test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/RegularUnionSourceGeneratorTests.cs

@@ -7,7 +7,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;
 public class SmartEnumSourceGeneratorTests : SourceGeneratorTestsBase
 {
    public SmartEnumSourceGeneratorTests(ITestOutputHelper output)
-      : base(output)
+      : base(output, 40_000)
    {
    }
 

test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/SmartEnumSourceGeneratorTests.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);

test/Thinktecture.Runtime.Extensions.SourceGenerator.Tests/SourceGeneratorTests/SourceGeneratorTestsBase.cs

@@ -10,7 +10,7 @@ namespace Thinktecture.Runtime.Tests.SourceGeneratorTests;
 public class ValueObjectSourceGeneratorTests : SourceGeneratorTestsBase
 {
    public ValueObjectSourceGeneratorTests(ITestOutputHelper output)
-      : base(output)
+      : base(output, 13_000)
    {
    }