@alloy-js/python 0.5.0-dev.0 → 0.5.0-dev.2

package/test/references.test.tsx

@@ -1,36 +1,37 @@
 import { refkey } from "@alloy-js/core";
-import { describe, it } from "vitest";
+import { describe, expect, it } from "vitest";
 import * as py from "../src/index.js";
-import { assertFileContents, toSourceTextMultiple } from "./utils.jsx";
+import { TestOutputDirectory } from "./utils.js";
 
 describe("Reference", () => {
   it("Verifies that a reference is correctly resolved", () => {
     const rk1 = refkey();
-    const res = toSourceTextMultiple([
-      <py.SourceFile path="models.py">
-        <py.ClassDeclaration name="User" refkey={rk1} />
-      </py.SourceFile>,
-      <py.SourceFile path="services.py">
-        <py.VariableDeclaration
-          name="current_user"
-          type={rk1}
-          initializer={
-            <py.ClassInstantiation target={rk1} args={['"Marvin"']} />
-          }
-        />
-      </py.SourceFile>,
-    ]);
-    assertFileContents(res, {
+    expect(
+      <TestOutputDirectory>
+        <py.SourceFile path="models.py">
+          <py.ClassDeclaration name="User" refkey={rk1} />
+        </py.SourceFile>
+        <py.SourceFile path="services.py">
+          <py.VariableDeclaration
+            name="current_user"
+            type={rk1}
+            initializer={
+              <py.ClassInstantiation target={rk1} args={['"Marvin"']} />
+            }
+          />
+        </py.SourceFile>
+      </TestOutputDirectory>,
+    ).toRenderTo({
       "models.py": `
-        class User:
-            pass
+          class User:
+              pass
 
-        `,
+          `,
       "services.py": `
-        from models import User
+          from models import User
 
-        current_user: User = User("Marvin")
-        `,
+          current_user: User = User("Marvin")
+          `,
     });
   });
 });