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

import {TestEnv, makeSuite} from './helpers/make-suite';
import {RAY, APPROVAL_AMOUNT_LENDING_POOL, ZERO_ADDRESS} from '../helpers/constants';
import {convertToCurrencyDecimals} from '../helpers/contracts-helpers';
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");

  });

});