restrict user flows pre-genesis

Joey Santoro · Joey Santoro · commit 51e22dc7ff5d · 2021-02-10T15:57:47.000-08:00
diff --git a/contracts/bondingcurve/BondingCurve.sol b/contracts/bondingcurve/BondingCurve.sol
@@ -77,7 +77,7 @@ abstract contract BondingCurve is IBondingCurve, OracleRef, PCVSplitter, Timed {
     }
 
     /// @notice batch allocate held PCV
-    function allocate() external override {
+    function allocate() external override postGenesis {
         uint256 amount = getTotalPCVHeld();
         require(amount != 0, "BondingCurve: No PCV held");
 
diff --git a/test/bondingcurve/EthBondingCurve.test.js b/test/bondingcurve/EthBondingCurve.test.js
@@ -421,6 +421,19 @@ describe('EthBondingCurve', function () {
   });
 
   describe('Allocate', function() {
+
+    describe('Pre Launch', function() {
+      beforeEach(async function() {
+        this.core = await getCore(false);
+        await this.core.init({from: governorAddress});
+        await this.bondingCurve.setCore(this.core.address, {from: governorAddress});
+      });
+
+      it('reverts', async function() {
+        await expectRevert(this.bondingCurve.allocate({from: keeperAddress}), "CoreRef: Still in Genesis Period"); 
+      });
+    });
+
     describe('No Purchase', function() {
       it('reverts', async function() {
         await expectRevert(this.bondingCurve.allocate({from: keeperAddress}), "BondingCurve: No PCV held"); 

Original file line number	Diff line number	Diff line change
`@@ -77,7 +77,7 @@ abstract contract BondingCurve is IBondingCurve, OracleRef, PCVSplitter, Timed {`
`77`	`77`	`}`
`78`	`78`
`79`	`79`	`/// @notice batch allocate held PCV`
`80`		`- function allocate() external override {`
	`80`	`+ function allocate() external override postGenesis {`
`81`	`81`	`uint256 amount = getTotalPCVHeld();`
`82`	`82`	`require(amount != 0, "BondingCurve: No PCV held");`
`83`	`83`