@kubb/plugin-faker 5.0.4 → 5.0.6

package/dist/index.cjs

@@ -467,10 +467,14 @@ function buildResponseUnionSchema(node, resolver) {
 	}
 	return kubb_kit.ast.factory.createSchema({
 		type: "union",
-		members: responses.map((response) => kubb_kit.ast.factory.createSchema({
-			type: "ref",
-			name: resolver.response.status(node, response.statusCode)
-		}))
+		members: responses.map((response) => {
+			const schema = (response.content?.length ?? 0) === 1 ? response.content?.[0]?.schema : void 0;
+			return kubb_kit.ast.factory.createSchema({
+				type: "ref",
+				name: resolver.response.status(node, response.statusCode),
+				schema
+			});
+		})
 	});
 }
 const SCALAR_TYPES$1 = /* @__PURE__ */ new Set([
@@ -969,27 +973,23 @@ const fakerKeywordMapper = {
 	email: () => "faker.internet.email()",
 	blob: () => "faker.image.url() as unknown as Blob"
 };
-function getEnumValues(node) {
-	if (node.namedEnumValues?.length) return node.namedEnumValues.map((item) => item.value);
-	return node.enumValues ?? [];
-}
 function parseEnumValue(value) {
 	if (typeof value === "string") return stringify(value);
 	return value;
 }
 /**
 * Reads the discriminator literal off a variant, or `undefined` when it can't be determined.
+*
+* A variant carries exactly one discriminator value, so a property resolving to several literals
+* (a `ref` to the enum shared by every branch, say) identifies no branch and narrows nothing.
 */
 function getDiscriminatorValue(member, discriminatorPropertyName) {
-	const prop = kubb_kit.ast.narrowSchema(member, "object")?.properties?.find((property) => property.name === discriminatorPropertyName);
-	const enumNode = prop ? kubb_kit.ast.narrowSchema(prop.schema, "enum") : null;
-	if (enumNode) return getEnumValues(enumNode)[0];
-	const refNode = kubb_kit.ast.narrowSchema(member, "ref");
-	if (refNode?.schema) return getDiscriminatorValue(refNode.schema, discriminatorPropertyName);
-	const intersectionNode = kubb_kit.ast.narrowSchema(member, "intersection");
-	for (const intersectionMember of intersectionNode?.members ?? []) {
-		const value = getDiscriminatorValue(intersectionMember, discriminatorPropertyName);
-		if (value !== void 0) return value;
+	for (const property of kubb_kit.ast.resolveSchemaProperties({
+		node: member,
+		propertyName: discriminatorPropertyName
+	})) {
+		const values = kubb_kit.ast.getSchemaLiteralValues(property.schema);
+		if (values.length === 1) return values[0];
 	}
 }
 /**
@@ -1055,7 +1055,7 @@ const printerFaker = kubb_kit.ast.createPrinter((options) => {
 				return `${resolvedName}${this.options.nestedInUnion && (node.schema?.type === "object" || node.schema?.type === "intersection") ? "<object>" : ""}()`;
 			},
 			enum(node) {
-				return fakerKeywordMapper.enum(getEnumValues(node).map(parseEnumValue), this.options.typeName);
+				return fakerKeywordMapper.enum(kubb_kit.ast.getSchemaLiteralValues(node).map(parseEnumValue), this.options.typeName);
 			},
 			union(node) {
 				const { discriminatorPropertyName } = node;
@@ -1295,7 +1295,8 @@ const fakerGenerator = (0, kubb_kit.defineGenerator)({
 				type: "union",
 				members: variants.map((variant) => kubb_kit.ast.factory.createSchema({
 					type: "ref",
-					name: variant.name
+					name: variant.name,
+					schema: variant.schema
 				}))
 			});
 			return [...variants.map((variant) => ({