Add test for generating comments for edited contract, move all test contracts to test/test-contracts folder.

Author: MakMuftic

test/index.test.es6

@@ -5,21 +5,32 @@ import { generateCommentsFromText } from '../src/index';
 test('Assert generate comments from text', (t) => {
   t.plan(1);
   let commentedContract =
-    generateCommentsFromText(fs.readFileSync('./test/Metacoin.sol', 'utf-8'));
+    generateCommentsFromText(fs.readFileSync('./test/test-contracts/Metacoin.sol', 'utf-8'));
   t.equal(
     commentedContract,
-    fs.readFileSync('./test/Metacoin.commented.sol', 'utf-8')
+    fs.readFileSync('./test/test-contracts/Metacoin.commented.sol', 'utf-8')
   );
 });
 
 test('Assert generate comments from text second time wont duplicate', (t) => {
   t.plan(1);
   let commentedContract =
-    generateCommentsFromText(fs.readFileSync('./test/Metacoin.sol', 'utf-8'));
+    generateCommentsFromText(fs.readFileSync('./test/test-contracts/Metacoin.sol', 'utf-8'));
   commentedContract =
     generateCommentsFromText(commentedContract, 'utf-8');
   t.equal(
     commentedContract,
-    fs.readFileSync('./test/Metacoin.commented.sol', 'utf-8')
+    fs.readFileSync('./test/test-contracts/Metacoin.commented.sol', 'utf-8')
   );
 });
+
+test('Assert generate comments from edited contract', (t) => {
+    t.plan(1);
+    let commentedContract =
+        generateCommentsFromText(fs.readFileSync('./test/test-contracts/Metacoin.edited.sol', 'utf-8'));
+
+    t.equal(
+        commentedContract,
+        fs.readFileSync('./test/test-contracts/Metacoin.edited.commented.sol', 'utf-8')
+    );
+});

test/lib/contract/contract-file.test.es6

@@ -13,28 +13,28 @@ test('Assert not founding contract file', (t) => {
 
 test('Assert creating new ContractFile instance', (t) => {
   t.plan(1);
-  let contract = new ContractFile('./test/Metacoin.sol');
+  let contract = new ContractFile('./test/test-contracts/Metacoin.sol');
   t.notEqual(contract, null);
 });
 
 test('Assert get filename', (t) => {
   t.plan(1);
-  let contract = new ContractFile('./test/Metacoin.sol');
+  let contract = new ContractFile('./test/test-contracts/Metacoin.sol');
   const filename = contract.getFilename();
   t.equal(filename, 'Metacoin.sol');
 });
 
 test('Assert get path', (t) => {
   t.plan(1);
-  let contract = new ContractFile('./test/Metacoin.sol');
+  let contract = new ContractFile('./test/test-contracts/Metacoin.sol');
   const path = contract.getPath();
-  t.equal(path, './test/Metacoin.sol');
+  t.equal(path, './test/test-contracts/Metacoin.sol');
 });
 
 test('Save file', async(t) => {
   t.plan(1);
-  let contract = new ContractFile('./test/Metacoin.sol');
+  let contract = new ContractFile('./test/test-contracts/Metacoin.sol');
   contract.insertTextBefore('test1\nbla', 'pragma solidity ^0.4.24;');
   await contract.save('/tmp/Metacoin-test.sol');
-  t.notEqual(fs.readFileSync('/tmp/Metacoin-test.sol'), fs.readFileSync('./test/Metacoin.sol'));
+  t.notEqual(fs.readFileSync('/tmp/Metacoin-test.sol'), fs.readFileSync('./test/test-contracts/Metacoin.sol'));
 });

test/lib/contract/contract.test.es6

@@ -65,7 +65,7 @@ test('Insert text before with offset', (t) => {
   contract.insertTextBefore('test1\nbla', 0);
   t.equal(contract.getLineAt(0), 'test1');
   t.equal(contract.getLineAt(1), 'bla');
-  contract.insertTextBefore('offset', 0);
+  contract.insertLinesBefore('offset'.split(), 0);
   t.equal(contract.offset(0), 2);
   t.equal(contract.getLineAt(2), 'offset');
 });

test/Metacoin.commented.sol …st/test-contracts/Metacoin.commented.soltest/Metacoin.commented.sol renamed to test/test-contracts/Metacoin.commented.sol test/Metacoin.commented.sol renamed to test/test-contracts/Metacoin.commented.sol

@@ -0,0 +1,51 @@
+pragma solidity ^0.4.24;
+
+
+/// @title MetaCoin v.2
+contract MetaCoin {
+
+    /// @notice
+    mapping (address => uint) balances;
+
+    event Transfer(address _from,address _to, uint _amount);
+
+    /// @notice
+    /// @dev
+    /// @return
+    constructor() {
+        balances[tx.origin] = 10000;
+    }
+
+    /// @notice
+    /// @dev
+    /// @param receiver
+    /// @return
+    function sendCoin(address receiver) public returns(bool sufficient) {
+        if (balances[msg.sender] < amount) return false;
+        address myAddress = this;
+        balances[msg.sender] -= amount;
+        balances[receiver] += amount;
+        Transfer(myAddress,receiver,amount);
+        return true;
+    }
+
+    /// @notice
+    /// @dev
+    /// @param addr Custom comment on parameter
+    /// @param number
+    /// @return
+    function getBalance(address addr, uint number) public returns(uint) {
+        return balances[addr];
+    }
+
+    /// @notice
+    /// @dev
+    /// @param code Token code
+    /// @return
+    function getToken(uint code) public returns(uint) {
+        return balances[addr];
+    }
+}
+
+
+

test/test-contracts/Metacoin.edited.commented.sol

@@ -0,0 +1,51 @@
+pragma solidity ^0.4.24;
+
+
+/// @title MetaCoin v.2
+contract MetaCoin {
+
+    mapping (address => uint) balances;
+
+    event Transfer(address _from,address _to, uint _amount);
+
+    /// @notice
+    /// @dev
+    /// @return
+    constructor() {
+        balances[tx.origin] = 10000;
+    }
+
+    /// @notice
+    /// @dev
+    /// @param receiver
+    /// @param amount
+    /// @return
+    function sendCoin(address receiver) public returns(bool sufficient) {
+        if (balances[msg.sender] < amount) return false;
+        address myAddress = this;
+        balances[msg.sender] -= amount;
+        balances[receiver] += amount;
+        Transfer(myAddress,receiver,amount);
+        return true;
+    }
+
+    /// @notice
+    /// @dev
+    /// @param addr Custom comment on parameter
+    /// @return
+    function getBalance(address addr, uint number) public returns(uint) {
+        return balances[addr];
+    }
+
+    /// @notice
+    /// @dev
+    /// @param code Token code
+    /// @param timestamp Token access timestamp
+    /// @return
+    function getToken(uint code) public returns(uint) {
+        return balances[addr];
+    }
+}
+
+
+