@unilogin/sdk/test/core/services/WalletStorageService.test.ts

SDK is a JS library, that communicates with relayer. SDK allows managing contract, by creating basic contract-calling messages.

import {expect} from 'chai';
import sinon from 'sinon';
import {Wallet} from 'ethers';
import {ETHER_NATIVE_TOKEN, ApplicationWallet, SerializableFutureWallet, TEST_GAS_PRICE} from '@unilogin/commons';
import {IStorageService} from '../../../src/core/models/IStorageService';
import {WalletStorageService} from '../../../src/core/services/WalletStorageService';
import {MemoryStorageService} from '../../../src/core/services/MemoryStorageService';
import {TEST_STORAGE_KEY} from '../../helpers/constants';

describe('WalletStorageService', () => {
  const baseWallet = {
    contractAddress: Wallet.createRandom().address,
    privateKey: Wallet.createRandom().privateKey,
  };

  const futureWallet: SerializableFutureWallet = {
    ...baseWallet,
    gasPrice: TEST_GAS_PRICE,
    ensName: 'name.mylogin.eth',
    gasToken: ETHER_NATIVE_TOKEN.address,
  };
  const applicationWallet: ApplicationWallet = {
    name: 'name',
    ...baseWallet,
  };

  const setup = () => {
    const storage: IStorageService = {
      get: sinon.fake(),
      set: sinon.fake(),
      remove: sinon.fake(),
    };
    const service = new WalletStorageService(storage, 'ganache');
    return {storage, service};
  };

  it('returns None state by default', () => {
    const {storage, service} = setup();
    storage.get = sinon.fake.returns(null);
    expect(service.load()).to.deep.eq({kind: 'None'});
    expect(storage.get).to.be.calledWith(TEST_STORAGE_KEY);
  });

  it('can load deployed state', () => {
    const {storage, service} = setup();
    storage.get = sinon.fake.returns(JSON.stringify({kind: 'Deployed', wallet: applicationWallet}));
    expect(service.load()).to.deep.eq({kind: 'Deployed', wallet: applicationWallet});
    expect(storage.get).to.be.calledWith(TEST_STORAGE_KEY);
  });

  it('can load future state', () => {
    const {storage, service} = setup();
    storage.get = sinon.fake.returns(JSON.stringify({kind: 'Future', name: 'name.mylogin.eth', wallet: futureWallet}));
    expect(service.load()).to.deep.eq({kind: 'Future', name: 'name.mylogin.eth', wallet: futureWallet});
    expect(storage.get).to.be.calledWith(TEST_STORAGE_KEY);
  });

  it('sanitizes data', () => {
    const {storage, service} = setup();
    storage.get = sinon.fake.returns(JSON.stringify({foo: 'bar'}));
    expect(() => service.load()).to.throw;
  });

  it('can save data', () => {
    const {storage, service} = setup();
    service.save({kind: 'Deployed', wallet: applicationWallet});
    expect(storage.set).to.be.calledWith(TEST_STORAGE_KEY, JSON.stringify({kind: 'Deployed', wallet: applicationWallet}));
  });

  it('can migrate data from previous versions', async () => {
    const {storage, service} = setup();
    const deployedRinkebyState = {
      kind: 'Deployed',
      wallet: {
        contractAddress: '0xCE824eA5d5a3810563B6e02114C66feFd0CaD13e',
        name: 'test.unilogin.test',
        privateKey: '0xe7f0024cb894893c855d5ba516969cb4c40838762c0dd0e5e730bfd4aa27b719',
      },
    };
    let savedState = JSON.stringify(deployedRinkebyState);
    storage.get = sinon.fake(() => savedState);
    storage.set = sinon.fake((key: string, val: string) => {
      savedState = val;
    });
    await service.migrate();
    expect(storage.get).to.be.calledWith('wallet');
    expect(storage.remove).to.be.calledWith('wallet');
    expect(storage.set).to.be.calledWith('wallet-rinkeby', savedState);
  });

  it('roundtrip', () => {
    const storage = new MemoryStorageService();
    const service = new WalletStorageService(storage, 'ganache');

    expect(service.load()).to.deep.eq({kind: 'None'});

    service.save({kind: 'Future', name: 'name.mylogin.eth', wallet: futureWallet});
    expect(service.load()).to.deep.eq({kind: 'Future', name: 'name.mylogin.eth', wallet: futureWallet});

    service.save({kind: 'Deployed', wallet: applicationWallet});
    expect(service.load()).to.deep.eq({kind: 'Deployed', wallet: applicationWallet});
  });
});