Fix SimpleRestrictedRuleEngine and add test case

atpar · Oct 8, 2020 · f70f54c · f70f54c
1 parent 810f1ae
commit f70f54c
Show file tree

Hide file tree

Showing 5 changed files with 875 additions and 4 deletions.
diff --git a/packages/ap-contracts/build/contracts/SimpleRestrictedRuleEngine.json b/packages/ap-contracts/build/contracts/SimpleRestrictedRuleEngine.json
diff --git a/packages/ap-contracts/contracts/FDT/CMTA20FDT/SimpleRestrictedRuleEngine.sol b/packages/ap-contracts/contracts/FDT/CMTA20FDT/SimpleRestrictedRuleEngine.sol
@@ -277,6 +277,11 @@ contract SimpleRestrictedRuleEngine is IRuleEngine, Whitelistable, Restrictable
     string public constant UNKNOWN_ERROR = "Unknown Error Code";
 
 
+    constructor(address owner) public {
+        super.__Ownable_init();
+        transferOwnership(owner);
+    }
+
     function setRules(IRule[] calldata rules) external override onlyOwner {
         revert("Can not set any additional rules");
     }

diff --git a/packages/ap-contracts/test/FDT/TestCMTA20FDT.js b/packages/ap-contracts/test/FDT/TestCMTA20FDT.js
@@ -2,7 +2,9 @@ const buidlerRuntime = require('@nomiclabs/buidler');
 const { BN, ether, shouldFail } = require('openzeppelin-test-helpers');
 
 const { expectEvent, ZERO_ADDRESS } = require('../helper/utils');
-const { getSnapshotTaker, deployPaymentToken, deployCMTA20FDT, deployRuleEngineMock } = require('../helper/setupTestEnvironment');
+const {
+  getSnapshotTaker, deployPaymentToken, deployCMTA20FDT, deployRuleEngineMock
+} = require('../helper/setupTestEnvironment');
 
 
 describe('CMTA20', () => {