1fb7266469
- Introduced Aggregator.sol for Chainlink-compatible oracle functionality, including round-based updates and access control. - Added OracleWithCCIP.sol to extend Aggregator with CCIP cross-chain messaging capabilities. - Created .gitmodules to include OpenZeppelin contracts as a submodule. - Developed a comprehensive deployment guide in NEXT_STEPS_COMPLETE_GUIDE.md for Phase 2 and smart contract deployment. - Implemented Vite configuration for the orchestration portal, supporting both Vue and React frameworks. - Added server-side logic for the Multi-Cloud Orchestration Portal, including API endpoints for environment management and monitoring. - Created scripts for resource import and usage validation across non-US regions. - Added tests for CCIP error handling and integration to ensure robust functionality. - Included various new files and directories for the orchestration portal and deployment scripts.
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 "@emoney/AccountWalletRegistry.sol";
|
|
import "@emoney/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));
|
|
}
|
|
}
|
|
|