@alloy-js/python 0.5.0-dev.1 → 0.5.0

package/dist/test/pydanticclassdeclarations.test.js

@@ -1,14 +1,14 @@
 import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
 import { Prose, code, refkey } from "@alloy-js/core";
-import { d } from "@alloy-js/core/testing";
 import { describe, expect, it } from "vitest";
 import { pydanticModule, pydanticSettingsModule, typingModule } from "../src/builtins/python.js";
 import * as py from "../src/index.js";
-import { assertFileContents, toSourceText, toSourceTextMultiple } from "./utils.js";
+import { TestOutput, TestOutputDirectory } from "./utils.js";
 describe("PydanticClassDeclaration", () => {
   it("forwards class-level decorators above `class`", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule, typingModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "User",
@@ -25,10 +25,7 @@ describe("PydanticClassDeclaration", () => {
           }
         });
       }
-    })], {
-      externals: [pydanticModule, typingModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic import BaseModel
         from typing import final
 
@@ -37,12 +34,12 @@ describe("PydanticClassDeclaration", () => {
         class User(BaseModel):
             id: int
 
-
       `);
   });
   it("emits a pydantic model with BaseModel, fields, and Field import", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "User",
@@ -63,10 +60,7 @@ describe("PydanticClassDeclaration", () => {
           }
         });
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import Field
 
@@ -75,13 +69,13 @@ describe("PydanticClassDeclaration", () => {
             id: int
             name: str = Field(default="anon")
 
-
       `);
   });
   it("inherits from another pydantic model via bases", () => {
     const baseRef = refkey();
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.StatementList, {
           get children() {
@@ -111,10 +105,7 @@ describe("PydanticClassDeclaration", () => {
           }
         });
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic import BaseModel
 
 
@@ -124,12 +115,12 @@ describe("PydanticClassDeclaration", () => {
         class Admin(User):
             role: str
 
-
       `);
   });
   it("supports required, optional, Field default, and plain default", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "Item",
@@ -163,10 +154,7 @@ describe("PydanticClassDeclaration", () => {
           }
         });
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import Field
 
@@ -177,7 +165,6 @@ describe("PydanticClassDeclaration", () => {
             label: str = Field(default="untitled")
             qty: int = 1
 
-
       `);
   });
   it("emits class docstring", () => {
@@ -188,8 +175,9 @@ describe("PydanticClassDeclaration", () => {
         })];
       }
     });
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "RequestBody",
@@ -204,10 +192,7 @@ describe("PydanticClassDeclaration", () => {
           }
         });
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic import BaseModel
 
 
@@ -218,68 +203,70 @@ describe("PydanticClassDeclaration", () => {
 
             value: str
 
-
       `);
   });
   it("resolves refkey across files with pydantic imports", () => {
     const modelRef = refkey();
-    const res = toSourceTextMultiple([_$createComponent(py.SourceFile, {
-      path: "models.py",
+    expect(_$createComponent(TestOutputDirectory, {
+      externals: [pydanticModule],
       get children() {
-        return _$createComponent(py.PydanticClassDeclaration, {
-          name: "User",
-          refkey: modelRef,
+        return [_$createComponent(py.SourceFile, {
+          path: "models.py",
           get children() {
-            return _$createComponent(py.VariableDeclaration, {
-              instanceVariable: true,
-              omitNone: true,
-              name: "id",
-              type: "int"
+            return _$createComponent(py.PydanticClassDeclaration, {
+              name: "User",
+              refkey: modelRef,
+              get children() {
+                return _$createComponent(py.VariableDeclaration, {
+                  instanceVariable: true,
+                  omitNone: true,
+                  name: "id",
+                  type: "int"
+                });
+              }
             });
           }
-        });
-      }
-    }), _$createComponent(py.SourceFile, {
-      path: "service.py",
-      get children() {
-        return _$createComponent(py.FunctionDeclaration, {
-          name: "load_user",
-          returnType: modelRef,
-          parameters: [{
-            name: "user_id",
-            type: "int"
-          }],
-          children: "return User(id=user_id)"
-        });
+        }), _$createComponent(py.SourceFile, {
+          path: "service.py",
+          get children() {
+            return _$createComponent(py.FunctionDeclaration, {
+              name: "load_user",
+              returnType: modelRef,
+              parameters: [{
+                name: "user_id",
+                type: "int"
+              }],
+              children: "return User(id=user_id)"
+            });
+          }
+        })];
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    assertFileContents(res, {
+    })).toRenderTo({
       "models.py": `
-        from pydantic import BaseModel
+          from pydantic import BaseModel
 
 
-        class User(BaseModel):
-            id: int
+          class User(BaseModel):
+              id: int
 
-        `,
+          `,
       "service.py": `
-        from typing import TYPE_CHECKING
+          from typing import TYPE_CHECKING
 
-        if TYPE_CHECKING:
-            from models import User
+          if TYPE_CHECKING:
+              from models import User
 
 
-        def load_user(user_id: int) -> User:
-            return User(id=user_id)
+          def load_user(user_id: int) -> User:
+              return User(id=user_id)
 
-        `
+          `
     });
   });
   it("emits model_config = ConfigDict(...) from top-level config props", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "User",
@@ -296,10 +283,7 @@ describe("PydanticClassDeclaration", () => {
           }
         });
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import ConfigDict
 
@@ -308,12 +292,12 @@ describe("PydanticClassDeclaration", () => {
             model_config = ConfigDict(frozen=True, extra="forbid", validate_assignment=True)
             id: int
 
-
       `);
   });
   it("emits model_config = ConfigDict(...) from modelConfig", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "User",
@@ -332,10 +316,7 @@ describe("PydanticClassDeclaration", () => {
           }
         });
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import ConfigDict
 
