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

* fix: use default passed to generate * lint * update text expectation
fabien0102 · Nov 1, 2024 · e8cc969 · e8cc969
1 parent a7055f5
commit e8cc969
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
@@ -1177,11 +1177,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
@@ -141,8 +141,7 @@ export function generate({
       );
 
       if (eligibleMapping) {
-        const schemaMethod =
-          eligibleMapping.getSchemaName || DEFAULT_GET_SCHEMA;
+        const schemaMethod = eligibleMapping.getSchemaName || getSchemaName;
 
         const identifiers = getImportIdentifiers(node);
         identifiers.forEach((i) =>