databag/app/sdk/__tests__/account.tests.ts

import { AccountModule } from '../src/account';
import { NoStore } from '../src/store';
import { Crypto } from '../src/crypto';
import { ConsoleLogging } from '../src/logging';
import { defaultAccountEntity } from '../src/entities';
import { AccountStatus } from '../src/types';
import { waitFor } from '../__mocks__/waitFor';
import axios from 'redaxios';

const testStatus = JSON.parse(JSON.stringify(defaultAccountEntity));

jest.mock('redaxios', () => {
  return {
    get: jest.fn().mockImplementation(() => {
      testStatus.storageUsed = 2;
      return Promise.resolve({ status: 200, data: testStatus });
    }),
    put: jest.fn().mockImplementation((url, body) => {
      if (url == 'http://test_url/account/notification?agent=test_token') {
        testStatus.pushEnabled = body;
      }
      if (url == 'http://test_url/account/searchable?agent=test_token') {
        testStatus.searchable = body;
      }
      if (url == 'http://test_url/account/seal?agent=test_token') {
        testStatus.seal = body;
      }
      return Promise.resolve({ status: 200 });
    }),
  }
})

class TestCrypto implements Crypto {

  public pbkdfSalt() {
    return { saltHex: 'SALT_HEX' }
  }

  public pbkdfKey(saltHex: string, password: string) {
    return { aesKeyHex: 'AES_KEY_HEX' }
  }

  public aesKey() {
    return { aesKeyHex: 'AES_KEY_HEX' };
  }

  public aesIv() {
    return { ivHex: 'IV_HEX' };
  }

  public aesEncrypt(data: string, ivHex: string, aesKeyHex: string) {
    return { encryptedDataB64: 'ENCRYPTED_DATA_B64' };
  }

  public aesDecrypt(encryptedDataB64: string, ivHex: string, aesKeyHex: string) {
    return { data: 'DATA' }
  }

  public rsaKey() {
    return { publicKeyB64: 'PUBLIC_KEY_B64', privateKeyB64: 'PRIVATE_KEY_B64' };
  }

  public rsaEncrypt(data: string, publicKeyB64: string) {
    return { encryptedDataB64: 'ENCRYPTED_DATA_B64' }
  }

  public rsaDecrypt(encryptedDataB64: string, privateKeyB64: string) {
    return { data: 'DATA' }
  }
}

class TestStore extends NoStore {
  public async getProfileRevision(): Promise<number> {
    return 4;
  }
}

test('allocates session correctly', async () => {
  let status: AccountStatus | null = null;
  const log = new ConsoleLogging();
  const store = new TestStore();
  const crypto = new TestCrypto();
  const account = new AccountModule(log, store, crypto, 'test_guid', 'test_token', 'test_url', false);
  account.addStatusListener((ev: AccountStatus) => { status = ev });
  account.setRevision(5);
  await waitFor(() => (status?.storageUsed == 2));
  account.enableRegistry();
  account.setRevision(6);
  await waitFor(() => Boolean(status?.searchable));
  account.disableRegistry();
  account.setRevision(7);
  await waitFor(() => !Boolean(status?.searchable));

  account.enableNotifications();
  account.setRevision(8);
  await waitFor(() => Boolean(status?.pushEnabled));
  account.disableNotifications();
  account.setRevision(9);
  await waitFor(() => !Boolean(status?.pushEnabled));

  account.setSeal('password');
  account.setRevision(10);
  await waitFor(() => Boolean(status?.sealSet));
});