From 41383dc86209a6a2dc9d360fc56d988642eb983f Mon Sep 17 00:00:00 2001
From: Mohit Tejani <mohit.tejani@pubnub.com>
Date: Thu, 11 Jan 2024 14:27:16 +0530
Subject: [PATCH] updated tests

---
 test/contract/definitions/event-engine.ts       | 17 +++++++++--------
 .../contract/steps/cryptoModule/cryptoModule.ts |  1 +
 2 files changed, 10 insertions(+), 8 deletions(-)

diff --git a/test/contract/definitions/event-engine.ts b/test/contract/definitions/event-engine.ts
index f64985165..34eed878f 100644
--- a/test/contract/definitions/event-engine.ts
+++ b/test/contract/definitions/event-engine.ts
@@ -1,7 +1,7 @@
 import { after, binding, given, then, when } from 'cucumber-tsflow';
 import { DemoKeyset } from '../shared/keysets';
 import { PubNub, PubNubManager } from '../shared/pubnub';
-import type { MessageEvent, StatusEvent } from 'pubnub';
+import type { MessageEvent, PresenceEvent, StatusEvent } from 'pubnub';
 import type { Change } from '../../../src/event-engine/core/change';
 import { DataTable } from '@cucumber/cucumber';
 import { expect } from 'chai';
@@ -32,6 +32,7 @@ class EventEngineSteps {
 
   private messagePromise?: Promise<MessageEvent>;
   private statusPromise?: Promise<StatusEvent>;
+  private presencePromise?: Promise<PresenceEvent>;
   private changelog: Change<any, any>[] = [];
   private configuration: any = {};
 
@@ -85,10 +86,10 @@ class EventEngineSteps {
     });
 
     this.statusPromise = new Promise<StatusEvent>((resolveStatus) => {
-      this.messagePromise = new Promise<MessageEvent>((resolveMessage) => {
+      this.presencePromise = new Promise<PresenceEvent>((resolvePresence) => {
         this.pubnub?.addListener({
-          message(messageEvent) {
-            resolveMessage(messageEvent);
+          presence(presenceEvent) {
+            resolvePresence(presenceEvent);
           },
           status(statusEvent) {
             resolveStatus(statusEvent);
@@ -100,9 +101,9 @@ class EventEngineSteps {
     });
   }
 
-  @then('I wait for getting Presence joined events')
+  @then('I wait for getting Presence joined events', undefined, 10000)
   async thenPresenceJoinEvent() {
-    const status = await this.messagePromise;
+    const status = await this.presencePromise;
   }
 
   @then('I wait {string} seconds')
@@ -208,14 +209,14 @@ class EventEngineSteps {
     const timetoken = await this.pubnub?.publish({ channel: 'test', message: { content: 'Hello world!' } });
   }
 
-  @then('I receive an error in my subscribe response')
+  @then('I receive an error in my subscribe response', undefined, 10000)
   async thenIReceiveError() {
     const status = await this.statusPromise;
 
     expect(status?.category).to.equal('PNConnectionErrorCategory');
   }
 
-  @then('I receive the message in my subscribe response')
+  @then('I receive the message in my subscribe response', undefined, 10000)
   async receiveMessage() {
     const message = await this.messagePromise;
   }
diff --git a/test/contract/steps/cryptoModule/cryptoModule.ts b/test/contract/steps/cryptoModule/cryptoModule.ts
index c7785b85d..956aeb014 100644
--- a/test/contract/steps/cryptoModule/cryptoModule.ts
+++ b/test/contract/steps/cryptoModule/cryptoModule.ts
@@ -169,6 +169,7 @@ Then('Successfully decrypt an encrypted file with legacy code', async function (
     fileStream.pipe(outputStream);
     outputStream.on('end', () => {
       const actualFileBuffer = fs.readFileSync(tempFilePath);
+      //@ts-ignore
       expect(actualFileBuffer).to.equalBytes(this.fileDataBuffer);
       fs.unlink(tempFilePath, () => {});
     });