From 0516e69ff7a906e5a82944017f7d41e389b23a0f Mon Sep 17 00:00:00 2001
From: Beaudan <beau@loki.network>
Date: Mon, 18 Feb 2019 15:53:18 +1100
Subject: [PATCH] Fix tests

---
 js/modules/loki_p2p_api.js             |  4 ++++
 libloki/test/node/loki_p2p_api_test.js | 10 +++++-----
 2 files changed, 9 insertions(+), 5 deletions(-)

diff --git a/js/modules/loki_p2p_api.js b/js/modules/loki_p2p_api.js
index cb22bbf6b..6475ccf9b 100644
--- a/js/modules/loki_p2p_api.js
+++ b/js/modules/loki_p2p_api.js
@@ -34,6 +34,10 @@ class LokiP2pAPI extends EventEmitter {
         pingTimer: null,
         isOnline: false,
       };
+      if (isPing) {
+        this.setContactOnline(pubKey);
+        return;
+      }
       // Try ping
       this.pingContact(pubKey);
       return;
diff --git a/libloki/test/node/loki_p2p_api_test.js b/libloki/test/node/loki_p2p_api_test.js
index 5875927ea..5e52ba019 100644
--- a/libloki/test/node/loki_p2p_api_test.js
+++ b/libloki/test/node/loki_p2p_api_test.js
@@ -1,7 +1,7 @@
 const { assert } = require('chai');
 const LokiP2pAPI = require('../../../js/modules/loki_p2p_api');
 
-describe('LocalLokiServer', () => {
+describe('LokiP2pAPI', () => {
   const usedKey = 'aPubKey';
   const usedAddress = 'anAddress';
   const usedPort = 'aPort';
@@ -64,16 +64,16 @@ describe('LocalLokiServer', () => {
       usedKey,
       usedAddress,
       usedPort,
-      true
+      false
     );
-    assert.isTrue(this.lokiP2pAPI.isOnline(usedKey));
+    assert.isFalse(this.lokiP2pAPI.isOnline(usedKey));
     this.lokiP2pAPI.updateContactP2pDetails(
       usedKey,
       usedAddress,
       usedPort,
-      false
+      true
     );
-    assert.isFalse(this.lokiP2pAPI.isOnline(usedKey));
+    assert.isTrue(this.lokiP2pAPI.isOnline(usedKey));
   });
 
   it('Should set a contact as offline', () => {