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
106 lines
3.8 KiB
Solidity
106 lines
3.8 KiB
Solidity
// SPDX-License-Identifier: MIT
|
|
pragma solidity ^0.8.20;
|
|
|
|
import "forge-std/Test.sol";
|
|
import "../../src/AccountWalletRegistry.sol";
|
|
import "../../src/interfaces/IAccountWalletRegistry.sol";
|
|
|
|
contract AccountWalletRegistryTest is Test {
|
|
AccountWalletRegistry public registry;
|
|
address public admin;
|
|
address public accountManager;
|
|
|
|
bytes32 public accountRefId1 = keccak256("account1");
|
|
bytes32 public walletRefId1 = keccak256("wallet1");
|
|
bytes32 public walletRefId2 = keccak256("wallet2");
|
|
bytes32 public provider1 = keccak256("METAMASK");
|
|
bytes32 public provider2 = keccak256("FIREBLOCKS");
|
|
|
|
function setUp() public {
|
|
admin = address(0x1);
|
|
accountManager = address(0x2);
|
|
|
|
registry = new AccountWalletRegistry(admin);
|
|
|
|
vm.startPrank(admin);
|
|
registry.grantRole(registry.ACCOUNT_MANAGER_ROLE(), accountManager);
|
|
vm.stopPrank();
|
|
}
|
|
|
|
function test_linkAccountToWallet() public {
|
|
vm.expectEmit(true, true, false, true);
|
|
emit IAccountWalletRegistry.AccountWalletLinked(accountRefId1, walletRefId1, provider1, uint64(block.timestamp));
|
|
|
|
vm.prank(accountManager);
|
|
registry.linkAccountToWallet(accountRefId1, walletRefId1, provider1);
|
|
|
|
assertTrue(registry.isLinked(accountRefId1, walletRefId1));
|
|
assertTrue(registry.isActive(accountRefId1, walletRefId1));
|
|
|
|
IAccountWalletRegistry.WalletLink[] memory wallets = registry.getWallets(accountRefId1);
|
|
assertEq(wallets.length, 1);
|
|
assertEq(wallets[0].walletRefId, walletRefId1);
|
|
assertEq(wallets[0].provider, provider1);
|
|
assertTrue(wallets[0].active);
|
|
}
|
|
|
|
function test_linkMultipleWallets() public {
|
|
vm.prank(accountManager);
|
|
registry.linkAccountToWallet(accountRefId1, walletRefId1, provider1);
|
|
|
|
vm.prank(accountManager);
|
|
registry.linkAccountToWallet(accountRefId1, walletRefId2, provider2);
|
|
|
|
IAccountWalletRegistry.WalletLink[] memory wallets = registry.getWallets(accountRefId1);
|
|
assertEq(wallets.length, 2);
|
|
assertEq(wallets[0].walletRefId, walletRefId1);
|
|
assertEq(wallets[1].walletRefId, walletRefId2);
|
|
}
|
|
|
|
function test_unlinkAccountFromWallet() public {
|
|
vm.prank(accountManager);
|
|
registry.linkAccountToWallet(accountRefId1, walletRefId1, provider1);
|
|
|
|
assertTrue(registry.isActive(accountRefId1, walletRefId1));
|
|
|
|
vm.expectEmit(true, true, false, false);
|
|
emit IAccountWalletRegistry.AccountWalletUnlinked(accountRefId1, walletRefId1);
|
|
|
|
vm.prank(accountManager);
|
|
registry.unlinkAccountFromWallet(accountRefId1, walletRefId1);
|
|
|
|
assertTrue(registry.isLinked(accountRefId1, walletRefId1)); // Still linked
|
|
assertFalse(registry.isActive(accountRefId1, walletRefId1)); // But inactive
|
|
}
|
|
|
|
function test_getAccounts() public {
|
|
bytes32 accountRefId2 = keccak256("account2");
|
|
|
|
vm.prank(accountManager);
|
|
registry.linkAccountToWallet(accountRefId1, walletRefId1, provider1);
|
|
|
|
vm.prank(accountManager);
|
|
registry.linkAccountToWallet(accountRefId2, walletRefId1, provider1);
|
|
|
|
bytes32[] memory accounts = registry.getAccounts(walletRefId1);
|
|
assertEq(accounts.length, 2);
|
|
}
|
|
|
|
function test_linkAccountToWallet_reactivate() public {
|
|
vm.prank(accountManager);
|
|
registry.linkAccountToWallet(accountRefId1, walletRefId1, provider1);
|
|
|
|
vm.prank(accountManager);
|
|
registry.unlinkAccountFromWallet(accountRefId1, walletRefId1);
|
|
|
|
assertFalse(registry.isActive(accountRefId1, walletRefId1));
|
|
|
|
// Reactivate
|
|
vm.prank(accountManager);
|
|
registry.linkAccountToWallet(accountRefId1, walletRefId1, provider1);
|
|
|
|
assertTrue(registry.isActive(accountRefId1, walletRefId1));
|
|
}
|
|
}
|
|
|