From e2a42d1b61d58141f26a5801a78f1e73d6ae2ad5 Mon Sep 17 00:00:00 2001
From: Audric Ackermann <audric@loki.network>
Date: Wed, 29 Jul 2020 11:25:36 +1000
Subject: [PATCH] add stubbing of messageQueue in window from ts tests

---
 preload.js                           |  3 +++
 ts/test/test-utils/utils/stubbing.ts | 23 ++++++++++++++++++++++-
 2 files changed, 25 insertions(+), 1 deletion(-)

diff --git a/preload.js b/preload.js
index acfec5268..38ed1056b 100644
--- a/preload.js
+++ b/preload.js
@@ -492,6 +492,9 @@ if (config.environment.includes('test-integration')) {
     debugMessageLogs: true,
     enableSenderKeys: true,
   };
+  /* eslint-disable global-require, import/no-extraneous-dependencies */
+  window.sinon = require('sinon');
+  /* eslint-enable global-require, import/no-extraneous-dependencies */
 }
 
 // Blocking
diff --git a/ts/test/test-utils/utils/stubbing.ts b/ts/test/test-utils/utils/stubbing.ts
index 9391204d9..ce004f8a2 100644
--- a/ts/test/test-utils/utils/stubbing.ts
+++ b/ts/test/test-utils/utils/stubbing.ts
@@ -1,6 +1,6 @@
 import * as sinon from 'sinon';
-import * as crypto from 'crypto';
 import * as DataShape from '../../../../js/modules/data';
+import { Application } from 'spectron';
 
 const globalAny: any = global;
 const sandbox = sinon.createSandbox();
@@ -54,6 +54,27 @@ export function stubWindow<K extends keyof Window>(
   };
 }
 
+export async function spyMessageQueueSend(app: Application) {
+  await app.webContents.executeJavaScript(
+    "var messageQueueSpy = sinon.spy(window.libsession.getMessageQueue(), 'send'); "
+  );
+}
+
+export async function getAllMessagesSent(app: Application) {
+  const messageQueueSpy = await app.webContents.executeJavaScript(
+    'messageQueueSpy.args;'
+  );
+  if (!messageQueueSpy) {
+    throw new Error(
+      'Be sure to call spyMessageQueueSend() on the correct app first.'
+    );
+  }
+  const messages = await app.webContents.executeJavaScript(
+    'messageQueueSpy.args'
+  );
+  return messages;
+}
+
 export function restoreStubs() {
   globalAny.window = undefined;
   sandbox.restore();