From dadb3b32f771533dcad0412bdf6f2907dde56a89 Mon Sep 17 00:00:00 2001
From: Alexey Svetliakov <asvetliakov@teledini.com>
Date: Mon, 20 Mar 2017 05:01:19 +0100
Subject: [PATCH] Added overload

---
 enzyme/index.d.ts | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/enzyme/index.d.ts b/enzyme/index.d.ts
index 184b20ec57..86c481ceb4 100644
--- a/enzyme/index.d.ts
+++ b/enzyme/index.d.ts
@@ -541,14 +541,16 @@ export interface MountRendererProps {
  * @param node
  * @param [options]
  */
-export function shallow<P, S>(node: ReactElement<P>, options?: ShallowRendererProps): ShallowWrapper<P, any>;
+export function shallow<P>(node: ReactElement<P>, options?: ShallowRendererProps): ShallowWrapper<P, any>;
+export function shallow<P, S>(node: ReactElement<P>, options?: ShallowRendererProps): ShallowWrapper<P, S>;
 
 /**
  * Mounts and renders a react component into the document and provides a testing wrapper around it.
  * @param node
  * @param [options]
  */
-export function mount<P, S>(node: ReactElement<P>, options?: MountRendererProps): ReactWrapper<P, any>;
+export function mount<P>(node: ReactElement<P>, options?: MountRendererProps): ReactWrapper<P, any>;
+export function mount<P, S>(node: ReactElement<P>, options?: MountRendererProps): ReactWrapper<P, S>;
 
 /**
  * Render react components to static HTML and analyze the resulting HTML structure.