bitwarden/jslib
1
0
Fork 0
mirror of https://github.com/bitwarden/jslib synced 2025-12-28 06:03:14 +00:00
Code Issues Releases Wiki Activity

Showcase how jest-mock-extended works

Browse Source
This commit is contained in:
Hinton
2022-03-31 10:54:44 +02:00
parent 47473b7ede
commit ab6c53542d
3 changed files with 19044 additions and 5667 deletions
Download Patch File Download Diff File Expand all files Collapse all files

54
common/spec/services/cipher.service.spec.ts
View File

@@ -1,6 +1,5 @@
import { Arg, Substitute, SubstituteOf } from "@fluffy-spoon/substitute";
import { any, mock, mockDeep, MockProxy } from "jest-mock-extended";
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";
@@ -13,35 +12,37 @@ 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 { ApiService } from "jslib-common/services/api.service";
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 stateService: SubstituteOf<StateService>;
let settingsService: SubstituteOf<SettingsService>;
let apiService: SubstituteOf<ApiService>;
let fileUploadService: SubstituteOf<FileUploadService>;
let i18nService: SubstituteOf<I18nService>;
let searchService: SubstituteOf<SearchService>;
let logService: SubstituteOf<LogService>;
let cryptoService: MockProxy<CryptoService>;
let stateService: MockProxy<StateService>;
let settingsService: MockProxy<SettingsService>;
let apiService: MockProxy<ApiService>;
let fileUploadService: MockProxy<FileUploadService>;
let i18nService: MockProxy<I18nService>;
let searchService: MockProxy<SearchService>;
let logService: MockProxy<LogService>;
let cipherService: CipherService;
beforeEach(() => {
cryptoService = Substitute.for<CryptoService>();
stateService = Substitute.for<StateService>();
settingsService = Substitute.for<SettingsService>();
apiService = Substitute.for<ApiService>();
fileUploadService = Substitute.for<FileUploadService>();
i18nService = Substitute.for<I18nService>();
searchService = Substitute.for<SearchService>();
logService = Substitute.for<LogService>();
cryptoService = mock<CryptoService>();
stateService = mock<StateService>();
settingsService = mock<SettingsService>();
apiService = mock<ApiService>();
fileUploadService = mock<FileUploadService>();
i18nService = mock<I18nService>();
searchService = mock<SearchService>();
logService = mock<LogService>();
cryptoService.encryptToBytes(Arg.any(), Arg.any()).resolves(ENCRYPTED_BYTES);
cryptoService.encrypt(Arg.any(), Arg.any()).resolves(new EncString(ENCRYPTED_TEXT));
cryptoService.makeEncKey.mockResolvedValue([jest.fn(), jest.fn()] as any);
cryptoService.encryptToBytes.mockResolvedValue(ENCRYPTED_BYTES);
cryptoService.encrypt.mockResolvedValue(new EncString(ENCRYPTED_TEXT));
cipherService = new CipherService(
cryptoService,
@@ -58,12 +59,17 @@ describe("Cipher Service", () => {
it("attachments upload encrypted file contents", async () => {
const fileName = "filename";
const fileData = new Uint8Array(10).buffer;
cryptoService.getOrgKey(Arg.any()).resolves(new SymmetricCryptoKey(new Uint8Array(32).buffer));
cryptoService.getOrgKey.mockResolvedValue(new SymmetricCryptoKey(new Uint8Array(32).buffer));
apiService.postCipherAttachment.mockResolvedValue("123" as any);
await cipherService.saveAttachmentRawWithServer(new Cipher(), fileName, fileData);
fileUploadService
.received(1)
.uploadCipherAttachment(Arg.any(), Arg.any(), new EncString(ENCRYPTED_TEXT), ENCRYPTED_BYTES);
expect(fileUploadService.uploadCipherAttachment).toHaveBeenCalledTimes(1);
expect(fileUploadService.uploadCipherAttachment).toHaveBeenCalledWith(
any(),
any(),
new EncString(ENCRYPTED_TEXT),
ENCRYPTED_BYTES
);
});
});