fix test mocks

contracts/mocks/TokenControllerMock.sol

@@ -4,7 +4,6 @@ pragma solidity ^0.5.0;
 
 import "../abstract/MasterAware.sol";
 import "../modules/token/NXMToken.sol";
-import "hardhat/console.sol";
 
 contract TokenControllerMock is MasterAware {
 

contracts/modules/cover/Cover.sol

@@ -7,7 +7,6 @@ import "../../interfaces/IPool.sol";
 import "../../abstract/MasterAwareV2.sol";
 import "../../interfaces/IMemberRoles.sol";
 import "../../interfaces/ICoverNFT.sol";
-import "hardhat/console.sol";
 import "../../interfaces/IMCR.sol";
 import "../../interfaces/ITokenController.sol";
 
@@ -133,8 +132,6 @@ contract Cover is ICover, MasterAwareV2 {
     require(premiumInAsset <= maxPremiumInAsset, "Cover: Price exceeds maxPremiumInAsset");
     retrievePayment(premiumInAsset, payoutAsset);
 
-
-    console.log("addressTC", address(tokenController()));
     tokenController().mint(owner, totalPremiumInNXM / 10);
 
     return coverId;

test/unit/Cover/setup.js

@@ -21,15 +21,18 @@ async function setup () {
   const MemberRolesMock = artifacts.require('MemberRolesMock');
   const CoverNFT = artifacts.require('CoverNFT');
   const TokenController = artifacts.require('TokenControllerMock');
+  const NXMToken = await artifacts.require('NXMTokenMock');
 
   const master = await MasterMock.new();
   const dai = await ERC20Mock.new();
   const stETH = await ERC20Mock.new();
   const memberRoles = await MemberRolesMock.new();
   const tokenController = await TokenController.new();
-
+  const nxm = await NXMToken.new();
   const cover = await Cover.new();
 
+  await master.setTokenAddress(nxm.address);
+
   const coverNFT = await CoverNFT.new('NexusMutual Cover', 'NXMC', cover.address);
   await cover.initialize(coverNFT.address);
 
@@ -61,7 +64,6 @@ async function setup () {
   await master.setLatestAddress(hex('QD'), quotationData.address);
   await master.setLatestAddress(hex('MR'), memberRoles.address);
   await master.setLatestAddress(hex('CO'), cover.address);
-
   await master.setLatestAddress(hex('TC'), tokenController.address);
 
   for (const member of accounts.members) {
@@ -83,8 +85,11 @@ async function setup () {
     await master.enrollGovernance(governanceContract);
   }
 
-  await cover.changeMasterAddress(master.address);
-  await cover.changeDependentContractAddress();
+  for (const contract of [cover, tokenController]) {
+    await contract.changeMasterAddress(master.address);
+    await contract.changeDependentContractAddress();
+    await master.enrollInternal(contract.address);
+  }
 
   // add products