Start first cycle when first time enable prod mode

buckyos · May 14, 2024 · 355a5d8 · 355a5d8
1 parent 663ae5c
commit 355a5d8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/contracts/exchange.sol b/contracts/exchange.sol
@@ -77,7 +77,7 @@ contract Exchange is Initializable, UUPSUpgradeable, OwnableUpgradeable {
 
         test_mode = true;
 
-        _newCycle();
+        //_newCycle();
     }
 
     function getCircleBalance(uint256 circle) public view returns (uint256) {
@@ -207,6 +207,11 @@ contract Exchange is Initializable, UUPSUpgradeable, OwnableUpgradeable {
 
         // 将剩余的DMC还给owner
         DMC(dmcToken).transfer(msg.sender, DMC(dmcToken).balanceOf(address(this)));
+
+        // 如果是首次开启生产模式，初始化第一个周期
+        if (current_circle == 0) {
+            _newCycle();
+        }
     }
 
     function addFreeDMCTestMintBalance(uint256 amount) public onlyOwner testEnabled {

diff --git a/test/test_exchange.ts b/test/test_exchange.ts
@@ -58,6 +58,8 @@ describe("Exchange", function () {
     })
 
     it("enable prod mode", async () => {
+        // 模拟在开启生产模式之前过了很久，超过一个周期的情况
+        mine(20, {interval: 1000});
         await (await exchange.enableProdMode()).wait();
 
         await expect(exchange.enableProdMode()).to.revertedWith("contract not in test mode");