From 145e8af07a0120fb82c83b493d5adbde895c6cb7 Mon Sep 17 00:00:00 2001
From: Honza Dvorsky <honza@swiftkey.com>
Date: Thu, 2 Apr 2015 20:13:08 +0200
Subject: [PATCH] on-finished fixed tests

---
 on-finished/on-finished-tests.ts | 11 +++++++----
 on-finished/on-finished.d.ts     |  8 +++++---
 2 files changed, 12 insertions(+), 7 deletions(-)

diff --git a/on-finished/on-finished-tests.ts b/on-finished/on-finished-tests.ts
index 8e2b4734b..15a757718 100644
--- a/on-finished/on-finished-tests.ts
+++ b/on-finished/on-finished-tests.ts
@@ -1,12 +1,15 @@
-/// <reference path="on-finished.d.ts" />
+/// <reference path="../node/node.d.ts" />
+/// <reference path="./on-finished.d.ts" />
+
+import events = require('events');
 
 function test_finished() {
 
-	var msg = {};
+	var e = new events.EventEmitter();
 
-	var ret = onFinished(msg, () => {
+	var ret: NodeJS.EventEmitter = OnFinished.onFinished(e, () => {
 		//callback
 		});
 
-	var finished: boolean = isFinished(msg);
+	var finished: boolean = OnFinished.isFinished(e);
 }
diff --git a/on-finished/on-finished.d.ts b/on-finished/on-finished.d.ts
index b3916f513..e4e35c995 100644
--- a/on-finished/on-finished.d.ts
+++ b/on-finished/on-finished.d.ts
@@ -3,8 +3,10 @@
 // Definitions by: Honza Dvorsky <http://github.com/czechboy0>
 // Definitions: https://github.com/borisyankov/DefinitelyTyped
 
-declare module 'on-finished' {
+/// <reference path="../node/node.d.ts" />
 
-    export function onFinished(msg:any, listener:Function):any;
-    export function isFinished(msg:any):boolean;
+declare module OnFinished {
+
+    export function onFinished(msg:NodeJS.EventEmitter, listener:Function): NodeJS.EventEmitter;
+    export function isFinished(msg:NodeJS.EventEmitter):boolean;
 }