compliance functions done except subscribe and unsubscribe

Author: davekay100

src/contract_wrappers/Compliance.js

@@ -168,21 +168,21 @@ export default class Compliance extends ContractWrapper {
 
   /**
    * Set an STO contract to be stored in the offerings mapping in Compliance.sol
-   * @param issuerAddress  Address of the offering contract owner or auditor
+   * @param stoDeveloperAddress  Address of the creator of the STO
    * @param stoAddress     Address of the STO contract deployed over the network
    * @param fee            Fee to be paid in poly to use that contract
    * @param vestingPeriod  Number of days investor binded to hold the Security token
    * @param quorum         Minimum percent of shareholders which need to vote to freeze
    */
   async setSTO(
-    issuerAddress: string,
+    stoDeveloperAddress: string,
     stoAddress: string,
     fee: BigNumber,
     vestingPeriod: BigNumber,
     quorum: BigNumber,
   ) {
     await this._contract.setSTO(stoAddress, fee, vestingPeriod, quorum, {
-      from: issuerAddress,
+      from: stoDeveloperAddress,
       gas: 200000,
     });
   }
@@ -306,4 +306,22 @@ export default class Compliance extends ContractWrapper {
       expires: template[3],
     };
   }
+
+  /**
+   * Returns all Template proposals
+   * @return An array of addresses
+   */
+  async getAllTemplateProposals(securityTokenAddress: string): Promise<Array<string>> {
+    return (await this._contract.getAllTemplateProposals.call(securityTokenAddress));
+  }
+
+    /**
+   * Returns all STO proposal addresses
+   * @return An array of addresses
+   */
+  async getAllOfferingProposals(securityTokenAddress: string): Promise<Array<string>> {
+    return (await this._contract.getAllOfferingProposals.call(securityTokenAddress));
+  }
+
+
 }

test/Compliance_test.js

@@ -13,9 +13,11 @@ import {
   makeSecurityTokenRegistrar,
   makeTemplateWithFinalized,
   makeSecurityTokenThroughRegistrar,
+  makeSelectedTemplateForSecurityToken,
+  makeSecurityTokenOffering,
 } from './util/make_examples';
 import { makeWeb3Wrapper } from './util/web3';
-import { fakeAddress } from './util/fake';
+import { fakeAddress, fakeBytes32 } from './util/fake';
 
 const { assert } = chai;
 
