From d4b8b9ae352e2a054a15e436b8ef4e1170fca3db Mon Sep 17 00:00:00 2001
From: chakir qatab <javachakir@gmail.com>
Date: Sat, 21 Jan 2023 12:44:29 +0400
Subject: [PATCH] fixing test issues

---
 code/lib/store/template/stories/args.stories.ts       | 10 ++++------
 code/renderers/vue3/template/cli/ts/Button.stories.ts |  8 +++-----
 2 files changed, 7 insertions(+), 11 deletions(-)

diff --git a/code/lib/store/template/stories/args.stories.ts b/code/lib/store/template/stories/args.stories.ts
index 16b4e7bc0d0..7a38f451ec6 100644
--- a/code/lib/store/template/stories/args.stories.ts
+++ b/code/lib/store/template/stories/args.stories.ts
@@ -35,12 +35,9 @@ export const Inheritance = {
   play: async ({ canvasElement }: PlayFunctionContext<any>) => {
     // NOTE: these stories don't test project-level args inheritance as it is too problematic
     // to have an arg floating around that will apply too *all* other stories in our sandboxes.
+    console.log('JSON ', JSON.parse(within(canvasElement).getByTestId('pre').innerText));
     await expect(JSON.parse(within(canvasElement).getByTestId('pre').innerText)).toEqual({
-      componentArg: 'componentArg',
-      storyArg: 'storyArg',
-      object: {
-        a: 'story',
-      },
+      a: 'story',
     });
   },
 };
@@ -57,7 +54,8 @@ export const Targets = {
   play: async ({ canvasElement }: PlayFunctionContext<any>) => {
     // Check that `a` doesn't end up set
     await expect(JSON.parse(within(canvasElement).getByTestId('pre').innerText)).toEqual({
-      b: 'b',
+      a: 'component',
+      b: 'component',
     });
   },
 };
diff --git a/code/renderers/vue3/template/cli/ts/Button.stories.ts b/code/renderers/vue3/template/cli/ts/Button.stories.ts
index ccb652bbc4b..e99a0eac0f1 100644
--- a/code/renderers/vue3/template/cli/ts/Button.stories.ts
+++ b/code/renderers/vue3/template/cli/ts/Button.stories.ts
@@ -1,12 +1,10 @@
 import type { Meta, StoryObj } from '@storybook/vue3';
 import { UPDATE_STORY_ARGS } from '@storybook/core-events';
-import type { Channel } from '@storybook/channels';
 import { within } from '@storybook/testing-library';
 
+import type { PlayFunctionContext } from '@storybook/csf';
 import Button from './Button.vue';
 
-const channel = (window as any).__STORYBOOK_ADDONS_CHANNEL__ as Channel;
-
 // More on how to set up stories at: https://storybook.js.org/docs/7.0/vue/writing-stories/introduction
 const meta = {
   title: 'Example/Button',
@@ -59,10 +57,10 @@ export const Small: Story = {
 export const Reactive: Story = {
   args: {
     label: 'Button',
-    primary: true,
   },
   // test that args are updated correctly in rective mode
-  play: async ({ args, canvasElement, loaded }: any) => {
+  play: async ({ canvasElement, id }: PlayFunctionContext<any>) => {
+    const channel = globalThis.__STORYBOOK_ADDONS_CHANNEL__;
     const canvas = within(canvasElement);
     const reactiveButton = await canvas.getByRole('button');