add contracts for token and crowdsale, and solidity tests

contracts/SampleToken.sol

@@ -30,4 +30,13 @@ contract SampleToken is MintableToken {
     return super.mint(_to, _amount);
   }
 
+  function transfer(address _to, uint256 _value) public returns (bool) {
+    require(jurisdiction.hasAttribute(_to, 'VALID'));
+    return super.transfer(_to, _value);
+  }
+
+  function transferFrom(address _from, address _to, uint256 _value) public returns (bool) {
+    require(jurisdiction.hasAttribute(_to, 'VALID'));
+    return super.transferFrom(_from, _to, _value);
+  }
 }

test/contracts/SampleCrowdsale.test.js

@@ -15,7 +15,7 @@ contract('SampleCrowdsale', function ([owner, wallet, investor]) {
 
     beforeEach(async function () {
         this.jurisdiction = await Jurisdiction.new();
-        this.jurisdiction.addValidator(owner);
+        await this.jurisdiction.addValidator(owner);
 
         this.token = await SampleToken.new(this.jurisdiction.address);
         this.crowdsale = await SampleCrowdsale.new(
@@ -24,9 +24,6 @@ contract('SampleCrowdsale', function ([owner, wallet, investor]) {
     });
 
     it('should create crowdsale with correct parameters', async function () {
-        this.crowdsale.should.exist;
-        this.token.should.exist;
-
         (await this.crowdsale.rate()).should.be.bignumber.equal(RATE);
         (await this.crowdsale.wallet()).should.be.equal(wallet);
     });

test/contracts/SampleToken.test.js

@@ -0,0 +1,129 @@
+const Jurisdiction = artifacts.require('Jurisdiction');
+const SampleToken = artifacts.require('SampleToken');
+
+const BigNumber = web3.BigNumber;
+
+var should = require('chai')
+      .use(require('chai-as-promised'))
+      .use(require('chai-bignumber')(BigNumber))
+      .should();
+
+contract('SampleToken', function ([owner, investor1, investor2]) {
+
+  beforeEach(async function () {
+    this.jurisdiction = await Jurisdiction.new();
+    await this.jurisdiction.addValidator(owner);
+
+    this.token = await SampleToken.new(this.jurisdiction.address);
+    await this.token.transferOwnership(owner);
+  });
+
+  it('should not accept not validated minting', async function () {
+    const investmentAmount = new BigNumber(web3.toWei(1, 'ether'));
+    await this.token.mint(
+      investor1, investmentAmount,
+      { from: investor1 }
+    ).should.be.rejectedWith('revert');
+    (await this.token.balanceOf(investor1))
+      .should.be.bignumber.equal(0);
+    (await this.token.totalSupply())
+      .should.be.bignumber.equal(0);
+  });
+
+  it('should accept validated minting', async function () {
+    await this.jurisdiction.addAttribute(investor1, 'VALID', 1);
+
+    const investmentAmount = new BigNumber(web3.toWei(1, 'ether'));
+    await this.token.mint(
+      investor1, investmentAmount,
+      { from: owner }
+    ).should.be.fulfilled;
+    (await this.token.balanceOf(investor1))
+      .should.be.bignumber.equal(investmentAmount);
+    (await this.token.totalSupply())
+      .should.be.bignumber.equal(investmentAmount);
+  });
+
+  it('should not accept not validated transfer', async function () {
+    await this.jurisdiction.addAttribute(investor1, 'VALID', 1);
+    const investmentAmount = new BigNumber(web3.toWei(1, 'ether'));
+    await this.token.mint(
+      investor1, investmentAmount,
+      { from: owner }
+    ).should.be.fulfilled;
+
+    await this.token.transfer(
+      investor2, investmentAmount,
+      { from: investor1 }
+    ).should.be.rejectedWith('revert');
+    (await this.token.balanceOf(investor1))
+      .should.be.bignumber.equal(investmentAmount);
+    (await this.token.balanceOf(investor2))
+      .should.be.bignumber.equal(0);
+  });
+
+  it('should accept validated transfer', async function () {
+    await this.jurisdiction.addAttribute(investor1, 'VALID', 1);
+    const investmentAmount = new BigNumber(web3.toWei(1, 'ether'));
+    await this.token.mint(
+      investor1, investmentAmount,
+      { from: owner }
+    ).should.be.fulfilled;
+
+    await this.jurisdiction.addAttribute(investor2, 'VALID', 1);
+    await this.token.transfer(
+      investor2, investmentAmount,
+      { from: investor1 }
+    ).should.be.fulfilled;
+    (await this.token.balanceOf(investor1))
+      .should.be.bignumber.equal(0);
+    (await this.token.balanceOf(investor2))
+      .should.be.bignumber.equal(investmentAmount);
+  });
+
+  it('should not accept not validated transferFrom', async function () {
+    await this.jurisdiction.addAttribute(investor1, 'VALID', 1);
+    const investmentAmount = new BigNumber(web3.toWei(1, 'ether'));
+    await this.token.mint(
+      investor1, investmentAmount,
+      { from: owner }
+    ).should.be.fulfilled;
+    await this.token.approve(
+      owner, investmentAmount,
+      { from: investor1 }
+    ).should.be.fulfilled;
+
+    await this.token.transferFrom(
+      investor1, investor2, investmentAmount,
+      { from: owner }
+    ).should.be.rejectedWith('revert');
+    (await this.token.balanceOf(investor1))
+      .should.be.bignumber.equal(investmentAmount);
+    (await this.token.balanceOf(investor2))
+      .should.be.bignumber.equal(0);
+  });
+
+  it('should accept validated transferFrom', async function () {
+    await this.jurisdiction.addAttribute(investor1, 'VALID', 1);
+    const investmentAmount = new BigNumber(web3.toWei(1, 'ether'));
+    await this.token.mint(
+      investor1, investmentAmount,
+      { from: owner }
+    ).should.be.fulfilled;
+    await this.token.approve(
+      owner, investmentAmount,
+      { from: investor1 }
+    ).should.be.fulfilled;
+
+    await this.jurisdiction.addAttribute(investor2, 'VALID', 1);
+    await this.token.transferFrom(
+      investor1, investor2, investmentAmount,
+      { from: owner }
+    ).should.be.fulfilled;
+    (await this.token.balanceOf(investor1))
+      .should.be.bignumber.equal(0);
+    (await this.token.balanceOf(investor2))
+      .should.be.bignumber.equal(investmentAmount);
+  });
+
+});