@kubb/plugin-faker 5.0.5 → 5.0.7

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([
@@ -532,11 +536,8 @@ function getScalarType(node, typeName) {
 * Determines the data type, return type, and whether it uses the type name.
 */
 function resolveFakerTypeUsage(node, typeName, canOverride) {
-	const isArray = node.type === "array";
-	const isTuple = node.type === "tuple";
 	const isScalar = SCALAR_TYPES$1.has(node.type);
 	let dataType = `Partial<${typeName}>`;
-	if (isArray || isTuple || node.type === "union" || node.type === "enum") dataType = typeName;
 	if (isScalar) dataType = getScalarType(node, typeName);
 	let returnType = canOverride ? typeName : null;
 	if (isScalar) returnType = getScalarType(node, typeName);
@@ -571,11 +572,12 @@ function Faker({ node, description, name, typeName, printer, canOverride }) {
 	const isObject = OBJECT_TYPES.has(node.type);
 	const isTuple = node.type === "tuple";
 	const isScalar = SCALAR_TYPES.has(node.type);
+	const isUnion = node.type === "union";
 	const useGenericOverride = canOverride && isObject;
 	const fakerTextWithOverride = (() => {
-		if (canOverride && isTuple) return `data || ${fakerText}`;
-		if (canOverride && isArray) return `[\n  ...${fakerText},\n  ...(data || [])\n]`;
-		if (canOverride && isScalar) return `data ?? ${fakerText}`;
+		if (canOverride && node.type === "tuple") return `data && data.length === ${node.items?.length ?? 0} && !data.includes(undefined) ? data : ${fakerText}`;
+		if (canOverride && isArray) return `[\n  ...${fakerText},\n  ...(data || []).filter((item) => item !== undefined),\n]`;
+		if (canOverride && (isScalar || isUnion)) return `data ?? ${fakerText}`;
 		return fakerText;
 	})();
 	const { dataType, returnType: resolvedReturnType } = resolveFakerTypeUsage(node, typeName, canOverride);
@@ -588,7 +590,7 @@ function Faker({ node, description, name, typeName, printer, canOverride }) {
 		})] });
 		const paramsSignature = declarationPrinter.print(params) ?? "";
 		const returnType = resolvedReturnType;
-		const returnExpression = node.type === "ref" && canOverride && returnType ? `${fakerTextWithOverride} as ${returnType}` : fakerTextWithOverride;
+		const returnExpression = canOverride && !!returnType && (node.type === "ref" || isArray || isTuple || isUnion) ? `${isTuple || isUnion ? `(${fakerTextWithOverride})` : fakerTextWithOverride} as ${returnType}` : fakerTextWithOverride;
 		return /* @__PURE__ */ (0, kubb_jsx_jsx_runtime.jsx)(kubb_jsx.File.Source, {
 			name,
 			isExportable: true,
@@ -1291,7 +1293,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) => ({