Skip to content

Commit cb6255f

Browse files
committed
add setUpGeneral that runs before each test and after each setUpFork / source / target
1 parent fc7fc89 commit cb6255f

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

src/testing/WormholeRelayerTest.sol

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,10 +37,15 @@ struct ActiveFork {
3737

3838
abstract contract WormholeRelayerTest is Test {
3939
/**
40-
* @dev virtual function to initialize active forks before each test
40+
* @dev required override to initialize active forks before each test
4141
*/
4242
function setUpFork(ActiveFork memory fork) public virtual;
4343

44+
/**
45+
* @dev optional override that runs after all forks have been set up
46+
*/
47+
function setUpGeneral() public virtual {}
48+
4449
uint256 constant DEVNET_GUARDIAN_PK = 0xcfb12303a19cde580bb4dd771639b0d26bc68353645571a8cff516ab2ee113a0;
4550

4651
// conveneince information to set up tests against testnet/mainnet forks
@@ -123,6 +128,9 @@ abstract contract WormholeRelayerTest is Test {
123128

124129
// Allow the offchain relayer to work on all forks
125130
vm.makePersistent(address(mockOffchainRelayer));
131+
132+
vm.selectFork(firstFork.fork);
133+
setUpGeneral();
126134
}
127135

128136
function performDelivery() public {
@@ -356,6 +364,8 @@ abstract contract WormholeRelayerBasicTest is WormholeRelayerTest {
356364
}
357365

358366
function setUp() public override {
367+
sourceFork = 0;
368+
targetFork = 1;
359369
_setUp();
360370
// aliases can't be set until after setUp
361371
guardianSource = activeForks[activeForksList[0]].guardian;

0 commit comments

Comments
 (0)