chore(zui): refactor testing + speed up some important tests (#382)

botpress · Sep 27, 2024 · 0acfad9 · 0acfad9
1 parent 5a77496
commit 0acfad9
Show file tree

Hide file tree

Showing 4 changed files with 278 additions and 264 deletions.
diff --git a/zui/src/setup.test.ts b/zui/src/setup.test.ts
@@ -1,5 +1,6 @@
 import { expect } from 'vitest'
 import { Project, Diagnostic } from 'ts-morph'
+import { format } from 'prettier'
 
 export function isValidTypescript(
   code: string,
@@ -31,11 +32,17 @@ expect.extend({
   },
 })
 
+const formatTs = async (code: string): Promise<string> => {
+  code = code.replace(/\s+/g, ' ')
+  code = await format(code, { parser: 'typescript' })
+  return code
+}
+
 expect.extend({
-  toMatchWithoutFormatting(received: string, expected: string, _) {
+  async toMatchWithoutFormatting(received: string, expected: string, _) {
     const { isNot } = this
-    const transformedReceived = received.replace(/\s+/g, '')
-    const transformedExpected = expected.replace(/\s+/g, '')
+    const transformedReceived = await formatTs(received)
+    const transformedExpected = await formatTs(expected)
 
     return {
       pass: transformedExpected === transformedReceived,