feat: add text to link to media model

prismicio · Aug 26, 2024 · 6d730f3 · 6d730f3
1 parent 82f9b29
commit 6d730f3
Show file tree

Hide file tree

Showing 4 changed files with 42 additions and 5 deletions.
diff --git a/src/model/linkToMedia.ts b/src/model/linkToMedia.ts
@@ -3,13 +3,27 @@ import * as changeCase from "change-case";
 
 import { createFaker } from "../lib/createFaker";
 
+import { LinkText } from "./link";
 import { MockModelConfig } from "../types";
 
-export type MockLinkToMediaModelConfig = MockModelConfig;
+type MockLinkToMediaModel<Text extends boolean = boolean> =
+	prismic.CustomTypeModelLinkToMediaField & {
+		config: Text extends true
+			? {
+					text: LinkText;
+				}
+			: {
+					text?: undefined;
+				};
+	};
+
+export type MockLinkToMediaModelConfig<Text extends boolean = boolean> = {
+	text?: Text;
+} & MockModelConfig;
 
-export const linkToMedia = (
-	config: MockLinkToMediaModelConfig,
-): prismic.CustomTypeModelLinkToMediaField => {
+export const linkToMedia = <Text extends boolean = boolean>(
+	config: MockLinkToMediaModelConfig<Text>,
+): MockLinkToMediaModel<Text> => {
 	const faker = config.faker || createFaker(config.seed);
 
 	return {
@@ -18,6 +32,11 @@ export const linkToMedia = (
 			label: changeCase.capitalCase(faker.word()),
 			placeholder: changeCase.sentenceCase(faker.words(3)),
 			select: prismic.CustomTypeModelLinkSelectType.Media,
+			text: config.text
+				? {
+						type: prismic.CustomTypeModelFieldType.Text,
+					}
+				: undefined,
 		},
-	};
+	} as MockLinkToMediaModel<Text>;
 };
diff --git a/test/model-linkToMedia.test.ts b/test/model-linkToMedia.test.ts
@@ -11,3 +11,17 @@ test("creates a mock Link To Media field model", snapshotTwiceMacro, (t) =>
 test("supports number seed", snapshotTwiceMacro, () =>
 	model.linkToMedia({ seed: 1 }),
 );
+
+test("can be configured to explicitly support the text property", (t) => {
+	const actualTrue = model.linkToMedia({
+		seed: t.title,
+		text: true,
+	});
+	t.is(actualTrue.config.text.type, "Text");
+
+	const actualFalse = model.linkToMedia({
+		seed: t.title,
+		text: false,
+	});
+	t.is(actualFalse.config.text, undefined);
+});
diff --git a/test/snapshots/model-linkToMedia.test.ts.md b/test/snapshots/model-linkToMedia.test.ts.md
@@ -14,6 +14,7 @@ Generated by [AVA](https://avajs.dev).
           label: 'Pharetra',
           placeholder: 'Tempor id eu',
           select: 'media',
+          text: undefined,
         },
         type: 'Link',
       },
@@ -22,6 +23,7 @@ Generated by [AVA](https://avajs.dev).
           label: 'Pharetra',
           placeholder: 'Tempor id eu',
           select: 'media',
+          text: undefined,
         },
         type: 'Link',
       },
@@ -37,6 +39,7 @@ Generated by [AVA](https://avajs.dev).
           label: 'Erat',
           placeholder: 'Ut porttitor leo',
           select: 'media',
+          text: undefined,
         },
         type: 'Link',
       },
@@ -45,6 +48,7 @@ Generated by [AVA](https://avajs.dev).
           label: 'Erat',
           placeholder: 'Ut porttitor leo',
           select: 'media',
+          text: undefined,
         },
         type: 'Link',
       },

diff --git a/test/snapshots/model-linkToMedia.test.ts.snap b/test/snapshots/model-linkToMedia.test.ts.snap