Merge pull request #4 from AlexanderMoskovkin/v0.1.1

Fix client side. Bump version.

Author: AndreyBelym

lib/client/index.js

@@ -1,14 +1,24 @@
 'use strict';
 
 (function () {
+    function getDriver (callback) {
+        var interval = window.setInterval(function () {
+            var testCafeDriver = window['%testCafeDriverInstance%'];
+
+            if (testCafeDriver) {
+                window.clearInterval(interval);
+                callback(testCafeDriver);
+            }
+        }, 50);
+    }
+
     // NOTE: enable interaction with a page when the last test is completed
-    window.setTimeout(function () {
-        const UNLOCK_PAGE_FLAG = 'testcafe-live|driver|unlock-page-flag';
+    var UNLOCK_PAGE_FLAG = 'testcafe-live|driver|unlock-page-flag';
 
-        // TestCafe > 0.18.5 required
-        const testCafeDriver = window['%testCafeDriverInstance%'];
-        const testCafeCore   = window['%testCafeCore%'];
-        const hammerhead     = window['%hammerhead%'];
+    // TestCafe > 0.18.5 required
+    getDriver(function (testCafeDriver) {
+        var testCafeCore = window['%testCafeCore%'];
+        var hammerhead   = window['%hammerhead%'];
 
         testCafeDriver.setCustomCommandHandlers('unlock-page', function () {
             testCafeCore.disableRealEventsPreventing();
@@ -18,7 +28,7 @@
             return hammerhead.Promise.resolve();
         });
 
-        const chain = testCafeDriver.contextStorage ? hammerhead.Promise.resolve() : testCafeDriver.readyPromise;
+        var chain = testCafeDriver.contextStorage ? hammerhead.Promise.resolve() : testCafeDriver.readyPromise;
 
         chain.then(function () {
             if (testCafeDriver.contextStorage.getItem(UNLOCK_PAGE_FLAG))

package.json

@@ -1,6 +1,6 @@
 {
   "name": "testcafe-live",
-  "version": "0.1.0",
+  "version": "0.1.1",
   "description": "A watcher utility for TestCafe. Watches for changes in test files and automatically runs tests when these changes occur.",
   "main": "lib/index.js",
   "bin": {