aave-protocol-v2/test/addresses-provider-registry.spec.ts

import {TestEnv, makeSuite} from './helpers/make-suite';
import {ZERO_ADDRESS} from '../helpers/constants';
import {ProtocolErrors} from '../helpers/types';

const {expect} = require('chai');

makeSuite('AddressesProviderRegistry', (testEnv: TestEnv) => {
  it('Checks the addresses provider is added to the registry', async () => {
    const {addressesProvider, registry} = testEnv;

    const providers = await registry.getAddressesProvidersList();

    expect(providers.length).to.be.equal(1, 'Invalid length of the addresses providers list');
    expect(providers[0].toString()).to.be.equal(
      addressesProvider.address,
      ' Invalid addresses provider added to the list'
    );
  });

  it('Registers a new mock addresses provider', async () => {
    const {users, registry} = testEnv;

    //simulating an addresses provider using the users[1] wallet address
    await registry.registerAddressesProvider(users[1].address, '2');

    const providers = await registry.getAddressesProvidersList();

    expect(providers.length).to.be.equal(2, 'Invalid length of the addresses providers list');
    expect(providers[1].toString()).to.be.equal(
      users[1].address,
      ' Invalid addresses provider added to the list'
    );
  });

  it('Removes the mock addresses provider', async () => {
    const {users, registry, addressesProvider} = testEnv;

    //checking the isAddressesProviderRegistered function
    const id = await registry.isAddressesProviderRegistered(users[1].address);

    expect(id).to.be.equal('2', 'Invalid isRegistered return value');

    await registry.unregisterAddressesProvider(users[1].address);

    const providers = await registry.getAddressesProvidersList();

    expect(providers.length).to.be.equal(2, 'Invalid length of the addresses providers list');
    expect(providers[0].toString()).to.be.equal(
      addressesProvider.address,
      ' Invalid addresses provider added to the list'
    );
    expect(providers[1].toString()).to.be.equal(ZERO_ADDRESS, ' Invalid addresses');
  });

  it('Tries to remove a unregistered addressesProvider', async () => {
    const {PROVIDER_NOT_REGISTERED} = ProtocolErrors;

    const {users, registry} = testEnv;

    await expect(registry.unregisterAddressesProvider(users[2].address)).to.be.revertedWith(
      PROVIDER_NOT_REGISTERED
    );
  });

  it('Tries to remove a unregistered addressesProvider', async () => {
    const {PROVIDER_NOT_REGISTERED} = ProtocolErrors;

    const {users, registry} = testEnv;

    await expect(registry.unregisterAddressesProvider(users[2].address)).to.be.revertedWith(
      PROVIDER_NOT_REGISTERED
    );
  });

  it('Tries to add an already added addressesProvider with a different id. Should overwrite the previous id', async () => {
    const {users, registry, addressesProvider} = testEnv;

    await registry.registerAddressesProvider(addressesProvider.address, '2');

    const providers = await registry.getAddressesProvidersList();

    const id = await registry.isAddressesProviderRegistered(addressesProvider.address);

    expect(providers.length).to.be.equal(2, 'Invalid length of the addresses providers list');

    expect(providers[0].toString()).to.be.equal(
      addressesProvider.address,
      ' Invalid addresses provider added to the list'
    );
    expect(providers[1].toString()).to.be.equal(ZERO_ADDRESS, ' Invalid addresses');
  });
});