fix!: use default passed to generate (#215)

* fix: use default passed to generate * lint * update text expectation
fabien0102 · Nov 25, 2024 · 09d6ebf · 09d6ebf
1 parent 7db07d5
commit 09d6ebf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/src/core/generate.test.ts b/src/core/generate.test.ts
@@ -1305,11 +1305,11 @@ describe("generate", () => {
         "// Generated by ts-to-zod
         import { z } from "zod";
 
-        import { heroSchema } from "${output}";
+        import { zHero } from "${output}";
 
         export const zPerson = z.object({
             id: z.number(),
-            hero: heroSchema
+            hero: zHero
         });
         "
       `);

diff --git a/src/core/generate.ts b/src/core/generate.ts
@@ -143,8 +143,7 @@ export function generate({
       );
 
       if (eligibleMapping) {
-        const schemaMethod =
-          eligibleMapping.getSchemaName || DEFAULT_GET_SCHEMA;
+        const schemaMethod = eligibleMapping.getSchemaName || getSchemaName;
 
         identifiers.forEach(({ name }) =>
           importedZodNamesAvailable.set(name, schemaMethod(name))