30 lines
1019 B
JavaScript
30 lines
1019 B
JavaScript
const { HardhatError } = require('hardhat/internal/core/errors');
|
|
|
|
function isExpectedError(e, suffix) {
|
|
// HH700: Artifact not found - from https://hardhat.org/hardhat-runner/docs/errors#HH700
|
|
return HardhatError.isHardhatError(e) && e.number === 700 && suffix !== '';
|
|
}
|
|
|
|
// Modifies the artifact require functions so that instead of X it loads the XUpgradeable contract.
|
|
// This allows us to run the same test suite on both the original and the transpiled and renamed Upgradeable contracts.
|
|
extendEnvironment(hre => {
|
|
const suffixes = ['UpgradeableWithInit', 'Upgradeable', ''];
|
|
|
|
// Ethers
|
|
const originalReadArtifact = hre.artifacts.readArtifact;
|
|
hre.artifacts.readArtifact = async function (name) {
|
|
for (const suffix of suffixes) {
|
|
try {
|
|
return await originalReadArtifact.call(this, name + suffix);
|
|
} catch (e) {
|
|
if (isExpectedError(e, suffix)) {
|
|
continue;
|
|
} else {
|
|
throw e;
|
|
}
|
|
}
|
|
}
|
|
throw new Error('Unreachable');
|
|
};
|
|
});
|