ThrowIfVerifyHasBeenRun in comparers (#758)

VerifyTests · Jan 9, 2023 · 96717dd · 96717dd
1 parent d66472e
commit 96717dd
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 5 deletions.
diff --git a/src/Verify.Tests/Comparer/ComparerTests.cs b/src/Verify.Tests/Comparer/ComparerTests.cs
@@ -23,11 +23,16 @@ public async Task Instance_with_message()
         Assert.Contains("theMessage", exception.Message);
     }
 
-    [Fact]
-    public async Task Static_with_message()
+    [ModuleInitializer]
+    public static void Static_with_messageInit()
     {
         FileExtensions.AddTextExtension("staticComparerExtMessage");
         VerifierSettings.RegisterStringComparer("staticComparerExtMessage", CompareWithMessage);
+    }
+
+    [Fact]
+    public async Task Static_with_message()
+    {
         var settings = new VerifySettings();
         settings.DisableDiff();
         settings.UseMethodName("Static_with_message_temp");
@@ -39,11 +44,16 @@ public async Task Static_with_message()
 
 #endif
 
-    [Fact]
-    public async Task Static()
+    [ModuleInitializer]
+    public static void StaticInit()
     {
         FileExtensions.AddTextExtension("staticComparerExt");
         VerifierSettings.RegisterStringComparer("staticComparerExt", Compare);
+    }
+
+    [Fact]
+    public async Task Static()
+    {
         await Verify("TheText", "staticComparerExt");
         PrefixUnique.Clear();
         await Verify("thetext", "staticComparerExt");

diff --git a/src/Verify/Compare/SharedSettings.cs b/src/Verify/Compare/SharedSettings.cs
@@ -27,16 +27,21 @@ internal static bool TryGetStringComparer(string extension, [NotNullWhen(true)]
 
     public static void RegisterStreamComparer(string extension, StreamCompare compare)
     {
+        InnerVerifier.ThrowIfVerifyHasBeenRun();
         Guard.AgainstBadExtension(extension, nameof(extension));
         streamComparers[extension] = compare;
     }
 
     public static void RegisterStringComparer(string extension, StringCompare compare)
     {
+        InnerVerifier.ThrowIfVerifyHasBeenRun();
         Guard.AgainstBadExtension(extension, nameof(extension));
         stringComparers[extension] = compare;
     }
 
-    public static void SetDefaultStringComparer(StringCompare compare) =>
+    public static void SetDefaultStringComparer(StringCompare compare)
+    {
+        InnerVerifier.ThrowIfVerifyHasBeenRun();
         defaultStringComparer = compare;
+    }
 }