From a23e44220acfdddb06215e4ed79f41a14b2ae4d6 Mon Sep 17 00:00:00 2001
From: Anh Thu Nguyen <anguyen@yext.com>
Date: Thu, 20 Jun 2024 14:57:03 -0400
Subject: [PATCH] added storybook stories for GDA component

J=CLIP-1226
TEST=manual

ran `npm run storybook`
---
 package-lock.json                             |  4 +-
 package.json                                  |  2 +-
 .../data/generativeDirectAnswer.ts            |  2 +-
 .../GenerativeDirectAnswer.stories.tsx        | 37 +++++++++++++++++++
 4 files changed, 41 insertions(+), 4 deletions(-)
 create mode 100644 tests/components/GenerativeDirectAnswer.stories.tsx

diff --git a/package-lock.json b/package-lock.json
index 692ba852..d350f745 100644
--- a/package-lock.json
+++ b/package-lock.json
@@ -1,12 +1,12 @@
 {
   "name": "@yext/search-ui-react",
-  "version": "1.7.0-beta",
+  "version": "1.7.0-beta.2",
   "lockfileVersion": 2,
   "requires": true,
   "packages": {
     "": {
       "name": "@yext/search-ui-react",
-      "version": "1.7.0-beta",
+      "version": "1.7.0-beta.2",
       "license": "BSD-3-Clause",
       "dependencies": {
         "@restart/ui": "^1.0.1",
diff --git a/package.json b/package.json
index ebd4542e..a50a5a76 100644
--- a/package.json
+++ b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "@yext/search-ui-react",
-  "version": "1.7.0-beta",
+  "version": "1.7.0-beta.2",
   "description": "A library of React Components for powering Yext Search integrations",
   "author": "slapshot@yext.com",
   "license": "BSD-3-Clause",
diff --git a/tests/__fixtures__/data/generativeDirectAnswer.ts b/tests/__fixtures__/data/generativeDirectAnswer.ts
index 8c2ce5eb..46a9d1da 100644
--- a/tests/__fixtures__/data/generativeDirectAnswer.ts
+++ b/tests/__fixtures__/data/generativeDirectAnswer.ts
@@ -1,7 +1,7 @@
 import { GenerativeDirectAnswerResponse } from '@yext/search-headless-react';
 
 export const generativeDirectAnswerResponse: GenerativeDirectAnswerResponse = {
-  directAnswer: 'an AI generated answer',
+  directAnswer: 'answer text',
   resultStatus: 'SUCCESS',
   citations: ['uid2','uid3']
 }
diff --git a/tests/components/GenerativeDirectAnswer.stories.tsx b/tests/components/GenerativeDirectAnswer.stories.tsx
new file mode 100644
index 00000000..a4fa2802
--- /dev/null
+++ b/tests/components/GenerativeDirectAnswer.stories.tsx
@@ -0,0 +1,37 @@
+import { Meta, StoryFn } from '@storybook/react';
+import { SearchHeadlessContext, State } from '@yext/search-headless-react';
+
+import { generateMockedHeadless } from '../__fixtures__/search-headless';
+import { GenerativeDirectAnswer, GenerativeDirectAnswerProps } from '../../src/components/GenerativeDirectAnswer';
+import { RecursivePartial } from '../__utils__/mocks';
+import { verticalResults } from '../__fixtures__/data/universalresults';
+import { generativeDirectAnswerResponse } from '../__fixtures__/data/generativeDirectAnswer';
+import React from 'react';
+
+const meta: Meta<typeof GenerativeDirectAnswer> = {
+  title: 'GenerativeDirectAnswer',
+  component: GenerativeDirectAnswer,
+  args: {
+    answerHeader: 'Answer Title',
+    citationsHeader: 'Citation Title',
+  }
+};
+export default meta;
+
+const mockedHeadlessState: RecursivePartial<State> = {
+  universal: {
+    verticals: verticalResults
+  },
+  generativeDirectAnswer: {
+    isLoading: false,
+    response: generativeDirectAnswerResponse
+  }
+};
+
+export const Primary: StoryFn<GenerativeDirectAnswerProps> = (args) => {
+  return (
+    <SearchHeadlessContext.Provider value={generateMockedHeadless(mockedHeadlessState)}>
+      <GenerativeDirectAnswer {...args} />
+    </SearchHeadlessContext.Provider>
+  );
+};