From 6a3137631aa844b3cd0547f0d41646c753455343 Mon Sep 17 00:00:00 2001
From: Rob <roblourens@gmail.com>
Date: Mon, 26 Oct 2015 23:08:31 -0700
Subject: [PATCH] Fix mock helper

---
 test/testUtils.ts                      |  2 --
 test/webkit/webKitDebugAdapter.test.ts | 12 +++++++-----
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/test/testUtils.ts b/test/testUtils.ts
index b5f2881..748e89b 100644
--- a/test/testUtils.ts
+++ b/test/testUtils.ts
@@ -2,8 +2,6 @@
  * Copyright (C) Microsoft Corporation. All rights reserved.
  *--------------------------------------------------------*/
 
-import * as assert from 'assert';
-
 export function setupUnhandledRejectionListener(): void {
     process.addListener('unhandledRejection', unhandledRejectionListener);
 }
diff --git a/test/webkit/webKitDebugAdapter.test.ts b/test/webkit/webKitDebugAdapter.test.ts
index d46a23f..e38375b 100644
--- a/test/webkit/webKitDebugAdapter.test.ts
+++ b/test/webkit/webKitDebugAdapter.test.ts
@@ -87,13 +87,15 @@ class DefaultMockWebKitConnection {
 }
 
 function registerMockWebKitConnection(partialImpl?: any): void {
-    const mock = {};
-    for (let name in DefaultMockWebKitConnection) {
+    const mock = () => { };
+    Object.getOwnPropertyNames(DefaultMockWebKitConnection).forEach(name => {
         mock[name] = DefaultMockWebKitConnection[name];
-    }
+    });
 
-    for (let name in partialImpl) {
-        mock[name] = partialImpl[name];
+    if (partialImpl) {
+        Object.getOwnPropertyNames(partialImpl).forEach(name => {
+            mock[name] = partialImpl[name];
+        });
     }
 
     mockery.registerMock('./webKitConnection', { WebKitConnection: mock });