@@ -344,12 +325,12 @@ describe("PydanticClassDeclaration", () => {
             model_config = ConfigDict(frozen=True, extra="forbid", validate_assignment=True)
             id: int
 
-
       `);
   });
   it("gives precedence to top-level config props over modelConfig", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "User",
@@ -361,10 +342,7 @@ describe("PydanticClassDeclaration", () => {
           extra: "forbid"
         });
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import ConfigDict
 
@@ -372,12 +350,12 @@ describe("PydanticClassDeclaration", () => {
         class User(BaseModel):
             model_config = ConfigDict(frozen=True, extra="forbid")
 
-
       `);
   });
   it("supports additional typed ConfigDict props", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "User",
@@ -390,10 +368,7 @@ describe("PydanticClassDeclaration", () => {
           valJsonBytes: "hex"
         });
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import ConfigDict
 
@@ -401,12 +376,12 @@ describe("PydanticClassDeclaration", () => {
         class User(BaseModel):
             model_config = ConfigDict(coerce_numbers_to_str=True, ser_json_bytes="base64", str_min_length=1, str_max_length=128, use_enum_values=True, val_json_bytes="hex", validate_return=True)
 
-
       `);
   });
   it("imports SecretStr when used as a field type", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "Credentials",
@@ -422,10 +397,7 @@ describe("PydanticClassDeclaration", () => {
           }
         });
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic import BaseModel
         from typing import TYPE_CHECKING
 
@@ -436,12 +408,12 @@ describe("PydanticClassDeclaration", () => {
         class Credentials(BaseModel):
             token: SecretStr
 
-
       `);
   });
   it("emits arbitrary model_config via modelConfigExpression", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "M",
@@ -450,10 +422,7 @@ describe("PydanticClassDeclaration", () => {
           }
         });
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import ConfigDict
 
@@ -461,12 +430,12 @@ describe("PydanticClassDeclaration", () => {
         class M(BaseModel):
             model_config = ConfigDict(frozen=True, extra="allow")
 
-
       `);
   });
   it("supports RootModel as explicit bases entry", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "Tags",
@@ -475,22 +444,19 @@ describe("PydanticClassDeclaration", () => {
           }
         });
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic import RootModel
 
 
         class Tags(RootModel[list[str]]):
             pass
 
-
       `);
   });
   it("places Pydantic validators above classmethod", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "User",
@@ -515,10 +481,7 @@ describe("PydanticClassDeclaration", () => {
           }
         });
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import field_validator
 
@@ -531,14 +494,14 @@ describe("PydanticClassDeclaration", () => {
                 return value.strip()
 
 
-      
       `);
   });
 });
 describe("Pydantic ecosystem emitters", () => {
   it("typing module resolves Any and similar annotations", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [typingModule],
       get children() {
         return _$createComponent(py.FunctionDeclaration, {
           name: "identity",
@@ -554,10 +517,7 @@ describe("Pydantic ecosystem emitters", () => {
           children: "return x"
         });
       }
-    })], {
-      externals: [typingModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from typing import TYPE_CHECKING
 
         if TYPE_CHECKING:
@@ -567,12 +527,12 @@ describe("Pydantic ecosystem emitters", () => {
         def identity(x: Any) -> Any:
             return x
 
-
       `);
   });
   it("pydantic.types constrains field annotations", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "Score",
@@ -588,10 +548,7 @@ describe("Pydantic ecosystem emitters", () => {
           }
         });
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic import BaseModel
         from typing import TYPE_CHECKING
 
