Add custom error classes for pre and post dialers error reporting

stacks-network · Feb 5, 2025 · 834596d · 834596d
1 parent c832641
commit 834596d
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 9 deletions.
diff --git a/dialer.ts b/dialer.ts
@@ -69,3 +69,17 @@ export class DialerRegistry {
     }
   }
 }
+
+export class PreDialerError extends Error {
+  constructor(message: string) {
+    super(message);
+    this.name = "Pre-dialer error";
+  }
+}
+
+export class PostDialerError extends Error {
+  constructor(message: string) {
+    super(message);
+    this.name = "Post-dialer error";
+  }
+}
diff --git a/invariant.ts b/invariant.ts
@@ -18,7 +18,7 @@ import {
   isTraitReferenceFunction,
 } from "./traits";
 import { EnrichedContractInterfaceFunction } from "./shared.types";
-import { DialerRegistry } from "./dialer";
+import { DialerRegistry, PostDialerError, PreDialerError } from "./dialer";
 
 /**
  * Runs invariant testing on the target contract and logs the progress. Reports
@@ -278,7 +278,7 @@ export const checkInvariants = async (
               });
             }
           } catch (e: any) {
-            throw new Error(`Error in pre dialer: ${e.message}`);
+            throw new PreDialerError(e.message);
           }
 
           try {
@@ -325,7 +325,7 @@ export const checkInvariants = async (
                   });
                 }
               } catch (e: any) {
-                throw new Error(`Error in post dialer: ${e.message}`);
+                throw new PostDialerError(e.message);
               }
             } else {
               radio.emit(
@@ -341,12 +341,10 @@ export const checkInvariants = async (
               );
             }
           } catch (error: any) {
-            // If the pre dialer fails, throw an error.
-            if (error.message.startsWith("Error in pre dialer")) {
-              throw error;
-            }
-            // If the post dialer fails, throw an error.
-            if (error.message.startsWith("Error in post dialer")) {
+            if (
+              error instanceof PreDialerError ||
+              error instanceof PostDialerError
+            ) {
               throw error;
             } else {
               // If the function call fails with a runtime error, log a dimmed