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

package/test/imports.test.tsx

@@ -8,17 +8,18 @@ import * as py from "../src/index.js";
 import { createPythonSymbol } from "../src/symbol-creation.js";
 import { ImportedSymbol, ImportRecords } from "../src/symbols/index.js";
 import {
-  assertFileContents,
   createPythonModuleScope,
-  toSourceText,
-  toSourceTextMultiple,
+  TestOutput,
+  TestOutputDirectory,
 } from "./utils.jsx";
 
 describe("ImportStatement", () => {
   it("renders module import", () => {
-    const result = toSourceText([<ImportStatement path="sys" />]);
-    const expected = `import sys`;
-    expect(result).toRenderTo(expected);
+    expect(
+      <TestOutput>
+        <ImportStatement path="sys" />
+      </TestOutput>,
+    ).toRenderTo("import sys");
   });
 
   it("renders named imports", () => {
@@ -28,11 +29,11 @@ describe("ImportStatement", () => {
       new ImportedSymbol(sqrtSymbol, sqrtSymbol),
       new ImportedSymbol(piSymbol, piSymbol),
     ]);
-    const result = toSourceText([
-      <ImportStatement path="math" symbols={symbols} />,
-    ]);
-    const expected = `from math import pi, sqrt`;
-    expect(result).toRenderTo(expected);
+    expect(
+      <TestOutput>
+        <ImportStatement path="math" symbols={symbols} />
+      </TestOutput>,
+    ).toRenderTo("from math import pi, sqrt");
   });
 });
 
@@ -59,14 +60,19 @@ describe("ImportStatements", () => {
       [sysModuleScope, { symbols: new Set<ImportedSymbol>() }],
     ]);
 
-    const result = toSourceText([<ImportStatements records={records} />]);
-    const expected = `
+    expect(
+      <TestOutput>
+        <ImportStatements records={records} />
+      </TestOutput>,
+    ).toRenderTo(
+      `
     from math import pi
     from math import sqrt
     from requests import get
-    import sys`;
-    expect(result).toRenderTo(expected);
+    import sys`,
+    );
   });
+
   it("renders multiple import statements, but joining imports from the same module", () => {
     const pythonModuleScope = createPythonModuleScope("math", undefined);
     const sqrtSymbol = createPythonSymbol("sqrt", undefined);
@@ -87,13 +93,15 @@ describe("ImportStatements", () => {
       [requestsScope, { symbols: requestsSymbols }],
     ]);
 
-    const result = toSourceText([
-      <ImportStatements records={records} joinImportsFromSameModule={true} />,
-    ]);
-    const expected = `
+    expect(
+      <TestOutput>
+        <ImportStatements records={records} joinImportsFromSameModule={true} />
+      </TestOutput>,
+    ).toRenderTo(
+      `
     from math import pi, sqrt
-    from requests import get, post`;
-    expect(result).toRenderTo(expected);
+    from requests import get, post`,
+    );
   });
 });
 
