651ff4f7eb
- Add Foundry project configuration (foundry.toml, foundry.lock) - Add Solidity contracts (TokenFactory138, BridgeVault138, ComplianceRegistry, etc.) - Add API definitions (OpenAPI, GraphQL, gRPC, AsyncAPI) - Add comprehensive test suite (unit, integration, fuzz, invariants) - Add API services (REST, GraphQL, orchestrator, packet service) - Add documentation (ISO20022 mapping, runbooks, adapter guides) - Add development tools (RBC tool, Swagger UI, mock server) - Update OpenZeppelin submodules to v5.0.0
104 lines
3.2 KiB
Solidity
104 lines
3.2 KiB
Solidity
// SPDX-License-Identifier: MIT
|
|
pragma solidity ^0.8.20;
|
|
|
|
import "forge-std/Test.sol";
|
|
import "../../src/RailEscrowVault.sol";
|
|
import "../../src/interfaces/IRailEscrowVault.sol";
|
|
import "../../src/libraries/RailTypes.sol";
|
|
import "@openzeppelin/contracts/token/ERC20/ERC20.sol";
|
|
|
|
contract MockERC20 is ERC20 {
|
|
constructor() ERC20("Mock Token", "MOCK") {
|
|
_mint(msg.sender, 1000000 * 10**18);
|
|
}
|
|
|
|
function mint(address to, uint256 amount) external {
|
|
_mint(to, amount);
|
|
}
|
|
}
|
|
|
|
contract RailEscrowVaultTest is Test {
|
|
RailEscrowVault public vault;
|
|
MockERC20 public token;
|
|
address public admin;
|
|
address public settlementOperator;
|
|
address public user;
|
|
|
|
function setUp() public {
|
|
admin = address(0x1);
|
|
settlementOperator = address(0x2);
|
|
user = address(0x10);
|
|
|
|
vault = new RailEscrowVault(admin);
|
|
token = new MockERC20();
|
|
|
|
vm.startPrank(admin);
|
|
vault.grantRole(vault.SETTLEMENT_OPERATOR_ROLE(), settlementOperator);
|
|
vm.stopPrank();
|
|
|
|
// Give user some tokens
|
|
token.mint(user, 10000 * 10**18);
|
|
}
|
|
|
|
function test_lock() public {
|
|
uint256 amount = 1000 * 10**18;
|
|
uint256 triggerId = 1;
|
|
|
|
vm.startPrank(user);
|
|
token.approve(address(vault), amount);
|
|
vm.stopPrank();
|
|
|
|
vm.expectEmit(true, true, false, true);
|
|
emit IRailEscrowVault.Locked(address(token), user, amount, triggerId, uint8(RailTypes.Rail.SWIFT));
|
|
|
|
vm.prank(settlementOperator);
|
|
vault.lock(address(token), user, amount, triggerId, RailTypes.Rail.SWIFT);
|
|
|
|
assertEq(vault.getEscrowAmount(address(token), triggerId), amount);
|
|
assertEq(vault.getTotalEscrow(address(token)), amount);
|
|
assertEq(token.balanceOf(address(vault)), amount);
|
|
}
|
|
|
|
function test_release() public {
|
|
uint256 amount = 1000 * 10**18;
|
|
uint256 triggerId = 1;
|
|
address recipient = address(0x20);
|
|
|
|
vm.startPrank(user);
|
|
token.approve(address(vault), amount);
|
|
vm.stopPrank();
|
|
|
|
vm.prank(settlementOperator);
|
|
vault.lock(address(token), user, amount, triggerId, RailTypes.Rail.SWIFT);
|
|
|
|
uint256 recipientBalanceBefore = token.balanceOf(recipient);
|
|
|
|
vm.expectEmit(true, true, false, true);
|
|
emit IRailEscrowVault.Released(address(token), recipient, amount, triggerId);
|
|
|
|
vm.prank(settlementOperator);
|
|
vault.release(address(token), recipient, amount, triggerId);
|
|
|
|
assertEq(vault.getEscrowAmount(address(token), triggerId), 0);
|
|
assertEq(vault.getTotalEscrow(address(token)), 0);
|
|
assertEq(token.balanceOf(recipient), recipientBalanceBefore + amount);
|
|
}
|
|
|
|
function test_release_insufficientEscrow() public {
|
|
uint256 amount = 1000 * 10**18;
|
|
uint256 triggerId = 1;
|
|
|
|
vm.startPrank(user);
|
|
token.approve(address(vault), amount);
|
|
vm.stopPrank();
|
|
|
|
vm.prank(settlementOperator);
|
|
vault.lock(address(token), user, amount, triggerId, RailTypes.Rail.SWIFT);
|
|
|
|
vm.prank(settlementOperator);
|
|
vm.expectRevert("RailEscrowVault: insufficient escrow");
|
|
vault.release(address(token), address(0x20), amount + 1, triggerId);
|
|
}
|
|
}
|
|
|