-
Notifications
You must be signed in to change notification settings - Fork 95
Expand file tree
/
Copy pathhelpers.ts
More file actions
348 lines (287 loc) · 9.98 KB
/
Copy pathhelpers.ts
File metadata and controls
348 lines (287 loc) · 9.98 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
import { Core, Core__factory } from '@custom-types/contracts';
import { NamedAddresses } from '@custom-types/types';
import Safe from '@gnosis.pm/safe-core-sdk';
import EthersAdapter from '@gnosis.pm/safe-ethers-lib';
import { SignerWithAddress } from '@nomiclabs/hardhat-ethers/signers';
import chai from 'chai';
import CBN from 'chai-bn';
import { BigNumber, BigNumberish, Contract, Signer } from 'ethers';
import hre, { artifacts, ethers, network } from 'hardhat';
// use default BigNumber
chai.use(CBN(ethers.BigNumber));
const toBN = ethers.BigNumber.from;
const { expect } = chai;
const WETH9 = artifacts.readArtifactSync('WETH9');
async function deployDevelopmentWeth(): Promise<void> {
await network.provider.send('hardhat_setCode', [
'0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2',
WETH9.deployedBytecode
]);
const weth = await ethers.getContractAt(WETH9.abi, '0xc02aaa39b223fe8d0a0e5c4f27ead9083c756cc2');
await weth.init();
}
async function getAddresses(): Promise<NamedAddresses> {
const [
userAddress,
secondUserAddress,
beneficiaryAddress1,
beneficiaryAddress2,
governorAddress,
genesisGroup,
keeperAddress,
pcvControllerAddress,
minterAddress,
burnerAddress,
guardianAddress
] = (await ethers.getSigners()).map((signer) => signer.address);
return {
userAddress,
secondUserAddress,
beneficiaryAddress1,
beneficiaryAddress2,
governorAddress,
genesisGroup,
keeperAddress,
pcvControllerAddress,
minterAddress,
burnerAddress,
guardianAddress
};
}
async function getImpersonatedSigner(address: string): Promise<SignerWithAddress> {
await hre.network.provider.request({
method: 'hardhat_impersonateAccount',
params: [address]
});
const signer = await ethers.getSigner(address);
return signer;
}
async function increaseTime(amount: number | string | BigNumberish): Promise<void> {
await time.increase(amount);
}
async function resetTime(): Promise<void> {
await resetFork();
}
async function resetFork(): Promise<void> {
if (process.env.NO_RESET) {
return;
}
await hre.network.provider.request({
method: 'hardhat_reset',
params: [
{
forking: hre.config.networks.hardhat.forking
? {
jsonRpcUrl: hre.config.networks.hardhat.forking.url,
blockNumber: hre.config.networks.hardhat.forking.blockNumber
}
: undefined
}
]
});
}
async function setNextBlockTimestamp(time: number): Promise<void> {
await hre.network.provider.request({
method: 'evm_setNextBlockTimestamp',
params: [time]
});
}
async function latestTime(): Promise<number> {
const { timestamp } = await ethers.provider.getBlock(await ethers.provider.getBlockNumber());
return timestamp as number;
}
async function mine(): Promise<void> {
await hre.network.provider.request({
method: 'hardhat_mine'
});
}
async function getCore(): Promise<Core> {
const { governorAddress, pcvControllerAddress, minterAddress, burnerAddress, guardianAddress } = await getAddresses();
await hre.network.provider.request({
method: 'hardhat_impersonateAccount',
params: [governorAddress]
});
const governorSigner = await ethers.getSigner(governorAddress);
const coreFactory = new Core__factory(governorSigner);
const core = await coreFactory.deploy();
await core.init();
await core.grantMinter(minterAddress);
await core.grantBurner(burnerAddress);
await core.grantPCVController(pcvControllerAddress);
await core.grantGuardian(guardianAddress);
return core;
}
const validateArraysEqual = (arrayA: string[], arrayB: string[]) => {
expect(arrayA.length).to.equal(arrayB.length);
arrayA.every((a) => expect(arrayB.map((b) => b.toLowerCase()).includes(a.toLowerCase())));
};
async function expectApprox(
actual: string | number | BigNumberish,
expected: string | number | BigNumberish,
magnitude = '1000'
): Promise<void> {
const actualBN = toBN(actual);
const expectedBN = toBN(expected);
const magnitudeBN = toBN(magnitude);
const diff = actualBN.sub(expectedBN);
const diffAbs = diff.abs();
if (expected.toString() == '0' || expected == 0 || expected == '0') {
expect(diffAbs).to.be.lt(magnitudeBN);
} else {
expect(diffAbs.div(expected).lt(magnitudeBN)).to.be.true;
}
}
// expectApproxAbs(a, b, c) checks if b is between [a-c, a+c]
async function expectApproxAbs(
actual: string | number | BigNumberish,
expected: string | number | BigNumberish,
diff = '1000000000000000000'
): Promise<void> {
const actualBN = toBN(actual);
const expectedBN = toBN(expected);
const diffBN = toBN(diff);
const lowerBound = expectedBN.sub(diffBN);
const upperBound = expectedBN.add(diffBN);
expect(actualBN).to.be.gte(lowerBound);
expect(actualBN).to.be.lte(upperBound);
}
async function expectEvent(tx: any, contract: any, event: string, args: any[]): Promise<void> {
await expect(tx)
.to.emit(contract, event)
.withArgs(...args);
}
async function expectRevert(tx: Promise<any>, errorMessage: string): Promise<void> {
await expect(tx).to.be.revertedWith(errorMessage);
}
async function expectUnspecifiedRevert(tx: Promise<any>): Promise<void> {
await expect(tx).to.be.reverted;
}
const ZERO_ADDRESS = ethers.constants.AddressZero;
const MAX_UINT256 = ethers.constants.MaxUint256;
const balance = {
current: async (address: string): Promise<BigNumber> => {
const balance = await ethers.provider.getBalance(address);
return balance;
}
};
async function overwriteChainlinkAggregator(aggregatorAddress: string, value: string, decimals: string) {
// Deploy new mock aggregator
const factory = await ethers.getContractFactory('MockChainlinkOracle');
const mockAggregator = await factory.deploy(value, decimals);
await mockAggregator.deployTransaction.wait();
// Overwrite storage at chainlink address to use mock aggregator for updates
const address = `0x00000000000000000000${mockAggregator.address.slice(2)}0005`;
await hre.network.provider.send('hardhat_setStorageAt', [aggregatorAddress, '0x2', address]);
}
const time = {
latest: async (): Promise<number> => latestTime(),
latestBlock: async (): Promise<number> => await ethers.provider.getBlockNumber(),
increase: async (duration: number | string | BigNumberish): Promise<void> => {
const durationBN = ethers.BigNumber.from(duration);
if (durationBN.lt(ethers.constants.Zero)) throw Error(`Cannot increase time by a negative amount (${duration})`);
if (durationBN.eq(ethers.constants.Zero)) {
await hre.network.provider.send('hardhat_mine');
} else {
await hre.network.provider.send('hardhat_mine', ['0x2', ethers.utils.hexStripZeros(durationBN.toHexString())]);
}
},
increaseTo: async (target: number | string | BigNumberish): Promise<void> => {
const targetBN = ethers.BigNumber.from(target);
const now = ethers.BigNumber.from(await time.latest());
if (targetBN.lt(now)) throw Error(`Cannot increase current time (${now}) to a moment in the past (${target})`);
const diff = targetBN.sub(now);
return time.increase(diff);
},
advanceBlockTo: async (target: number | string | BigNumberish): Promise<void> => {
target = ethers.BigNumber.from(target);
const currentBlock = await time.latestBlock();
if (target.lt(currentBlock))
throw Error(`Target block #(${target}) is lower than current block #(${currentBlock})`);
const diff = target.sub(currentBlock);
await hre.network.provider.send('hardhat_mine', [ethers.utils.hexStripZeros(diff.toHexString())]);
},
advanceBlock: async (): Promise<void> => {
await hre.network.provider.send('hardhat_mine');
}
};
async function performDAOAction(
feiDAO: Contract,
multisigAddress: string,
calldatas: string[],
targets: string[],
values: number[]
): Promise<void> {
const description = '';
await hre.network.provider.request({
method: 'hardhat_impersonateAccount',
params: [multisigAddress]
});
const signer = await ethers.getSigner(multisigAddress);
// Propose
// note ethers.js requires using this notation when two overloaded methods exist)
// https://docs.ethers.io/v5/migration/web3/#migration-from-web3-js--contracts--overloaded-functions
await feiDAO.connect(signer)['propose(address[],uint256[],bytes[],string)'](targets, values, calldatas, description);
const pid = await feiDAO.hashProposal(targets, values, calldatas, ethers.utils.keccak256(description));
const startBlock = (await feiDAO.proposals(pid)).startBlock;
await time.advanceBlockTo(startBlock.toString());
// vote
await feiDAO.connect(signer).castVote(pid, 1);
const endBlock = (await feiDAO.proposals(pid)).endBlock;
await time.advanceBlockTo(endBlock.toString());
// queue
await feiDAO['queue(address[],uint256[],bytes[],bytes32)'](
targets,
values,
calldatas,
ethers.utils.keccak256(description)
);
await time.increase('1000000');
// execute
await feiDAO['execute(address[],uint256[],bytes[],bytes32)'](
targets,
values,
calldatas,
ethers.utils.keccak256(description)
);
}
async function initialiseGnosisSDK(safeOwner: Signer, safeAddress: string): Promise<Safe> {
const ethAdapter = new EthersAdapter({
ethers,
signer: safeOwner
});
const { chainId } = await safeOwner.provider!.getNetwork();
const contractNetworks = {
[chainId]: {
multiSendAddress: '0x8D29bE29923b68abfDD21e541b9374737B49cdAD',
safeMasterCopyAddress: '0x34CfAC646f301356fAa8B21e94227e3583Fe3F5F',
safeProxyFactoryAddress: '0x76E2cFc1F5Fa8F6a5b3fC4c8F4788F0116861F9B'
}
};
return Safe.create({ ethAdapter, safeAddress, contractNetworks });
}
export {
ZERO_ADDRESS,
MAX_UINT256,
time,
balance,
expectEvent,
expectRevert,
expectUnspecifiedRevert,
// functions
mine,
getCore,
getAddresses,
increaseTime,
latestTime,
expectApprox,
expectApproxAbs,
deployDevelopmentWeth,
getImpersonatedSigner,
setNextBlockTimestamp,
resetTime,
resetFork,
overwriteChainlinkAggregator,
performDAOAction,
initialiseGnosisSDK,
validateArraysEqual
};