Merge pull request #8 from i-am-bee/feat/7-vitest-logging

test: improve vitest logging

tests/setup.ts

@@ -15,6 +15,9 @@
  */
 
 import dotenv from "dotenv";
+import { FrameworkError } from "bee-agent-framework/errors";
+import * as util from "node:util";
+import { hasProps } from "bee-agent-framework/internals/helpers/object";
 dotenv.config();
 dotenv.config({
   path: ".env.test",
@@ -24,3 +27,43 @@ dotenv.config({
   path: ".env.test.local",
   override: true,
 });
+
+function isFrameworkErrorLike(error: unknown): error is Record<keyof FrameworkError, any> {
+  const keys = ["errors", "context", "isRetryable", "isFatal"] as (keyof FrameworkError)[];
+  return hasProps(keys)(error as Record<keyof FrameworkError, any>);
+}
+
+afterEach(() => {
+  onTestFailed((testCase) => {
+    const errors = testCase.errors ?? [];
+    for (const error of errors) {
+      if (isFrameworkErrorLike(error)) {
+        error.message = util
+          .inspect(
+            {
+              message: error.message,
+              context: error.context,
+              cause: error.cause,
+              isFatal: error.isFatal,
+              isRetryable: error.isRetryable,
+              errors: error.errors,
+            },
+            {
+              compact: false,
+              depth: Infinity,
+            },
+          )
+          .replaceAll("[Object: null prototype]", "");
+      }
+    }
+  });
+});
+
+expect.addSnapshotSerializer({
+  serialize(val: FrameworkError): string {
+    return val.explain();
+  },
+  test(val): boolean {
+    return val && val instanceof FrameworkError;
+  },
+});