From 3faa4f71f33d9e4780be22b76d4d2fb7b8d4a1dc Mon Sep 17 00:00:00 2001
From: Maxim Shishmarev <msgmaxim@gmail.com>
Date: Mon, 13 Jul 2020 16:38:28 +1000
Subject: [PATCH] Fix unit tests

---
 .../session/crypto/MessageEncrypter_test.ts    | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

diff --git a/ts/test/session/crypto/MessageEncrypter_test.ts b/ts/test/session/crypto/MessageEncrypter_test.ts
index ecf534690..7be40cad2 100644
--- a/ts/test/session/crypto/MessageEncrypter_test.ts
+++ b/ts/test/session/crypto/MessageEncrypter_test.ts
@@ -7,6 +7,8 @@ import { Stubs, TestUtils } from '../../test-utils';
 import { UserUtil } from '../../../util';
 import { SignalService } from '../../../protobuf';
 
+import * as Ratchet from '../../../session/medium_group/ratchet';
+
 // tslint:disable-next-line: max-func-body-length
 describe('MessageEncrypter', () => {
   const sandbox = sinon.createSandbox();
@@ -33,6 +35,7 @@ describe('MessageEncrypter', () => {
     TestUtils.stubWindow('libloki', {
       crypto: {
         FallBackSessionCipher: Stubs.FallBackSessionCipherStub,
+        encryptForPubkey: sinon.fake.returns(''),
       } as any,
     });
 
@@ -46,15 +49,22 @@ describe('MessageEncrypter', () => {
 
   describe('EncryptionType', () => {
     describe('MediumGroup', () => {
-      it('should throw an error', async () => {
+      it('should return a MEDIUM_GROUP_CIPHERTEXT envelope type', async () => {
         const data = crypto.randomBytes(10);
-        const promise = MessageEncrypter.encrypt(
+
+        sandbox.replace(
+          Ratchet,
+          'encryptWithSenderKey',
+          sinon.fake.returns({ ciphertext: '' })
+        );
+
+        const result = await MessageEncrypter.encrypt(
           TestUtils.generateFakePubKey(),
           data,
           EncryptionType.MediumGroup
         );
-        await expect(promise).to.be.rejectedWith(
-          'Encryption is not yet supported'
+        expect(result.envelopeType).to.deep.equal(
+          SignalService.Envelope.Type.MEDIUM_GROUP_CIPHERTEXT
         );
       });
     });