withSafeTimeout: Also provide clearTimeout

WordPress · Feb 5, 2018 · 360a51d · 360a51d
1 parent ff410da
commit 360a51d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/components/higher-order/with-safe-timeout/index.js b/components/higher-order/with-safe-timeout/index.js
@@ -22,6 +22,7 @@ function withSafeTimeout( OriginalComponent ) {
 			super( ...arguments );
 			this.timeouts = [];
 			this.setTimeout = this.setTimeout.bind( this );
+			this.clearTimeout = this.clearTimeout.bind( this );
 			this.clear = this.clear.bind( this );
 		}
 
@@ -38,6 +39,11 @@ function withSafeTimeout( OriginalComponent ) {
 			return id;
 		}
 
+		clearTimeout( id ) {
+			clearTimeout( id );
+			this.clear( id );
+		}
+
 		clear( id ) {
 			this.timeouts = this.timeouts.filter( t => t !== id );
 		}
@@ -47,6 +53,7 @@ function withSafeTimeout( OriginalComponent ) {
 				<OriginalComponent
 					{ ...this.props }
 					setTimeout={ this.setTimeout }
+					clearTimeout={ this.clearTimeout }
 				/>
 			);
 		}