@daostack/arc/test/organizationregister.js

A platform for building DAOs

const helpers = require('./helpers');

const OrganizationRegister = artifacts.require('./OrganizationRegister.sol');
const ERC20Mock = artifacts.require('./test/ERC20Mock.sol');
const DaoCreator = artifacts.require("./DaoCreator.sol");
const ControllerCreator = artifacts.require("./ControllerCreator.sol");
const DAOTracker = artifacts.require("./DAOTracker.sol");

class OrganizationRegisterParams {
  constructor() {
  }
}

const setupOrganizationRegisterParams = async function(
                                            organizationRegister,
                                            tokenAddress,
                                            beneficiary,

                                            ) {
  var organizationRegisterParams = new OrganizationRegisterParams();
  organizationRegisterParams.votingMachine = await helpers.setupAbsoluteVote();
  await organizationRegister.setParameters(tokenAddress,13,beneficiary);
  organizationRegisterParams.paramsHash = await organizationRegister.getParametersHash(tokenAddress,13,beneficiary);
  return organizationRegisterParams;
};

const setup = async function (accounts) {
   var testSetup = new helpers.TestSetup();
   testSetup.fee = 10;
   testSetup.standardTokenMock = await ERC20Mock.new(accounts[1],100);
   testSetup.organizationRegister = await OrganizationRegister.new();
   var controllerCreator = await ControllerCreator.new();
   var daoTracker = await DAOTracker.new();
   testSetup.daoCreator = await DaoCreator.new(controllerCreator.address,daoTracker.address);
   testSetup.org = await helpers.setupOrganization(testSetup.daoCreator,accounts[0],1000,1000);
   testSetup.organizationRegisterParams= await setupOrganizationRegisterParams(testSetup.organizationRegister,testSetup.standardTokenMock.address,accounts[2]);
   var permissions = "0x00000000";
   await testSetup.daoCreator.setSchemes(testSetup.org.avatar.address,[testSetup.organizationRegister.address],[testSetup.organizationRegisterParams.paramsHash],[permissions],"metaData");

   return testSetup;
};

contract('OrganizationRegister',accounts => {
  before(function() {
     helpers.etherForEveryone(accounts);
  });

   it("setParameters", async() => {
     var organizationRegister = await OrganizationRegister.new();
     await organizationRegister.setParameters(accounts[3],13,accounts[2]);
     var paramHash = await organizationRegister.getParametersHash(accounts[3],13,accounts[2]);
     var parameters = await organizationRegister.parameters(paramHash);
     assert.equal(parameters[1],accounts[3]);
     assert.equal(parameters[0],13);
     assert.equal(parameters[2],accounts[2]);
     });


     it("addOrPromoteAddress add and promote ", async function() {
       var testSetup = await setup(accounts);
       var record = accounts[4];
       var amount = 13;


       await testSetup.standardTokenMock.approve(testSetup.organizationRegister.address,100,{from:accounts[1]});

       var tx = await testSetup.organizationRegister.addOrPromoteAddress(testSetup.org.avatar.address,
                                                                     record,
                                                                     amount,{from:accounts[1]});
       assert.equal(tx.logs.length, 2);
       assert.equal(tx.logs[0].event, "OrgAdded");
       assert.equal(tx.logs[1].event, "Promotion");
       var  registery = await helpers.getValueFromLogs(tx, '_registry');
       var org = await helpers.getValueFromLogs(tx, '_org');
       var registeryAmount  = await testSetup.organizationRegister.organizationsRegistry(registery,org);
       assert.equal(amount,registeryAmount);
       //now try to promote.
       tx = await testSetup.organizationRegister.addOrPromoteAddress(testSetup.org.avatar.address,
                                                                     record,
                                                                     1,{from:accounts[1]});
       assert.equal(tx.logs.length, 1);
       assert.equal(tx.logs[0].event, "Promotion");
       registery = await helpers.getValueFromLogs(tx, '_registry');
       org = await helpers.getValueFromLogs(tx, '_org');
       registeryAmount  = await testSetup.organizationRegister.organizationsRegistry(registery,org);
       assert.equal(amount+1,registeryAmount);
      });

      it("addOrPromoteAddress add without enough fee should fail ", async function() {
        var testSetup = await setup(accounts);
        var record = accounts[4];
        var amount = 12;


        await testSetup.standardTokenMock.approve(testSetup.organizationRegister.address,100,{from:accounts[1]});

        try{
        await testSetup.organizationRegister.addOrPromoteAddress(testSetup.org.avatar.address,
                                                                      record,
                                                                      amount,{from:accounts[1]});
        assert(false,"addOrPromoteAddress should  fail - due to amount<fee !");
        }catch(ex){
           helpers.assertVMException(ex);
        }
       });
});