Check old factory

cybernetics · Aug 26, 2020 · 6ac37ef · 6ac37ef
1 parent c79ae5e
commit 6ac37ef
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/contracts/Migrator.sol b/contracts/Migrator.sol
@@ -6,19 +6,27 @@ import "./uniswapv2/interfaces/IUniswapV2Factory.sol";
 
 contract Migrator {
     address public chef;
+    address public oldFactory;
     IUniswapV2Factory public factory;
     uint256 public notBeforeBlock;
     uint256 public desiredLiquidity = uint256(-1);
 
-    constructor(address _chef, IUniswapV2Factory _factory, uint256 _notBeforeBlock) public {
+    constructor(
+        address _chef,
+        address _oldFactory,
+        IUniswapV2Factory _factory,
+        uint256 _notBeforeBlock
+    ) public {
         chef = _chef;
+        oldFactory = _oldFactory;
         factory = _factory;
         notBeforeBlock = _notBeforeBlock;
     }
 
     function migrate(IUniswapV2Pair orig) public returns (IUniswapV2Pair) {
-        require(msg.sender == chef, "not the chef");
+        require(msg.sender == chef, "not from master chef");
         require(block.number >= notBeforeBlock, "too early to migrate");
+        require(orig.factory() == oldFactory, "not from old factory");
         address token0 = orig.token0();
         address token1 = orig.token1();
         IUniswapV2Pair pair = IUniswapV2Pair(factory.getPair(token0, token1));

diff --git a/test/Migrator.test.js b/test/Migrator.test.js
@@ -16,7 +16,7 @@ contract('Migrator', ([alice, bob, dev, minter]) => {
         this.lp1 = await UniswapV2Pair.at((await this.factory1.createPair(this.weth.address, this.token.address)).logs[0].args.pair);
         this.lp2 = await UniswapV2Pair.at((await this.factory2.createPair(this.weth.address, this.token.address)).logs[0].args.pair);
         this.chef = await MasterChef.new(this.sushi.address, dev, '1000', '0', '100000', { from: alice });
-        this.migrator = await Migrator.new(this.chef.address, this.factory2.address, '0');
+        this.migrator = await Migrator.new(this.chef.address, this.factory1.address, this.factory2.address, '0');
         await this.sushi.transferOwnership(this.chef.address, { from: alice });
         await this.chef.add('100', this.lp1.address, true, { from: alice });
     });