Skip to content

Commit

Permalink
test: add GhoStableDebtToken unit tests
Browse files Browse the repository at this point in the history
  • Loading branch information
cedephrase committed Apr 27, 2023
1 parent 9551aa7 commit dc8b27a
Show file tree
Hide file tree
Showing 2 changed files with 222 additions and 0 deletions.
12 changes: 12 additions & 0 deletions src/contracts/foundry-test/TestEnv.sol
View file
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@ import {MockedPool} from './mocks/MockedPool.sol';
import {MockedProvider} from './mocks/MockedProvider.sol';
import {MockedAclManager} from './mocks/MockedAclManager.sol';
import {GhoVariableDebtToken} from '../facilitators/aave/tokens/GhoVariableDebtToken.sol';
import {GhoStableDebtToken} from '../facilitators/aave/tokens/GhoStableDebtToken.sol';
import {GhoFlashMinter} from '../facilitators/flashMinter/GhoFlashMinter.sol';
import {MockFlashBorrower} from '../facilitators/flashMinter/mocks/MockFlashBorrower.sol';
import {IERC3156FlashBorrower} from '@openzeppelin/contracts/interfaces/IERC3156FlashBorrower.sol';
Expand Down Expand Up @@ -52,6 +53,7 @@ contract TestEnv is Test {
MockedConfigurator CONFIGURATOR;
WETH9Mock WETH;
GhoVariableDebtToken GHO_DEBT_TOKEN;
GhoStableDebtToken GHO_STABLE_DEBT_TOKEN;
GhoAToken GHO_ATOKEN;
GhoFlashMinter GHO_FLASH_MINTER;
GhoDiscountRateStrategy GHO_DISCOUNT_STRATEGY;
Expand Down Expand Up @@ -89,6 +91,7 @@ contract TestEnv is Test {
IPool iPool = IPool(address(POOL));
WETH = new WETH9Mock('Wrapped Ether', 'WETH', faucet);
GHO_DEBT_TOKEN = new GhoVariableDebtToken(iPool);
GHO_STABLE_DEBT_TOKEN = new GhoStableDebtToken(iPool);
GHO_ATOKEN = new GhoAToken(iPool);
GHO_DEBT_TOKEN.initialize(
iPool,
Expand All @@ -99,6 +102,15 @@ contract TestEnv is Test {
'GHOVarDebt',
empty
);
GHO_STABLE_DEBT_TOKEN.initialize(
iPool,
ghoToken,
IAaveIncentivesController(address(0)),
18,
'GHO Stable Debt',
'GHOStaDebt',
empty
);
GHO_ATOKEN.initialize(
iPool,
treasury,
Expand Down
210 changes: 210 additions & 0 deletions src/contracts/foundry-test/TestGhoStableDebtToken.t.sol
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,210 @@
// SPDX-License-Identifier: MIT
pragma solidity ^0.8.0;

import 'forge-std/Test.sol';

import './TestEnv.sol';
import {IPool} from '@aave/core-v3/contracts/interfaces/IPool.sol';
import {Errors} from '@aave/core-v3/contracts/protocol/libraries/helpers/Errors.sol';
import {DebtUtils} from './libraries/DebtUtils.sol';
import {GhoActions} from './libraries/GhoActions.sol';

contract TestGhoStableDebtToken is Test, GhoActions {
address public alice;
address public bob;
address public carlos;

function setUp() public {
alice = users[0];
bob = users[1];
carlos = users[2];
}

function testConstructor() public {
GhoStableDebtToken debtToken = new GhoStableDebtToken(IPool(address(POOL)));
assertEq(debtToken.name(), 'GHO_STABLE_DEBT_TOKEN_IMPL', 'Wrong default ERC20 name');
assertEq(debtToken.symbol(), 'GHO_STABLE_DEBT_TOKEN_IMPL', 'Wrong default ERC20 symbol');
assertEq(debtToken.decimals(), 0, 'Wrong default ERC20 decimals');
}

function testInitialize() public {
GhoStableDebtToken debtToken = new GhoStableDebtToken(IPool(address(POOL)));
string memory tokenName = 'GHO Stable Debt';
string memory tokenSymbol = 'GhoStaDebt';
bytes memory empty;
debtToken.initialize(
IPool(address(POOL)),
address(GHO_TOKEN),
IAaveIncentivesController(address(0)),
18,
tokenName,
tokenSymbol,
empty
);

assertEq(debtToken.name(), tokenName, 'Wrong initialized name');
assertEq(debtToken.symbol(), tokenSymbol, 'Wrong initialized symbol');
assertEq(debtToken.decimals(), 18, 'Wrong ERC20 decimals');
}

function testInitializePoolRevert() public {
string memory tokenName = 'GHO Stable Debt';
string memory tokenSymbol = 'GhoStaDebt';
bytes memory empty;

GhoStableDebtToken debtToken = new GhoStableDebtToken(IPool(address(POOL)));
vm.expectRevert(bytes(Errors.POOL_ADDRESSES_DO_NOT_MATCH));
debtToken.initialize(
IPool(address(0)),
address(GHO_TOKEN),
IAaveIncentivesController(address(0)),
18,
tokenName,
tokenSymbol,
empty
);
}

function testReInitRevert() public {
string memory tokenName = 'GHO Stable Debt';
string memory tokenSymbol = 'GhoStaDebt';
bytes memory empty;

vm.expectRevert(bytes('Contract instance has already been initialized'));
GHO_STABLE_DEBT_TOKEN.initialize(
IPool(address(POOL)),
address(GHO_TOKEN),
IAaveIncentivesController(address(0)),
18,
tokenName,
tokenSymbol,
empty
);
}

function testUnderlying() public {
assertEq(
GHO_STABLE_DEBT_TOKEN.UNDERLYING_ASSET_ADDRESS(),
address(GHO_TOKEN),
'Underlying should match token'
);
}

function testTransferRevert() public {
vm.startPrank(alice);
vm.expectRevert(bytes(Errors.OPERATION_NOT_SUPPORTED));
GHO_STABLE_DEBT_TOKEN.transfer(carlos, 1);
}

function testTransferFromRevert() public {
vm.startPrank(alice);
vm.expectRevert(bytes(Errors.OPERATION_NOT_SUPPORTED));
GHO_STABLE_DEBT_TOKEN.transferFrom(alice, carlos, 1);
}

function testApproveRevert() public {
vm.startPrank(alice);
vm.expectRevert(bytes(Errors.OPERATION_NOT_SUPPORTED));
GHO_STABLE_DEBT_TOKEN.approve(carlos, 1);
}

function testIncreaseAllowanceRevert() public {
vm.startPrank(alice);
vm.expectRevert(bytes(Errors.OPERATION_NOT_SUPPORTED));
GHO_STABLE_DEBT_TOKEN.increaseAllowance(carlos, 1);
}

function testDecreaseAllowanceRevert() public {
vm.startPrank(alice);
vm.expectRevert(bytes(Errors.OPERATION_NOT_SUPPORTED));
GHO_STABLE_DEBT_TOKEN.decreaseAllowance(carlos, 1);
}

function testAllowanceRevert() public {
vm.startPrank(alice);
vm.expectRevert(bytes(Errors.OPERATION_NOT_SUPPORTED));
GHO_STABLE_DEBT_TOKEN.allowance(carlos, alice);
}

function testPrincipalBalanceOfZero() public {
assertEq(GHO_STABLE_DEBT_TOKEN.principalBalanceOf(alice), 0, 'Unexpected principal balance');
assertEq(GHO_STABLE_DEBT_TOKEN.principalBalanceOf(bob), 0, 'Unexpected principal balance');
assertEq(GHO_STABLE_DEBT_TOKEN.principalBalanceOf(carlos), 0, 'Unexpected principal balance');
}

function testMintRevert() public {
vm.prank(address(POOL));
vm.expectRevert(bytes(Errors.OPERATION_NOT_SUPPORTED));
GHO_STABLE_DEBT_TOKEN.mint(alice, alice, 0, 0);
}

function testMintByOther() public {
vm.startPrank(alice);
vm.expectRevert(bytes(Errors.CALLER_MUST_BE_POOL));
GHO_STABLE_DEBT_TOKEN.mint(alice, alice, 0, 0);
}

function testBurnRevert() public {
vm.prank(address(POOL));
vm.expectRevert(bytes(Errors.OPERATION_NOT_SUPPORTED));
GHO_STABLE_DEBT_TOKEN.burn(alice, 0);
}

function testBurnByOther() public {
vm.startPrank(alice);
vm.expectRevert(bytes(Errors.CALLER_MUST_BE_POOL));
GHO_STABLE_DEBT_TOKEN.burn(alice, 0);
}

function testGetAverageStableRateZero() public {
uint256 result = GHO_STABLE_DEBT_TOKEN.getAverageStableRate();
assertEq(result, 0, 'Unexpected stable rate');
}

function testGetUserLastUpdatedZero() public {
assertEq(GHO_STABLE_DEBT_TOKEN.getUserLastUpdated(alice), 0, 'Unexpected stable rate');
assertEq(GHO_STABLE_DEBT_TOKEN.getUserLastUpdated(bob), 0, 'Unexpected stable rate');
assertEq(GHO_STABLE_DEBT_TOKEN.getUserLastUpdated(carlos), 0, 'Unexpected stable rate');
}

function testGetUserStableRateZero() public {
assertEq(GHO_STABLE_DEBT_TOKEN.getUserStableRate(alice), 0, 'Unexpected stable rate');
assertEq(GHO_STABLE_DEBT_TOKEN.getUserStableRate(bob), 0, 'Unexpected stable rate');
assertEq(GHO_STABLE_DEBT_TOKEN.getUserStableRate(carlos), 0, 'Unexpected stable rate');
}

function testGetUserBalanceZero() public {
assertEq(GHO_STABLE_DEBT_TOKEN.balanceOf(alice), 0, 'Unexpected stable rate');
assertEq(GHO_STABLE_DEBT_TOKEN.balanceOf(bob), 0, 'Unexpected stable rate');
assertEq(GHO_STABLE_DEBT_TOKEN.balanceOf(carlos), 0, 'Unexpected stable rate');
}

function testGetSupplyDataZero() public {
(
uint256 totalSupply,
uint256 calcTotalSupply,
uint256 avgRate,
uint40 timestamp
) = GHO_STABLE_DEBT_TOKEN.getSupplyData();
assertEq(totalSupply, 0, 'Unexpected total supply');
assertEq(calcTotalSupply, 0, 'Unexpected total supply');
assertEq(avgRate, 0, 'Unexpected average rate');
assertEq(timestamp, 0, 'Unexpected timestamp');
}

function testGetTotalSupplyAvgRateZero() public {
(uint256 calcTotalSupply, uint256 avgRate) = GHO_STABLE_DEBT_TOKEN.getTotalSupplyAndAvgRate();
assertEq(calcTotalSupply, 0, 'Unexpected total supply');
assertEq(avgRate, 0, 'Unexpected average rate');
}

function testTotalSupplyZero() public {
uint256 result = GHO_STABLE_DEBT_TOKEN.totalSupply();
assertEq(result, 0, 'Unexpected total supply');
}

function testTotalSupplyLastUpdatedZero() public {
uint40 result = GHO_STABLE_DEBT_TOKEN.getTotalSupplyLastUpdated();
assertEq(result, 0, 'Unexpected timestamp');
}
}

0 comments on commit dc8b27a

Please sign in to comment.