fix: add data-testid props

ccontrols · Apr 28, 2021 · c50907a · c50907a
1 parent ee3f298
commit c50907a
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 4 deletions.
diff --git a/ui/blocks/src/Stories/Stories.tsx b/ui/blocks/src/Stories/Stories.tsx
@@ -34,7 +34,12 @@ export const Stories: FC<StoriesProps> = ({ id, name, title, ...rest }) => {
     return null;
   }
   return (
-    <StoryBlockContainer story={story} title={title} {...rest}>
+    <StoryBlockContainer
+      story={story}
+      title={title}
+      data-testid="stories"
+      {...rest}
+    >
       {stories.map((id: string) => {
         const story = store.stories[id];
         return (

diff --git a/ui/blocks/src/Story/Story.tsx b/ui/blocks/src/Story/Story.tsx
@@ -47,7 +47,7 @@ export const Story: FC<StoryProps> = forwardRef(function Story(
   const story = useStory({ id, name });
   if (story && story.id && story.renderFn) {
     return (
-      <StoryBlockContainer {...rest} story={story}>
+      <StoryBlockContainer {...rest} data-testid="story" story={story}>
         <StoryRender
           ref={ref}
           story={story}

diff --git a/ui/blocks/src/StoryConfig/StoryConfig.tsx b/ui/blocks/src/StoryConfig/StoryConfig.tsx
@@ -33,7 +33,7 @@ export const StoryConfig: FC<StoryConfigProps> = ({
   const doc = useCurrentDocument();
   const docPackage = usePackage(doc?.package);
   return (
-    <StoryBlockContainer story={story} {...rest}>
+    <StoryBlockContainer data-testid="story-config" story={story} {...rest}>
       <BaseStoryConfig
         story={story}
         doc={doc}

diff --git a/ui/blocks/src/StorySource/StorySource.tsx b/ui/blocks/src/StorySource/StorySource.tsx
@@ -45,7 +45,7 @@ export const StorySource: FC<StorySourceProps> = ({
   const docPackage = usePackage(doc?.package);
 
   return (
-    <StoryBlockContainer story={story} {...rest}>
+    <StoryBlockContainer story={story} data-testid="story-source" {...rest}>
       <BaseStorySource
         story={story}
         doc={doc}