test: zod union type

packages/astro/test/content-collections.test.js

@@ -103,6 +103,25 @@ describe('Content Collections', () => {
 				const slugs = json.withSlugConfig.map((item) => item.slug);
 				expect(slugs).to.deep.equal(['fancy-one.md', 'excellent-three.md', 'interesting-two.md']);
 			});
+
+			it('Returns `with union schema` collection', async () => {
+				expect(json).to.haveOwnProperty('withUnionSchema');
+				expect(Array.isArray(json.withUnionSchema)).to.equal(true);
+
+				const post = json.withUnionSchema.find((item) => item.id === 'post.md');
+				expect(post).to.not.be.undefined;
+				expect(post.data).to.deep.equal({
+					type: 'post',
+					title: 'My Post',
+					description: 'This is my post',
+				});
+				const newsletter = json.withUnionSchema.find((item) => item.id === 'newsletter.md');
+				expect(newsletter).to.not.be.undefined;
+				expect(newsletter.data).to.deep.equal({
+					type: 'newsletter',
+					subject: 'My Newsletter',
+				});
+			});
 		});
 
 		describe('Entry', () => {
@@ -130,6 +149,16 @@ describe('Content Collections', () => {
 				expect(json).to.haveOwnProperty('twoWithSlugConfig');
 				expect(json.twoWithSlugConfig.slug).to.equal('interesting-two.md');
 			});
+
+			it('Returns `with union schema` collection entry', async () => {
+				expect(json).to.haveOwnProperty('postWithUnionSchema');
+				expect(json.postWithUnionSchema.id).to.equal('post.md');
+				expect(json.postWithUnionSchema.data).to.deep.equal({
+					type: 'post',
+					title: 'My Post',
+					description: 'This is my post',
+				});
+			});
 		});
 	});
 

packages/astro/test/fixtures/content-collections/src/content/config.ts

@@ -2,11 +2,12 @@ import { z, defineCollection } from 'astro:content';
 
 const withSlugConfig = defineCollection({
 	slug({ id, data }) {
+		console.log({id, data})
 		return `${data.prefix}-${id}`;
 	},
 	schema: z.object({
 		prefix: z.string(),
-	})
+	}),
 });
 
 const withSchemaConfig = defineCollection({
@@ -18,7 +19,22 @@ const withSchemaConfig = defineCollection({
 	})
 });
 
+const withUnionSchema = defineCollection({
+	schema: z.discriminatedUnion('type', [
+		z.object({
+			type: z.literal('post'),
+			title: z.string(),
+			description: z.string(),
+		}),
+		z.object({
+			type: z.literal('newsletter'),
+			subject: z.string(),
+		}),
+	]),
+});
+
 export const collections = {
 	'with-slug-config': withSlugConfig,
 	'with-schema-config': withSchemaConfig,
+	'with-union-schema': withUnionSchema,
 }

packages/astro/test/fixtures/content-collections/src/content/with-union-schema/newsletter.md

@@ -0,0 +1,6 @@
+---
+type: newsletter
+subject: My Newsletter
+---
+
+# It's a newsletter!

packages/astro/test/fixtures/content-collections/src/content/with-union-schema/post.md

@@ -0,0 +1,7 @@
+---
+type: post
+title: My Post
+description: This is my post
+---
+
+# It's a post!

packages/astro/test/fixtures/content-collections/src/pages/collections.json.js

@@ -6,7 +6,9 @@ export async function get() {
 	const withoutConfig = stripAllRenderFn(await getCollection('without-config'));
 	const withSchemaConfig = stripAllRenderFn(await getCollection('with-schema-config'));
 	const withSlugConfig = stripAllRenderFn(await getCollection('with-slug-config'));
+	const withUnionSchema = stripAllRenderFn(await getCollection('with-union-schema'));
+
 	return {
-		body: devalue.stringify({withoutConfig, withSchemaConfig, withSlugConfig}),
+		body: devalue.stringify({withoutConfig, withSchemaConfig, withSlugConfig, withUnionSchema}),
 	}
 }

packages/astro/test/fixtures/content-collections/src/pages/entries.json.js

@@ -6,7 +6,9 @@ export async function get() {
 	const columbiaWithoutConfig = stripRenderFn(await getEntry('without-config', 'columbia.md'));
 	const oneWithSchemaConfig = stripRenderFn(await getEntry('with-schema-config', 'one.md'));
 	const twoWithSlugConfig = stripRenderFn(await getEntry('with-slug-config', 'two.md'));
+	const postWithUnionSchema = stripRenderFn(await getEntry('with-union-schema', 'post.md'));
+
 	return {
-		body: devalue.stringify({columbiaWithoutConfig, oneWithSchemaConfig, twoWithSlugConfig}),
+		body: devalue.stringify({columbiaWithoutConfig, oneWithSchemaConfig, twoWithSlugConfig, postWithUnionSchema}),
 	}
 }