From f45c7bc2564e2ed45966dcef249a2353126b5061 Mon Sep 17 00:00:00 2001
From: Jonathan Rainville <rainville.jonathan@gmail.com>
Date: Fri, 24 Aug 2018 13:11:06 -0400
Subject: [PATCH] use secureSend for SimpleStorage set as it can cause hangs on
 WS

---
 test_apps/test_app/test/simple_storage_spec.js | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/test_apps/test_app/test/simple_storage_spec.js b/test_apps/test_app/test/simple_storage_spec.js
index 0e2e711f..5596da69 100644
--- a/test_apps/test_app/test/simple_storage_spec.js
+++ b/test_apps/test_app/test/simple_storage_spec.js
@@ -1,6 +1,7 @@
 /*global contract, config, it, assert, web3*/
 const SimpleStorage = require('Embark/contracts/SimpleStorage');
 let accounts;
+const Utils = require('embarkjs').Utils;
 
 config({
   contracts: {
@@ -21,10 +22,16 @@ contract("SimpleStorage", function () {
     assert.strictEqual(parseInt(result, 10), 100);
   });
 
-  it("set storage value", async function () {
-    await SimpleStorage.methods.set(150).send();
-    let result = await SimpleStorage.methods.get().call();
-    assert.strictEqual(parseInt(result, 10), 150);
+  it("set storage value", function (done) {
+    // await SimpleStorage.methods.set(150).send();
+    Utils.secureSend(web3, SimpleStorage.methods.set(150), {}, false, async function(err) {
+      if (err) {
+        return done(err);
+      }
+      let result = await SimpleStorage.methods.get().call();
+      assert.strictEqual(parseInt(result, 10), 150);
+      done();
+    });
   });
 
   it("should set defaultAccount", async function () {