refactor: create custom error class

artilleryio · Nov 23, 2023 · 3700de4 · 3700de4
1 parent ce620dc
commit 3700de4
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/packages/artillery-plugin-expect/index.js b/packages/artillery-plugin-expect/index.js
@@ -83,7 +83,7 @@ function expectationsPluginOnError(
   events,
   done
 ) {
-  if (scenarioErr.message.startsWith('Failed expectations for request')) {
+  if (scenarioErr instanceof FailedExpectationError) {
     return done();
   }
   if (userContext.expectationsPlugin.formatter === 'json') {
@@ -186,7 +186,9 @@ function expectationsPluginCheckExpectations(
         ? req.name
         : req.url;
     return done(
-      new Error(`Failed expectations for request ${filteredRequestName}`)
+      new FailedExpectationError(
+        `Failed expectations for request ${filteredRequestName}`
+      )
     );
   }
 
@@ -213,3 +215,10 @@ function maybeParseBody(res) {
     return res.body;
   }
 }
+
+class FailedExpectationError extends Error {
+  constructor(message) {
+    super(message);
+    this.name = 'FailedExpectationError';
+  }
+}