From c3b8d5b4cf2c3de51a9a43805c0d5f474de96671 Mon Sep 17 00:00:00 2001
From: karak <karak97@gmail.com>
Date: Sat, 11 Nov 2017 00:59:21 +0900
Subject: [PATCH] Fixed return types of on(), off().

The methods on, off returns this, Benchmark or Benchmark.Suite respectively.
---
 types/benchmark/benchmark-tests.ts | 21 +++++++++++++++++++++
 types/benchmark/index.d.ts         |  8 ++++----
 2 files changed, 25 insertions(+), 4 deletions(-)

diff --git a/types/benchmark/benchmark-tests.ts b/types/benchmark/benchmark-tests.ts
index 8fbe2b1d92..cb02940b37 100644
--- a/types/benchmark/benchmark-tests.ts
+++ b/types/benchmark/benchmark-tests.ts
@@ -229,6 +229,27 @@ suite.add({
   'onComplete': onComplete
 });
 
+// unregister a listener for an event type
+suite.off('cycle', listener) as Benchmark.Suite;
+
+// unregister a listener for multiple event types
+suite.off('start cycle', listener) as Benchmark.Suite;
+
+// unregister all listeners for an event type
+suite.off('cycle') as Benchmark.Suite;
+
+// unregister all listeners for multiple event types
+suite.off('start cycle complete') as Benchmark.Suite;
+
+// unregister all listeners for all event types
+suite.off() as Benchmark.Suite;
+
+// register a listener for an event type
+suite.on('cycle', listener) as Benchmark.Suite;
+
+// register a listener for multiple event types
+suite.on('start cycle', listener) as Benchmark.Suite;
+
 // basic usage
 suite.run();
 
diff --git a/types/benchmark/index.d.ts b/types/benchmark/index.d.ts
index 01bb379a48..d8c72aef68 100644
--- a/types/benchmark/index.d.ts
+++ b/types/benchmark/index.d.ts
@@ -170,10 +170,10 @@ declare namespace Benchmark {
         join(separator?: string): string;
         listeners(type: string): Function[];
         map(callback: Function): any[];
-        off(type?: string, callback?: Function): Benchmark;
-        off(types: string[]): Benchmark;
-        on(type?: string, callback?: Function): Benchmark;
-        on(types: string[]): Benchmark;
+        off(type?: string, callback?: Function): Suite;
+        off(types: string[]): Suite;
+        on(type?: string, callback?: Function): Suite;
+        on(types: string[]): Suite;
         pluck(property: string): any[];
         pop(): Function;
         push(benchmark: Benchmark): number;