feat(SHAPE-8085): fix tests

storyblok · Jan 15, 2025 · 11297f8 · 11297f8
1 parent d0e8573
commit 11297f8
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 11 deletions.
diff --git a/...teFieldPlugin/createPluginActions/createPluginMessageListener/handlePluginMessage.test.ts b/...teFieldPlugin/createPluginActions/createPluginMessageListener/handlePluginMessage.test.ts
@@ -148,7 +148,7 @@ describe('handlePluginMessage', () => {
       action: 'prompt-ai',
       uid,
       callbackId: 'test-callback-id',
-      output: 'Some output',
+      aiGeneratedText: 'Some AI generated text',
     }
 
     const callbacks = mockCallbacks()

diff --git a/...ugin/src/messaging/pluginMessage/containerToPluginMessage/PromptAIResponseMessage.test.ts b/...ugin/src/messaging/pluginMessage/containerToPluginMessage/PromptAIResponseMessage.test.ts
@@ -7,7 +7,7 @@ const stub: PromptAIResponseMessage = {
   action: 'prompt-ai',
   uid: '-preview',
   callbackId: 'test-callback-id',
-  output: 'test-output',
+  aiGeneratedText: 'test-ai-generated-text',
 }
 
 describe('PromptAIResponseMessage', function () {
@@ -30,50 +30,50 @@ describe('PromptAIResponseMessage', function () {
       ).toEqual(false)
     })
   })
-  describe('the output property', () => {
+  describe('the aiGeneratedText property', () => {
     it('cannot be undefined', () => {
       expect(
         isPromptAIMessage({
           ...stub,
-          output: undefined,
+          aiGeneratedText: undefined,
         }),
       ).toEqual(false)
     })
     it('is a string', () => {
       expect(
         isPromptAIMessage({
           ...stub,
-          output: 'any-string',
+          aiGeneratedText: 'any-string',
         }),
       ).toEqual(true)
       expect(
         isPromptAIMessage({
           ...stub,
-          output: 123,
+          aiGeneratedText: 123,
         }),
       ).toEqual(false)
       expect(
         isPromptAIMessage({
           ...stub,
-          output: null,
+          aiGeneratedText: null,
         }),
       ).toEqual(false)
       expect(
         isPromptAIMessage({
           ...stub,
-          output: [],
+          aiGeneratedText: [],
         }),
       ).toEqual(false)
       expect(
         isPromptAIMessage({
           ...stub,
-          output: {},
+          aiGeneratedText: {},
         }),
       ).toEqual(false)
       expect(
         isPromptAIMessage({
           ...stub,
-          output: false,
+          aiGeneratedText: false,
         }),
       ).toEqual(false)
     })

diff --git a/packages/lib-helpers/test/src/index.ts b/packages/lib-helpers/test/src/index.ts
@@ -115,7 +115,8 @@ const getContainer = (sendToFieldPlugin: (data: unknown) => void) => {
           action: 'prompt-ai',
           uid,
           callbackId: data.callbackId,
-          output: 'output fake for the prompt: ' + data.payload.text,
+          aiGeneratedText:
+            'Fake AI generated text for the prompt: ' + data.payload.text,
         })
       } else if (isGetUserContextMessage(data)) {
         sendToFieldPlugin({