@@ -102,36 +110,41 @@ describe("Imports being used", () => {
     const rk1 = refkey();
     const rk2 = refkey();
     const rk3 = refkey();
-    const result = toSourceTextMultiple([
-      <py.SourceFile path="test_1.py">
-        <py.VariableDeclaration name="conflict" refkey={rk1} />
-      </py.SourceFile>,
-      <py.SourceFile path="test_3.py">
-        <py.VariableDeclaration name="conflict" refkey={rk3} />
-      </py.SourceFile>,
-      <py.SourceFile path="test_2.py">
-        <py.VariableDeclaration name="conflict" refkey={rk2} />
-      </py.SourceFile>,
-      <py.SourceFile path="test.py">
-        <py.StatementList>
-          <py.VariableDeclaration name="one" initializer={rk1} />
-          <py.VariableDeclaration name="three" initializer={rk3} />
-          <py.VariableDeclaration name="two" initializer={rk2} />
-        </py.StatementList>
-      </py.SourceFile>,
-    ]);
-    assertFileContents(result, {
+    expect(
+      <TestOutputDirectory>
+        <py.SourceFile path="test_1.py">
+          <py.VariableDeclaration name="conflict" refkey={rk1} />
+        </py.SourceFile>
+        <py.SourceFile path="test_3.py">
+          <py.VariableDeclaration name="conflict" refkey={rk3} />
+        </py.SourceFile>
+        <py.SourceFile path="test_2.py">
+          <py.VariableDeclaration name="conflict" refkey={rk2} />
+        </py.SourceFile>
+        <py.SourceFile path="test.py">
+          <py.StatementList>
+            <py.VariableDeclaration name="one" initializer={rk1} />
+            <py.VariableDeclaration name="three" initializer={rk3} />
+            <py.VariableDeclaration name="two" initializer={rk2} />
+          </py.StatementList>
+        </py.SourceFile>
+      </TestOutputDirectory>,
+    ).toRenderTo({
       "test.py": `
-        from test_1 import conflict
-        from test_2 import conflict as conflict_3_test_2
-        from test_3 import conflict as conflict_2_test_3
-
-        one = conflict
-        three = conflict_2_test_3
-        two = conflict_3_test_2
-      `,
+          from test_1 import conflict
+          from test_2 import conflict as conflict_3_test_2
+          from test_3 import conflict as conflict_2_test_3
+
+          one = conflict
+          three = conflict_2_test_3
+          two = conflict_3_test_2
+        `,
+      "test_1.py": "conflict = None",
+      "test_2.py": "conflict = None",
+      "test_3.py": "conflict = None",
     });
   });
+
   it("works with importing the same name many times from different files and with the correct order", () => {
     const rk1 = refkey();
     const rk2 = refkey();
@@ -140,50 +153,65 @@ describe("Imports being used", () => {
     const rk5 = refkey();
     const rk6 = refkey();
     const rk7 = refkey();
-    const result = toSourceTextMultiple([
-      <py.SourceFile path="test_1.py">
-        <py.VariableDeclaration name="conflict" refkey={rk1} />
-        <py.VariableDeclaration name="something_else" refkey={rk4} />
-      </py.SourceFile>,
-      <py.SourceFile path="test_2.py">
-        <py.VariableDeclaration name="conflict" refkey={rk2} />
-        <py.VariableDeclaration name="something" refkey={rk6} />
-        <py.VariableDeclaration name="something_else" refkey={rk5} />
-      </py.SourceFile>,
-      <py.SourceFile path="test_3.py">
-        <py.VariableDeclaration name="conflict" refkey={rk3} />
-        <py.VariableDeclaration name="something" refkey={rk7} />
-      </py.SourceFile>,
-      <py.SourceFile path="test.py">
-        <py.StatementList>
-          <py.VariableDeclaration name="one" initializer={rk1} />
-          <py.VariableDeclaration name="two" initializer={rk2} />
-          <py.VariableDeclaration name="three" initializer={rk3} />
-          <py.VariableDeclaration name="something_else" initializer={rk4} />
-          <py.VariableDeclaration name="something_else_two" initializer={rk5} />
-          <py.VariableDeclaration name="something" initializer={rk6} />
-          <py.VariableDeclaration name="something_two" initializer={rk7} />
-        </py.StatementList>
-      </py.SourceFile>,
-    ]);
-    assertFileContents(result, {
+    expect(
+      <TestOutputDirectory>
+        <py.SourceFile path="test_1.py">
+          <py.VariableDeclaration name="conflict" refkey={rk1} />
+          <py.VariableDeclaration name="something_else" refkey={rk4} />
+        </py.SourceFile>
+        <py.SourceFile path="test_2.py">
+          <py.VariableDeclaration name="conflict" refkey={rk2} />
+          <py.VariableDeclaration name="something" refkey={rk6} />
+          <py.VariableDeclaration name="something_else" refkey={rk5} />
+        </py.SourceFile>
+        <py.SourceFile path="test_3.py">
+          <py.VariableDeclaration name="conflict" refkey={rk3} />
+          <py.VariableDeclaration name="something" refkey={rk7} />
+        </py.SourceFile>
+        <py.SourceFile path="test.py">
+          <py.StatementList>
+            <py.VariableDeclaration name="one" initializer={rk1} />
+            <py.VariableDeclaration name="two" initializer={rk2} />
+            <py.VariableDeclaration name="three" initializer={rk3} />
+            <py.VariableDeclaration name="something_else" initializer={rk4} />
+            <py.VariableDeclaration
+              name="something_else_two"
+              initializer={rk5}
+            />
+            <py.VariableDeclaration name="something" initializer={rk6} />
+            <py.VariableDeclaration name="something_two" initializer={rk7} />
+          </py.StatementList>
+        </py.SourceFile>
+      </TestOutputDirectory>,
+    ).toRenderTo({
       "test.py": `
-        from test_1 import conflict
-        from test_1 import something_else as something_else_2_test_1
-        from test_2 import conflict as conflict_2_test_2
-        from test_2 import something as something_2_test_2
-        from test_2 import something_else as something_else_3_test_2
-        from test_3 import conflict as conflict_3_test_3
-        from test_3 import something as something_3_test_3
-
-        one = conflict
-        two = conflict_2_test_2
-        three = conflict_3_test_3
-        something_else = something_else_2_test_1
-        something_else_two = something_else_3_test_2
-        something = something_2_test_2
-        something_two = something_3_test_3
-      `,
+          from test_1 import conflict
+          from test_1 import something_else as something_else_2_test_1
+          from test_2 import conflict as conflict_2_test_2
+          from test_2 import something as something_2_test_2
+          from test_2 import something_else as something_else_3_test_2
+          from test_3 import conflict as conflict_3_test_3
+          from test_3 import something as something_3_test_3
+
+          one = conflict
+          two = conflict_2_test_2
+          three = conflict_3_test_3
+          something_else = something_else_2_test_1
+          something_else_two = something_else_3_test_2
+          something = something_2_test_2
+          something_two = something_3_test_3
+        `,
+      "test_1.py": `conflict = None
+
+something_else = None`,
+      "test_2.py": `conflict = None
+
+something = None
+
+something_else = None`,
+      "test_3.py": `conflict = None
+
+something = None`,
     });
   });
 });