hyperlane-monorepo/solidity/optics-core/test/Upgrade.test.js

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

describe('Upgrade', async () => {
  let proxy, upgradeBeacon, upgradeBeaconController;
  const a = 5;
  const b = 10;
  const stateVar = 17;

  before(async () => {
    // SETUP CONTRACT SUITE
    const { contracts } = await optics.deployUpgradeSetupAndProxy(
      'MysteryMathV1',
    );

    proxy = contracts.proxyWithImplementation;
    upgradeBeacon = contracts.upgradeBeacon;
    upgradeBeaconController = contracts.upgradeBeaconController;

    // Set state of proxy
    await proxy.setState(stateVar);
  });

  it('Pre-Upgrade returns version 1', async () => {
    const versionResult = await proxy.version();
    expect(versionResult).to.equal(1);
  });

  it('Pre-Upgrade returns the math from implementation v1', async () => {
    const mathResult = await proxy.doMath(a, b);
    expect(mathResult).to.equal(a + b);
  });

  it('Pre-Upgrade returns the expected state variable', async () => {
    const stateResult = await proxy.getState();
    expect(stateResult).to.equal(stateVar);
  });

  it('Upgrades without problem', async () => {
    // Deploy Implementation 2
    const implementation = await optics.deployImplementation('MysteryMathV2');

    // Upgrade to implementation 2
    await upgradeBeaconController.upgrade(
      upgradeBeacon.address,
      implementation.address,
    );
  });

  it('Post-Upgrade returns version 2', async () => {
    const versionResult = await proxy.version();
    expect(versionResult).to.equal(2);
  });

  it('Post-Upgrade returns the math from implementation v2', async () => {
    const mathResult = await proxy.doMath(a, b);
    expect(mathResult).to.equal(a * b);
  });

  it('Post-Upgrade preserved the state variable', async () => {
    const stateResult = await proxy.getState();
    expect(stateResult).to.equal(stateVar);
  });
});