test: adding new test

src/core/generate.test.ts

@@ -762,6 +762,37 @@ describe("generate", () => {
       `);
       });
     });
+
+    describe("import as array of union of imported types", () => {
+      const sourceText = `
+      import { Hero } from "module"
+    
+      import { Villain } from "module"
+    
+      export type Person = (Hero | Villain)[]
+      `;
+
+      const { getZodSchemasFile } = generate({
+        sourceText,
+      });
+
+      it("should generate the zod schemas", () => {
+        expect(getZodSchemasFile("./source")).toMatchInlineSnapshot(`
+      "// Generated by ts-to-zod
+      import { z } from \\"zod\\";
+
+      import { Hero } from \\"module\\";
+      import { Villain } from \\"module\\";
+
+      const heroSchema = z.instanceof(Hero);
+
+      const villainSchema = z.instanceof(Villain);
+
+      export const personSchema = z.array(z.union([heroSchema, villainSchema]));
+      "
+      `);
+      });
+    });
   });
 
   describe("with input/output mappings to manage imports", () => {