update snapshot

storybookjs · Oct 28, 2017 · dd4857f · dd4857f
1 parent c753d74
commit dd4857f
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 5 deletions.
diff --git a/lib/codemod/src/transforms/__testfixtures__/update-addon-info/update-addon-info.input.js b/lib/codemod/src/transforms/__testfixtures__/update-addon-info/update-addon-info.input.js
@@ -195,5 +195,4 @@ storiesOf('shared/ProgressBar', module)
     <ProgressBar progress={number('progress', 25)}
       delay={number('delay', 500)}
     />
-
   ));
diff --git a/lib/codemod/src/transforms/__testfixtures__/update-addon-info/update-addon-info.output.js b/lib/codemod/src/transforms/__testfixtures__/update-addon-info/update-addon-info.output.js
@@ -184,3 +184,11 @@ storiesOf('Button').add('with custom styles', withInfo({
     return stylesheet
   }
 })(() => <Button label="The Button" onClick={action('onClick')} />))
+
+storiesOf('shared/ProgressBar', module)
+  .addDecorator(withKnobs)
+  .add('default style', withInfo('default style')(() => (
+    <ProgressBar progress={number('progress', 25)}
+      delay={number('delay', 500)}
+    />
+  )));
diff --git a/lib/codemod/src/transforms/update-addon-info.js b/lib/codemod/src/transforms/update-addon-info.js
@@ -66,10 +66,9 @@ export default function transformer(file, api) {
     node.callee.property.name = 'add';
     node.arguments = [
       args[0],
-      j.callExpression(
-        j.callExpression(j.identifier('withInfo'), getOptions(args)),
-        storyComponent
-      ),
+      j.callExpression(j.callExpression(j.identifier('withInfo'), getOptions(args)), [
+        storyComponent,
+      ]),
     ];
 
     return node;
-Original file line number
+Diff line change
@@ Expand Up / @@ -195,5 +195,4 @@ storiesOf('shared/ProgressBar', module) @@
         <ProgressBar progress={number('progress', 25)}
           delay={number('delay', 500)}
         />
       ));