From 71c45adb98a91c3fae951bb40de07c1e23aa3de0 Mon Sep 17 00:00:00 2001
From: Nicholas Zuber <zuber.nicholas@gmail.com>
Date: Wed, 7 Nov 2018 23:58:39 -0500
Subject: [PATCH] Improve tooltip being unmounted

---
 src/enhance/index.js | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/src/enhance/index.js b/src/enhance/index.js
index 88181cf..d933954 100644
--- a/src/enhance/index.js
+++ b/src/enhance/index.js
@@ -53,7 +53,7 @@ class Tooltip extends React.Component {
     }, 500);
   }
 
-  onMouseLeave = () => {
+  removeTooltip = () => {
     clearTimeout(this.timeout);
     const tooltipElement = this.getTooltipElement();
     if (tooltipElement) {
@@ -64,16 +64,18 @@ class Tooltip extends React.Component {
   render () {
     return this.props.children({
       onMouseEnter: this.onMouseEnter,
-      onMouseLeave: this.onMouseLeave,
+      onMouseLeave: this.removeTooltip,
+      onMouseDown: this.removeTooltip,
     });
   }
 }
 
 export const withTooltip = WrappedComponent => ({tooltip, tooltipOffsetX, tooltipOffsetY, ...props}) => (
   <Tooltip
-      message={tooltip}
-      tooltipOffsetX={tooltipOffsetX}
-      tooltipOffsetY={tooltipOffsetY}>
+    message={tooltip}
+    tooltipOffsetX={tooltipOffsetX}
+    tooltipOffsetY={tooltipOffsetY}
+  >
     {mouseEvents => tooltip ? (
       <WrappedComponent {...props} {...mouseEvents} />
     ) : (