fix: 🐛 CTM example now works!

Author: shuffledex

examples/countTransferManager.ts

@@ -41,9 +41,9 @@ window.addEventListener('load', async () => {
   });
 
   // Register a ticker
-  await polymathAPI.securityTokenRegistry.registerTicker({
+  await polymathAPI.securityTokenRegistry.registerNewTicker({
+    owner: myAddress,
     ticker: ticker!,
-    tokenName: tokenName!,
   });
 
   // Get the st launch fee and approve the security token registry to spend
@@ -54,11 +54,13 @@ window.addEventListener('load', async () => {
   });
 
   // Generate a security token
-  await polymathAPI.securityTokenRegistry.generateSecurityToken({
+  await polymathAPI.securityTokenRegistry.generateNewSecurityToken({
     name: tokenName!,
     ticker: ticker!,
-    details: 'http://',
-    divisible: false,
+    tokenDetails: 'http://',
+    divisible: true,
+    treasuryWallet: myAddress,
+    protocolVersion: '0',
   });
 
   // Create a Security Token Instance
@@ -82,16 +84,17 @@ window.addEventListener('load', async () => {
     names.push(instanceFactory.name());
   });
   const resultNames = await Promise.all(names);
-
-  const finalNames = resultNames.map(name => {
-    return bytes32ToString(name);
-  });
-  const index = finalNames.indexOf(moduleStringName);
+  const index = resultNames.indexOf(moduleStringName);
 
   // Get setup cost
   const factory = await polymathAPI.moduleFactory.getModuleFactory(modules[index]);
   const setupCost = await factory.setupCostInPoly();
 
+  const randomBeneficiaries = [
+    '0x3444444444444444444444444444444444444444',
+    '0x5544444444444444444444444444444444444444',
+  ];
+
   // Call to add count transfer manager module with max 3 holders
   await tickerSecurityTokenInstance.addModule({
     moduleName,
@@ -100,15 +103,14 @@ window.addEventListener('load', async () => {
     budget: setupCost,
     archived: false,
     data: {
-      maxHolderCount: 3,
+      maxHolderCount: randomBeneficiaries.length + 1,
     },
   });
 
   // Get Count TM address and create count transfer manager
   const countTMAddress = (await tickerSecurityTokenInstance.getModulesByName({
     moduleName: ModuleName.CountTransferManager,
   }))[0];
-  console.log(countTMAddress);
   const countTM = await polymathAPI.moduleFactory.getModuleInstance({
     name: ModuleName.CountTransferManager,
     address: countTMAddress,
@@ -122,18 +124,36 @@ window.addEventListener('load', async () => {
     name: ModuleName.GeneralTransferManager,
     address: generalTMAddress,
   });
-//  await generalTM.changeAllowAllTransfers({ allowAllTransfers: true });
 
-  const randomBeneficiary1 = '0x3444444444444444444444444444444444444444';
-  const randomBeneficiary2 = '0x5544444444444444444444444444444444444444';
-  const randomBeneficiary3 = '0x6644444444444444444444444444444444444444';
+  // Add owner address in the whitelist to allow mint tokens
+  await generalTM.modifyKYCData({
+    investor: myAddress,
+    canSendAfter: new Date(),
+    canReceiveAfter: new Date(),
+    expiryTime: new Date(2020, 0),
+    txData: {
+      from: await polymathAPI.getAccount(),
+    },
+  });
 
   // Mint yourself some tokens and make some transfers
-  await tickerSecurityTokenInstance.issue({ investor: myAddress, value: new BigNumber(200), data: '' });
-  await tickerSecurityTokenInstance.transfer({ to: randomBeneficiary1, value: new BigNumber(10) });
-  await tickerSecurityTokenInstance.transfer({ to: randomBeneficiary2, value: new BigNumber(20) });
+  await tickerSecurityTokenInstance.issue({
+    investor: myAddress,
+    value: new BigNumber(200),
+    data: '0x00',
+  });
+
+  // Add beneficiaries address to whitelist
+  await generalTM.modifyKYCDataMulti({
+    investors: randomBeneficiaries,
+    canSendAfter: [new Date(), new Date()],
+    canReceiveAfter: [new Date(), new Date()],
+    expiryTime: [new Date(2020, 0), new Date(2020, 0)],
+  });
+
+  await tickerSecurityTokenInstance.transfer({ to: randomBeneficiaries[0], value: new BigNumber(10) });
+  await tickerSecurityTokenInstance.transfer({ to: randomBeneficiaries[1], value: new BigNumber(20) });
 
-  // We need to increase max holder account to get another transfer through (We are currently at max)
   // Subscribe to event of modify holder count
   await countTM.subscribeAsync({
     eventName: CountTransferManagerEvents.ModifyHolderCount,
@@ -146,13 +166,28 @@ window.addEventListener('load', async () => {
       }
     },
   });
-  await countTM.changeHolderCount({ maxHolderCount: 4 });
-  await tickerSecurityTokenInstance.transfer({ to: randomBeneficiary3, value: new BigNumber(30) });
+
+  // We need to increase max holder account to get another transfer through (We are currently at max)
+  randomBeneficiaries.push('0x6644444444444444444444444444444444444444');
+  await countTM.changeHolderCount({ maxHolderCount: randomBeneficiaries.length + 1 });
+
+  // And adding to our whitelist
+  await generalTM.modifyKYCData({
+    investor: randomBeneficiaries[2],
+    canSendAfter: new Date(),
+    canReceiveAfter: new Date(),
+    expiryTime: new Date(2020, 0),
+    txData: {
+      from: await polymathAPI.getAccount(),
+    },
+  });
+
+  await tickerSecurityTokenInstance.transfer({ to: randomBeneficiaries[2], value: new BigNumber(30) });
 
   // Show the balances of our token holders
-  console.log(await tickerSecurityTokenInstance.balanceOf({ owner: randomBeneficiary1 }));
-  console.log(await tickerSecurityTokenInstance.balanceOf({ owner: randomBeneficiary2 }));
-  console.log(await tickerSecurityTokenInstance.balanceOf({ owner: randomBeneficiary3 }));
+  console.log(await tickerSecurityTokenInstance.balanceOf({ owner: randomBeneficiaries[0] }));
+  console.log(await tickerSecurityTokenInstance.balanceOf({ owner: randomBeneficiaries[1] }));
+  console.log(await tickerSecurityTokenInstance.balanceOf({ owner: randomBeneficiaries[2] }));
   console.log(await tickerSecurityTokenInstance.balanceOf({ owner: myAddress }));
 
   console.log('Funds transferred');