@@ -602,53 +559,56 @@ describe("Pydantic ecosystem emitters", () => {
         class Score(BaseModel):
             points: PositiveInt
 
-
       `);
   });
   it("postponed annotations support forward references in fields", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
-      path: "models.py",
-      get futureImports() {
-        return [_$createComponent(py.FutureStatement, {
-          feature: "annotations"
-        })];
-      },
+    expect(_$createComponent(TestOutputDirectory, {
+      externals: [pydanticModule],
       get children() {
-        return _$createComponent(py.PydanticClassDeclaration, {
-          name: "Node",
-          get children() {
-            return [_$createComponent(py.VariableDeclaration, {
-              instanceVariable: true,
-              omitNone: true,
-              name: "label",
-              type: "str"
-            }), _$createComponent(py.VariableDeclaration, {
-              instanceVariable: true,
-              name: "child",
-              type: code`"Node" | None`
+        return _$createComponent(py.SourceFile, {
+          path: "models.py",
+          get futureImports() {
+            return [_$createComponent(py.FutureStatement, {
+              feature: "annotations"
             })];
+          },
+          get children() {
+            return _$createComponent(py.PydanticClassDeclaration, {
+              name: "Node",
+              get children() {
+                return [_$createComponent(py.VariableDeclaration, {
+                  instanceVariable: true,
+                  omitNone: true,
+                  name: "label",
+                  type: "str"
+                }), _$createComponent(py.VariableDeclaration, {
+                  instanceVariable: true,
+                  name: "child",
+                  type: code`"Node" | None`
+                })];
+              }
+            });
           }
         });
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
-        from __future__ import annotations
-
-        from pydantic import BaseModel
+    })).toRenderTo({
+      "models.py": `
+          from __future__ import annotations
 
+          from pydantic import BaseModel
 
-        class Node(BaseModel):
-            label: str
-            child: "Node" | None = None
 
+          class Node(BaseModel):
+              label: str
+              child: "Node" | None = None
 
-      `);
+        `
+    });
   });
   it("model_config can use pydantic.alias_generators", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "M",
@@ -657,10 +617,7 @@ describe("Pydantic ecosystem emitters", () => {
           }
         });
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import ConfigDict
         from pydantic.alias_generators import to_camel
@@ -669,12 +626,12 @@ describe("Pydantic ecosystem emitters", () => {
         class M(BaseModel):
             model_config = ConfigDict(alias_generator=to_camel)
 
-
       `);
   });
   it("pydantic_settings exposes BaseSettings", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "config.py",
+      externals: [pydanticSettingsModule],
       get children() {
         return _$createComponent(py.ClassDeclaration, {
           name: "AppSettings",
@@ -683,22 +640,19 @@ describe("Pydantic ecosystem emitters", () => {
           }
         });
       }
-    })], {
-      externals: [pydanticSettingsModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic_settings import BaseSettings
 
 
         class AppSettings(BaseSettings):
             pass
 
-
       `);
   });
   it("emits @computed_field above @property via PropertyDeclaration", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "Square",
@@ -719,10 +673,7 @@ describe("Pydantic ecosystem emitters", () => {
           }
         });
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import computed_field
 
@@ -735,12 +686,12 @@ describe("Pydantic ecosystem emitters", () => {
                 return self.width ** 2
 
 
-      
       `);
   });
   it("emits computed_field on an instance method", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "Square",
@@ -761,10 +712,7 @@ describe("Pydantic ecosystem emitters", () => {
           }
         });
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import computed_field
 
@@ -776,12 +724,12 @@ describe("Pydantic ecosystem emitters", () => {
                 return self.width ** 2
 
 
-      
       `);
   });
   it("emits model_validator above classmethod", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "Bag",
@@ -806,10 +754,7 @@ describe("Pydantic ecosystem emitters", () => {
           }
         });
       }
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import model_validator
 
@@ -822,7 +767,6 @@ describe("Pydantic ecosystem emitters", () => {
                 return data
 
 
-      
       `);
   });
 });