import assert from 'assert';
import migration31 from '../../../app/scripts/migrations/031';

describe('migration #31', function () {
  it('should set completedOnboarding to true if vault exists', function (done) {
    const oldStorage = {
      meta: {},
      data: {
        PreferencesController: {
          tokens: [
            { address: '0xa', symbol: 'A', decimals: 4 },
            { address: '0xb', symbol: 'B', decimals: 4 },
          ],
          identities: {
            '0x6d14': {},
            '0x3695': {},
          },
        },
        KeyringController: {
          vault: {
            data: 'test0',
            iv: 'test1',
            salt: 'test2',
          },
        },
      },
    };

    migration31
      .migrate(oldStorage)
      .then((newStorage) => {
        assert.equal(
          newStorage.data.PreferencesController.completedOnboarding,
          true,
        );
        done();
      })
      .catch(done);
  });

  it('should set completedOnboarding to false if vault does not exist', function (done) {
    const oldStorage = {
      meta: {},
      data: {
        PreferencesController: {
          tokens: [
            { address: '0xa', symbol: 'A', decimals: 4 },
            { address: '0xb', symbol: 'B', decimals: 4 },
          ],
          identities: {
            '0x6d14': {},
            '0x3695': {},
          },
        },
        KeyringController: {},
      },
    };

    migration31
      .migrate(oldStorage)
      .then((newStorage) => {
        assert.equal(
          newStorage.data.PreferencesController.completedOnboarding,
          false,
        );
        done();
      })
      .catch(done);
  });
});