86 lines
2.2 KiB
TypeScript
86 lines
2.2 KiB
TypeScript
|
// Copyright 2021 Signal Messenger, LLC
|
||
|
|
// SPDX-License-Identifier: AGPL-3.0-only
|
||
|
|
|
||
|
|
import { assert } from 'chai';
|
||
|
|
import { v4 as uuid } from 'uuid';
|
||
|
|
|
||
|
|
import Crypto from '../../textsecure/Crypto';
|
||
|
|
import {
|
||
|
|
arrayBufferToBase64,
|
||
|
|
base64ToArrayBuffer,
|
||
|
|
stringFromBytes,
|
||
|
|
trimForDisplay,
|
||
|
|
} from '../../Crypto';
|
||
|
|
import { encryptProfileData } from '../../util/encryptProfileData';
|
||
|
|
|
||
|
|
describe('encryptProfileData', () => {
|
||
|
|
it('encrypts and decrypts properly', async () => {
|
||
|
|
const keyBuffer = Crypto.getRandomBytes(32);
|
||
|
|
const conversation = {
|
||
|
|
aboutEmoji: '🐢',
|
||
|
|
aboutText: 'I like turtles',
|
||
|
|
familyName: 'Kid',
|
||
|
|
firstName: 'Zombie',
|
||
|
|
profileKey: arrayBufferToBase64(keyBuffer),
|
||
|
|
uuid: uuid(),
|
||
|
|
|
||
|
|
// To satisfy TS
|
||
|
|
acceptedMessageRequest: true,
|
||
|
|
id: '',
|
||
|
|
isMe: true,
|
||
|
|
sharedGroupNames: [],
|
||
|
|
title: '',
|
||
|
|
type: 'direct' as const,
|
||
|
|
};
|
||
|
|
|
||
|
|
const [encrypted] = await encryptProfileData(conversation);
|
||
|
|
|
||
|
|
assert.isDefined(encrypted.version);
|
||
|
|
assert.isDefined(encrypted.name);
|
||
|
|
assert.isDefined(encrypted.commitment);
|
||
|
|
|
||
|
|
const decryptedProfileNameBytes = await Crypto.decryptProfileName(
|
||
|
|
encrypted.name,
|
||
|
|
keyBuffer
|
||
|
|
);
|
||
|
|
assert.equal(
|
||
|
|
stringFromBytes(decryptedProfileNameBytes.given),
|
||
|
|
conversation.firstName
|
||
|
|
);
|
||
|
|
if (decryptedProfileNameBytes.family) {
|
||
|
|
assert.equal(
|
||
|
|
stringFromBytes(decryptedProfileNameBytes.family),
|
||
|
|
conversation.familyName
|
||
|
|
);
|
||
|
|
} else {
|
||
|
|
assert.isDefined(decryptedProfileNameBytes.family);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (encrypted.about) {
|
||
|
|
const decryptedAboutBytes = await Crypto.decryptProfile(
|
||
|
|
base64ToArrayBuffer(encrypted.about),
|
||
|
|
keyBuffer
|
||
|
|
);
|
||
|
|
assert.equal(
|
||
|
|
stringFromBytes(trimForDisplay(decryptedAboutBytes)),
|
||
|
|
conversation.aboutText
|
||
|
|
);
|
||
|
|
} else {
|
||
|
|
assert.isDefined(encrypted.about);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (encrypted.aboutEmoji) {
|
||
|
|
const decryptedAboutEmojiBytes = await Crypto.decryptProfile(
|
||
|
|
base64ToArrayBuffer(encrypted.aboutEmoji),
|
||
|
|
keyBuffer
|
||
|
|
);
|
||
|
|
assert.equal(
|
||
|
|
stringFromBytes(trimForDisplay(decryptedAboutEmojiBytes)),
|
||
|
|
conversation.aboutEmoji
|
||
|
|
);
|
||
|
|
} else {
|
||
|
|
assert.isDefined(encrypted.aboutEmoji);
|
||
|
|
}
|
||
|
|
});
|
||
|
|
});
|