diff --git a/test/token-voting/LlamaTokenVotingTestSetup.sol b/test/token-voting/LlamaTokenVotingTestSetup.sol index dbb82e7..627ba4b 100644 --- a/test/token-voting/LlamaTokenVotingTestSetup.sol +++ b/test/token-voting/LlamaTokenVotingTestSetup.sol @@ -4,9 +4,6 @@ pragma solidity ^0.8.23; import {Test, console2} from "forge-std/Test.sol"; import {Vm} from "forge-std/Vm.sol"; -import {ERC20Votes} from "@openzeppelin/token/ERC20/extensions/ERC20Votes.sol"; -import {ERC721Votes} from "@openzeppelin/token/ERC721/extensions/ERC721Votes.sol"; - import {MockERC20Votes} from "test/mock/MockERC20Votes.sol"; import {MockERC721Votes} from "test/mock/MockERC721Votes.sol"; import {LlamaPeripheryTestSetup} from "test/LlamaPeripheryTestSetup.sol"; @@ -24,11 +21,9 @@ contract LlamaTokenVotingTestSetup is LlamaPeripheryTestSetup, DeployLlamaTokenV uint256 public constant ERC721_MIN_APPROVAL_PCT = 1; uint256 public constant ERC721_MIN_DISAPPROVAL_PCT = 1; - // Tokens - MockERC20Votes public mockERC20Votes; - ERC20Votes public erc20VotesToken; - MockERC721Votes public mockERC721Votes; - ERC721Votes public erc721VotesToken; + // Votes Tokens + MockERC20Votes public erc20VotesToken; + MockERC721Votes public erc721VotesToken; function setUp() public virtual override { LlamaPeripheryTestSetup.setUp(); @@ -37,20 +32,18 @@ contract LlamaTokenVotingTestSetup is LlamaPeripheryTestSetup, DeployLlamaTokenV DeployLlamaTokenVotingFactory.run(); // Deploy the ERC20 and ERC721 tokens. - mockERC20Votes = new MockERC20Votes(); - erc20VotesToken = ERC20Votes(address(mockERC20Votes)); - mockERC721Votes = new MockERC721Votes(); - erc721VotesToken = ERC721Votes(address(mockERC721Votes)); + erc20VotesToken = new MockERC20Votes(); + erc721VotesToken = new MockERC721Votes(); // Mint tokens to core team members. - mockERC20Votes.mint(coreTeam1, 100); - mockERC20Votes.mint(coreTeam2, 100); - mockERC20Votes.mint(coreTeam3, 100); - mockERC20Votes.mint(coreTeam4, 100); - mockERC721Votes.mint(coreTeam1, 0); - mockERC721Votes.mint(coreTeam2, 1); - mockERC721Votes.mint(coreTeam3, 2); - mockERC721Votes.mint(coreTeam4, 3); + erc20VotesToken.mint(coreTeam1, 100); + erc20VotesToken.mint(coreTeam2, 100); + erc20VotesToken.mint(coreTeam3, 100); + erc20VotesToken.mint(coreTeam4, 100); + erc721VotesToken.mint(coreTeam1, 0); + erc721VotesToken.mint(coreTeam2, 1); + erc721VotesToken.mint(coreTeam3, 2); + erc721VotesToken.mint(coreTeam4, 3); // Mine block so that the ERC20 and ERC721 supply will be available when doing a past timestamp check. mineBlock();