From bb27e58c3e1b51fa4a0c64d61f0ca4b41fe377e8 Mon Sep 17 00:00:00 2001
From: Adhiraj Singh <adhirajsingh1001@gmail.com>
Date: Sat, 18 Mar 2023 17:40:35 +0530
Subject: [PATCH] test: override session

---
 src/Tests/test.libsignal.ts | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

diff --git a/src/Tests/test.libsignal.ts b/src/Tests/test.libsignal.ts
index 15a475f..a71fd5c 100644
--- a/src/Tests/test.libsignal.ts
+++ b/src/Tests/test.libsignal.ts
@@ -23,6 +23,27 @@ describe('Signal Tests', () => {
 		expect(dec).toEqual(msg)
 	})
 
+	it('should correctly override a session', async() => {
+		const user1 = makeUser()
+		const user2 = makeUser()
+
+		const msg = Buffer.from('hello there!')
+
+		for(let preKeyId = 2; preKeyId <= 3;preKeyId++) {
+			await prepareForSendingMessage(user1, user2, preKeyId)
+
+			const result = await user1.repository.encryptMessage(
+				{ jid: user2.jid, data: msg }
+			)
+
+			const dec = await user2.repository.decryptMessage(
+				{ jid: user1.jid, ...result }
+			)
+
+			expect(dec).toEqual(msg)
+		}
+	})
+
 	it('should correctly encrypt/decrypt multiple messages', async() => {
 		const user1 = makeUser()
 		const user2 = makeUser()
@@ -98,8 +119,8 @@ function makeUser() {
 async function prepareForSendingMessage(
 	sender: User,
 	receiver: User,
+	preKeyId = 2
 ) {
-	const preKeyId = 2
 	const preKey = Curve.generateKeyPair()
 	await sender.repository.injectE2ESession(
 		{