mirror of
https://github.com/bitwarden/browser.git
synced 2024-12-02 13:23:29 +01:00
ea90aea013
* Use EncString type to enforce encryption on filename in Cipher attachment upload * Fix Cipher attachment test
62 lines
3.1 KiB
TypeScript
62 lines
3.1 KiB
TypeScript
import { Arg, Substitute, SubstituteOf } from '@fluffy-spoon/substitute';
|
|
|
|
import { ApiService } from 'jslib-common/abstractions/api.service';
|
|
import { CryptoService } from 'jslib-common/abstractions/crypto.service';
|
|
import { FileUploadService } from 'jslib-common/abstractions/fileUpload.service';
|
|
import { I18nService } from 'jslib-common/abstractions/i18n.service';
|
|
import { SearchService } from 'jslib-common/abstractions/search.service';
|
|
import { SettingsService } from 'jslib-common/abstractions/settings.service';
|
|
import { StorageService } from 'jslib-common/abstractions/storage.service';
|
|
import { UserService } from 'jslib-common/abstractions/user.service';
|
|
import { Utils } from 'jslib-common/misc/utils';
|
|
import { Cipher } from 'jslib-common/models/domain/cipher';
|
|
import { EncArrayBuffer } from 'jslib-common/models/domain/encArrayBuffer';
|
|
import { EncString } from 'jslib-common/models/domain/encString';
|
|
import { SymmetricCryptoKey } from 'jslib-common/models/domain/symmetricCryptoKey';
|
|
|
|
import { CipherService } from 'jslib-common/services/cipher.service';
|
|
|
|
const ENCRYPTED_TEXT = 'This data has been encrypted';
|
|
const ENCRYPTED_BYTES = new EncArrayBuffer(Utils.fromUtf8ToArray(ENCRYPTED_TEXT).buffer);
|
|
|
|
describe('Cipher Service', () => {
|
|
let cryptoService: SubstituteOf<CryptoService>;
|
|
let userService: SubstituteOf<UserService>;
|
|
let settingsService: SubstituteOf<SettingsService>;
|
|
let apiService: SubstituteOf<ApiService>;
|
|
let fileUploadService: SubstituteOf<FileUploadService>;
|
|
let storageService: SubstituteOf<StorageService>;
|
|
let i18nService: SubstituteOf<I18nService>;
|
|
let searchService: SubstituteOf<SearchService>;
|
|
|
|
let cipherService: CipherService;
|
|
|
|
beforeEach(() => {
|
|
cryptoService = Substitute.for<CryptoService>();
|
|
userService = Substitute.for<UserService>();
|
|
settingsService = Substitute.for<SettingsService>();
|
|
apiService = Substitute.for<ApiService>();
|
|
fileUploadService = Substitute.for<FileUploadService>();
|
|
storageService = Substitute.for<StorageService>();
|
|
i18nService = Substitute.for<I18nService>();
|
|
searchService = Substitute.for<SearchService>();
|
|
|
|
cryptoService.encryptToBytes(Arg.any(), Arg.any()).resolves(ENCRYPTED_BYTES);
|
|
cryptoService.encrypt(Arg.any(), Arg.any()).resolves(new EncString(ENCRYPTED_TEXT));
|
|
|
|
cipherService = new CipherService(cryptoService, userService, settingsService, apiService, fileUploadService,
|
|
storageService, i18nService, () => searchService);
|
|
});
|
|
|
|
it('attachments upload encrypted file contents', async () => {
|
|
const key = new SymmetricCryptoKey(new Uint8Array(32).buffer);
|
|
const fileName = 'filename';
|
|
const fileData = new Uint8Array(10).buffer;
|
|
cryptoService.getOrgKey(Arg.any()).resolves(new SymmetricCryptoKey(new Uint8Array(32).buffer));
|
|
|
|
await cipherService.saveAttachmentRawWithServer(new Cipher(), fileName, fileData);
|
|
|
|
fileUploadService.received(1).uploadCipherAttachment(Arg.any(), Arg.any(), new EncString(ENCRYPTED_TEXT), ENCRYPTED_BYTES);
|
|
});
|
|
});
|