Skip to content

Commit

Permalink
refactor: update tests
Browse files Browse the repository at this point in the history
  • Loading branch information
ChiTimesChi committed Nov 25, 2024
1 parent 22ab91d commit 209fbb1
Show file tree
Hide file tree
Showing 5 changed files with 15 additions and 15 deletions.
2 changes: 1 addition & 1 deletion packages/contracts-rfq/test/FastBridgeV2.Dst.Base.t.sol
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,6 @@ contract FastBridgeV2DstBaseTest is FastBridgeV2Test {
{
bytes memory request = BridgeTransactionV2Lib.encodeV2(bridgeTx);
vm.prank({msgSender: caller, txOrigin: caller});
fastBridge.relay{value: msgValue}(request, relayer);
fastBridge.relayV2{value: msgValue}(request, relayer);
}
}
6 changes: 3 additions & 3 deletions packages/contracts-rfq/test/FastBridgeV2.Dst.t.sol
Original file line number Diff line number Diff line change
Expand Up @@ -453,19 +453,19 @@ contract FastBridgeV2DstTest is FastBridgeV2DstBaseTest {
// V1 doesn't have any version field
expectRevertUnsupportedVersion(0);
vm.prank({msgSender: relayerA, txOrigin: relayerA});
fastBridge.relay(mockRequestV1, relayerB);
fastBridge.relayV2(mockRequestV1, relayerB);
}

function test_relay_withRelayerAddress_revert_invalidRequestV2() public {
expectRevertInvalidEncodedTx();
vm.prank({msgSender: relayerA, txOrigin: relayerA});
fastBridge.relay(invalidRequestV2, relayerB);
fastBridge.relayV2(invalidRequestV2, relayerB);
}

function test_relay_withRelayerAddress_revert_requestV3() public {
expectRevertUnsupportedVersion(3);
vm.prank({msgSender: relayerA, txOrigin: relayerA});
fastBridge.relay(mockRequestV3, relayerB);
fastBridge.relayV2(mockRequestV3, relayerB);
}

function test_relay_withRelayerAddress_revert_chainIncorrect() public {
Expand Down
8 changes: 4 additions & 4 deletions packages/contracts-rfq/test/FastBridgeV2.Src.Base.t.sol
Original file line number Diff line number Diff line change
Expand Up @@ -68,12 +68,12 @@ abstract contract FastBridgeV2SrcBaseTest is FastBridgeV2Test {
public
{
vm.prank({msgSender: caller, txOrigin: caller});
fastBridge.bridge{value: msgValue}(params, paramsV2);
fastBridge.bridgeV2{value: msgValue}(params, paramsV2);
}

function prove(address caller, bytes32 transactionId, bytes32 destTxHash, address relayer) public {
vm.prank({msgSender: caller, txOrigin: caller});
fastBridge.prove(transactionId, destTxHash, relayer);
fastBridge.proveV2(transactionId, destTxHash, relayer);
}

function prove(address caller, IFastBridgeV2.BridgeTransactionV2 memory bridgeTx, bytes32 destTxHash) public {
Expand All @@ -83,7 +83,7 @@ abstract contract FastBridgeV2SrcBaseTest is FastBridgeV2Test {

function claim(address caller, IFastBridgeV2.BridgeTransactionV2 memory bridgeTx) public {
vm.prank({msgSender: caller, txOrigin: caller});
fastBridge.claim(BridgeTransactionV2Lib.encodeV2(bridgeTx));
fastBridge.claimV2(BridgeTransactionV2Lib.encodeV2(bridgeTx));
}

function claim(address caller, IFastBridgeV2.BridgeTransactionV2 memory bridgeTx, address to) public {
Expand All @@ -98,7 +98,7 @@ abstract contract FastBridgeV2SrcBaseTest is FastBridgeV2Test {

function cancel(address caller, IFastBridgeV2.BridgeTransactionV2 memory bridgeTx) public virtual {
vm.prank({msgSender: caller, txOrigin: caller});
fastBridge.cancel(BridgeTransactionV2Lib.encodeV2(bridgeTx));
fastBridge.cancelV2(BridgeTransactionV2Lib.encodeV2(bridgeTx));
}

function test_nonce() public view {
Expand Down
12 changes: 6 additions & 6 deletions packages/contracts-rfq/test/FastBridgeV2.Src.t.sol
Original file line number Diff line number Diff line change
Expand Up @@ -998,19 +998,19 @@ contract FastBridgeV2SrcTest is FastBridgeV2SrcBaseTest {
// V1 doesn't have any version field
expectRevertUnsupportedVersion(0);
vm.prank({msgSender: relayerA, txOrigin: relayerA});
fastBridge.claim(mockRequestV1);
fastBridge.claimV2(mockRequestV1);
}

function test_claim_revert_invalidRequestV2() public {
expectRevertInvalidEncodedTx();
vm.prank({msgSender: relayerA, txOrigin: relayerA});
fastBridge.claim(invalidRequestV2);
fastBridge.claimV2(invalidRequestV2);
}

function test_claim_revert_requestV3() public {
expectRevertUnsupportedVersion(3);
vm.prank({msgSender: relayerA, txOrigin: relayerA});
fastBridge.claim(mockRequestV3);
fastBridge.claimV2(mockRequestV3);
}

function test_claim_toDiffAddress_revert_requestV1() public {
Expand All @@ -1036,18 +1036,18 @@ contract FastBridgeV2SrcTest is FastBridgeV2SrcBaseTest {
// V1 doesn't have any version field
expectRevertUnsupportedVersion(0);
vm.prank({msgSender: relayerA, txOrigin: relayerA});
fastBridge.cancel(mockRequestV1);
fastBridge.cancelV2(mockRequestV1);
}

function test_cancel_revert_invalidRequestV2() public {
expectRevertInvalidEncodedTx();
vm.prank({msgSender: relayerA, txOrigin: relayerA});
fastBridge.cancel(invalidRequestV2);
fastBridge.cancelV2(invalidRequestV2);
}

function test_cancel_revert_requestV3() public {
expectRevertUnsupportedVersion(3);
vm.prank({msgSender: relayerA, txOrigin: relayerA});
fastBridge.cancel(mockRequestV3);
fastBridge.cancelV2(mockRequestV3);
}
}
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@ contract FastBridgeV2TokenZapV1SrcTest is TokenZapV1IntegrationTest {
public
{
vm.prank({msgSender: user, txOrigin: user});
fastBridge.bridge{value: isToken ? 0 : SRC_AMOUNT}(params, paramsV2);
fastBridge.bridgeV2{value: isToken ? 0 : SRC_AMOUNT}(params, paramsV2);
}

function expectEventBridgeRequested(
Expand Down

0 comments on commit 209fbb1

Please sign in to comment.