✅ basic tess for createAuction and auctionReady

Kwenta · Oct 17, 2024 · 0ea697e · 0ea697e
1 parent 1e0be95
commit 0ea697e
Show file tree

Hide file tree

Showing 2 changed files with 70 additions and 3 deletions.
diff --git a/test/KSXVault.t.sol b/test/KSXVault.t.sol
@@ -16,7 +16,7 @@ contract KSXVaultTest is Bootstrap {
 
         depositToken = new MockERC20("Deposit Token", "DT");
         stakingRewards = new MockStakingRewards(address(depositToken));
-        initializeLocal(address(depositToken),  address(stakingRewards), DECIMAL_OFFSET);
+        initializeLocal(address(depositToken),  address(stakingRewards), DECIMAL_OFFSET, 0);
 
         depositToken.mint(alice, 10 ether);
         depositToken.mint(bob, 10 ether);
@@ -100,4 +100,71 @@ contract KSXVaultTest is Bootstrap {
         vm.stopPrank();
     }
 
+    function test_auctionReady() public {
+        assertEq(ksxVault.auctionReady(), false);
+        assertEq(block.timestamp, 1);
+        vm.warp(block.timestamp + 1 weeks - 2);
+        assertEq(ksxVault.auctionReady(), false);
+        vm.warp(block.timestamp + 1);
+        assertEq(ksxVault.auctionReady(), true);
+    }
+
+    function test_auctionReady_next_week() public {
+        assertEq(ksxVault.auctionReady(), false);
+        assertEq(block.timestamp, 1);
+        vm.warp(block.timestamp + 1 weeks - 2);
+        assertEq(ksxVault.auctionReady(), false);
+        vm.warp(block.timestamp + 1);
+        assertEq(ksxVault.auctionReady(), true);
+
+        ksxVault.createAuction(address(this), address(0), address(0), 100, 100);
+
+        vm.warp(block.timestamp + 1 weeks - 1);
+        assertEq(ksxVault.auctionReady(), false);
+        vm.warp(block.timestamp + 1);
+        assertEq(ksxVault.auctionReady(), true);
+    }
+
+    function test_auctionReady_offset() public {
+        vm.warp(block.timestamp + 2 weeks);
+        initializeLocal(address(depositToken),  address(stakingRewards), DECIMAL_OFFSET, 1);
+        ksxVault.createAuction(address(this), address(0), address(0), 100, 100);
+        assertEq(ksxVault.auctionReady(), false);
+        assertEq(block.timestamp, 2 weeks + 1);
+        vm.warp(block.timestamp + 1 days - 2);
+        assertEq(ksxVault.auctionReady(), false);
+        vm.warp(block.timestamp + 1);
+        assertEq(ksxVault.auctionReady(), true);
+    }
+
+    function test_auctionReady_offset_next_week() public {
+        vm.warp(block.timestamp + 2 weeks);
+        initializeLocal(address(depositToken),  address(stakingRewards), DECIMAL_OFFSET, 1);
+        ksxVault.createAuction(address(this), address(0), address(0), 100, 100);
+        assertEq(ksxVault.auctionReady(), false);
+        assertEq(block.timestamp, 2 weeks + 1);
+        vm.warp(block.timestamp + 1 days - 2);
+        assertEq(ksxVault.auctionReady(), false);
+        vm.warp(block.timestamp + 1);
+        assertEq(ksxVault.auctionReady(), true);
+
+        ksxVault.createAuction(address(this), address(0), address(0), 100, 100);
+
+        vm.warp(block.timestamp + 1 weeks - 1);
+        assertEq(ksxVault.auctionReady(), false);
+        vm.warp(block.timestamp + 1);
+        assertEq(ksxVault.auctionReady(), true);
+    }
+
+    function test_createAuction() public {
+        vm.warp(block.timestamp + 1 weeks);
+        ksxVault.createAuction(address(this), address(0), address(0), 100, 100);
+        assertEq(ksxVault.lastAuctionStartTime(), block.timestamp);
+    }
+
+    function test_createAuction_AuctionNotReady() public {
+        vm.expectRevert(KSXVault.AuctionNotReady.selector);
+        ksxVault.createAuction(address(this), address(0), address(0), 100, 100);
+    }
+
 }
diff --git a/test/utils/Bootstrap.sol b/test/utils/Bootstrap.sol
@@ -31,9 +31,9 @@ contract Bootstrap is Test, Constants {
     address constant alice = address(0xAAAA);
     address constant bob = address(0xBBBB);
 
-    function initializeLocal(address _token, address _stakingRewards, uint8 _decimalsOffset) internal {
+    function initializeLocal(address _token, address _stakingRewards, uint8 _decimalsOffset, uint256 _daysToOffsetBy) internal {
         BootstrapLocal bootstrap = new BootstrapLocal();
-        (address ksxVaultAddress) = bootstrap.init(_token, _stakingRewards, _decimalsOffset, 0);
+        (address ksxVaultAddress) = bootstrap.init(_token, _stakingRewards, _decimalsOffset, _daysToOffsetBy);
 
         decimalsOffset = _decimalsOffset;
         TOKEN = IERC20(_token);