@@ -67,6 +69,14 @@ describe('Compliance wrapper', () => {
       new BigNumber(10).toPower(18).times(100000),
       accounts[2],
     );
+    await polyToken.generateNewTokens(
+      new BigNumber(10).toPower(18).times(100000),
+      accounts[3],
+    );
+    await polyToken.generateNewTokens(
+      new BigNumber(10).toPower(18).times(100000),
+      accounts[4],
+    );
   });
 
   it('createTemplate', async () => {
@@ -81,7 +91,7 @@ describe('Compliance wrapper', () => {
     assert.isAbove(templateAddress.length, 0);
   });
 
-  it('proposeTemplate, templateReputation, getTemplateAddressByProposal, cancelTemplateProposal', async () => {
+  it('proposeTemplate, templateReputation, getTemplateAddressByProposal, cancelTemplateProposal, getAllTemplateProposals', async () => {
     await makeKYCProvider(customers, accounts[1]);
     await makeLegalDelegate(polyToken, customers, accounts[1], accounts[2]);
     const templateAddress = await makeTemplateWithFinalized(
@@ -93,7 +103,7 @@ describe('Compliance wrapper', () => {
     // Propose Template
     await compliance.proposeTemplate(accounts[2], securityToken.address, templateAddress);
     const logs = await compliance.getLogs('LogNewTemplateProposal', {}, { fromBlock: 1 });
-    assert.equal(logs[0].args._template, templateAddress, 'Template address does not match the logged version');
+    assert.equal(logs[0].args._template, templateAddress, 'Template address does not match the logged version');
 
     // Reputation
     let templateReputation = await compliance.getTemplateReputation(templateAddress);
@@ -103,8 +113,12 @@ describe('Compliance wrapper', () => {
     const address = await compliance.getTemplateAddressByProposal(securityToken.address, 0)
     assert.equal(address, templateAddress, 'Proposal returned the wrong template address');
 
+    //Test all templateProposals
+    let arrayOfTemplates = await compliance.getAllTemplateProposals(securityToken.address)
+    assert.equal(arrayOfTemplates[0], templateAddress, 'Template address does not match the getter function return');
+
     // Cancel Proposal
-    await compliance.cancelTemplateProposal(accounts[2],securityToken.address, 0);
+    await compliance.cancelTemplateProposal(accounts[2], securityToken.address, 0);
 
     const addressShouldBeZero = await compliance.getTemplateAddressByProposal(securityToken.address, 0)
     assert.equal(addressShouldBeZero, 0, 'Proposal returned the wrong template address');
@@ -113,25 +127,103 @@ describe('Compliance wrapper', () => {
 
   });
 
-  it('setSTO', async () => {
-    await makeKYCProvider(customers, accounts[1]);
-
-    await compliance.setSTO(
-      accounts[0],
-      fakeAddress,
-      new BigNumber(10),
-      new BigNumber(9888888),
-      new BigNumber(20),
-    );
-  });
 
   it('getMinimumVestingPeriod', async () => {
     let minimum = await compliance.getMinimumVestingPeriod();
     assert.equal(minimum, 60 * 60 * 24 * 100, "Does not equal 100 days, when it should")
   })
 
-  // it('getTemplateReputation', async() => {
+  //so we need to have a securityToken actually created through STRegistrar
+  //and so me of the stuff has to match up
+  //then we have an actual one in offeringProposals
+  it('setSTO, proposeSTO, cancleSTO, getSTOProposal, getSTOAddressByProposal, getAllOfferingProposals', async () => {
+    const owner = accounts[0];
+    const legalDelegate = accounts[2];
+    const kycProvider = accounts[1];
+
+    // STO variables
+    const auditor = accounts[4];
+    const startTime = new BigNumber(
+      Math.floor(new Date().getTime() / 1000)
+    ).plus(200);
+
+    const endTime = new BigNumber(Math.floor(new Date().getTime() / 1000)).plus(
+      2592000,
+    ); // 1 Month duration
+
+    await makeKYCProvider(customers, kycProvider);
+
+    await makeLegalDelegate(polyToken, customers, kycProvider, legalDelegate);
+    const templateAddress = await makeTemplateWithFinalized(
+      compliance,
+      kycProvider,
+      legalDelegate,
+    );
+
+    await makeSelectedTemplateForSecurityToken(
+      securityToken,
+      compliance,
+      polyToken,
+      owner,
+      legalDelegate,
+      kycProvider,
+      fakeBytes32,
+      templateAddress,
+    );
+
+    await polyToken.approve(auditor, customers.address, 100);
+
+    await customers.verifyCustomer(
+      kycProvider,
+      auditor,
+      'US',
+      'CA',
+      'investor',
+      true,
+      new BigNumber(Math.floor(new Date().getTime() / 1000)).plus(10000),
+    );
+
+    //this make example does setSTO and proposeSTO, and we will test below
+    const offering = await makeSecurityTokenOffering(
+      web3Wrapper,
+      polyToken,
+      securityToken,
+      compliance,
+      auditor,
+      startTime,
+      endTime,
+    );
+
+    //to confirm setSTO, we need to check offerings for the msg.sender addr
+    //which is using getOfferingByProposal
+    //in setSTO we
+    let getSTO = await compliance.getSTOProposal(securityToken.address, 0)
+    assert.equal(getSTO.auditorAddress, auditor, "Auditor address not read properly");
+
+    //to confirm proposeSTO worked, we check offeringProposals, which is getSTOAddress by proposal
+    let getSTOProposal = await compliance.getSTOAddressByProposal(securityToken.address, 0)
+    assert.equal(getSTOProposal, getSTO.stoContractAddress, "STO address not read properly");
+
+    let getAllOfferings = await compliance.getAllOfferingProposals(securityToken.address, 0)
+    assert.equal(getAllOfferings[0], getSTO.stoContractAddress, "STO array of addresses not read properly");
+
+    // Cancel Proposal
+    await compliance.cancelSTOProposal(auditor, securityToken.address, 0);
+
+    const addressShouldBeZero = await compliance.getSTOAddressByProposal(securityToken.address, 0)
+    console.log(addressShouldBeZero)
+    assert.equal(addressShouldBeZero, 0, 'Proposal did not return zero, which it should have for being cancelled');
+
+
+
+  })
+
+  it('getAllOfferingProposals', async () => {
+
+  })
+
+  it('subscribe, unsubscribe, unsubscribeAll', async () => {
 
-  // })
+  })
 
 });