From f6ddc0feca5750e8796b8614bcea96a283d1ccbd Mon Sep 17 00:00:00 2001
From: Matthew Lehner <matthew@mpl.io>
Date: Mon, 15 Jul 2019 09:59:55 -0700
Subject: [PATCH] [@storybook/addon-storyshots] callable renderWithOptions
 (#36768)

---
 types/storybook__addon-storyshots/index.d.ts           |  8 +++++++-
 .../storybook__addon-storyshots-tests.ts               | 10 ++++++++++
 2 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/types/storybook__addon-storyshots/index.d.ts b/types/storybook__addon-storyshots/index.d.ts
index f7955b2b01..cb58baa59f 100644
--- a/types/storybook__addon-storyshots/index.d.ts
+++ b/types/storybook__addon-storyshots/index.d.ts
@@ -22,10 +22,16 @@ export type RenderTree = (
     options?: SnapshotOptions
 ) => undefined | void | Promise<void>;
 
-export interface SnapshotOptions {
+export interface SnapshotOptionsObject {
     createNodeMock?: (element: any) => any;
 }
 
+export interface SnapshotOptionsFn {
+    (story: StoryObject): SnapshotOptionsObject;
+}
+
+export type SnapshotOptions = SnapshotOptionsObject | SnapshotOptionsFn;
+
 export interface StoryContext {
     kind: string;
     story: string;
diff --git a/types/storybook__addon-storyshots/storybook__addon-storyshots-tests.ts b/types/storybook__addon-storyshots/storybook__addon-storyshots-tests.ts
index 9e2b44b57b..cac02b4cf7 100644
--- a/types/storybook__addon-storyshots/storybook__addon-storyshots-tests.ts
+++ b/types/storybook__addon-storyshots/storybook__addon-storyshots-tests.ts
@@ -45,6 +45,16 @@ initStoryshots({
     })
 });
 
+initStoryshots({
+    configPath: '',
+    test: renderWithOptions(story => ({
+        createNodeMock: () => {
+            story.name;
+            return undefined;
+        },
+    })),
+});
+
 initStoryshots({
     configPath: "",
     test: imageSnapshot({