Remove lodash from tests (#1323)

OpenZeppelin · Sep 17, 2018 · d062352 · d062352
1 parent b4f87bb
commit d062352
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 10 deletions.
diff --git a/test/helpers/sendTransaction.js b/test/helpers/sendTransaction.js
@@ -1,9 +1,8 @@
-const _ = require('lodash');
 const ethjsABI = require('ethjs-abi');
 
 function findMethod (abi, name, args) {
   for (let i = 0; i < abi.length; i++) {
-    const methodArgs = _.map(abi[i].inputs, 'type').join(',');
+    const methodArgs = abi[i].inputs.map(input => input.type).join(',');
     if ((abi[i].name === name) && (methodArgs === args)) {
       return abi[i];
     }

diff --git a/test/token/ERC721/ERC721.behavior.js b/test/token/ERC721/ERC721.behavior.js
@@ -2,7 +2,6 @@ const { shouldSupportInterfaces } = require('../../introspection/SupportsInterfa
 const { assertRevert } = require('../../helpers/assertRevert');
 const { decodeLogs } = require('../../helpers/decodeLogs');
 const { sendTransaction } = require('../../helpers/sendTransaction');
-const _ = require('lodash');
 
 const ERC721Receiver = artifacts.require('ERC721ReceiverMock.sol');
 const BigNumber = web3.BigNumber;
@@ -175,7 +174,9 @@ function shouldBehaveLikeERC721 (
 
           it('keeps same tokens by index', async function () {
             if (!this.token.tokenOfOwnerByIndex) return;
-            const tokensListed = await Promise.all(_.range(2).map(i => this.token.tokenOfOwnerByIndex(owner, i)));
+            const tokensListed = await Promise.all(
+              [0, 1].map(i => this.token.tokenOfOwnerByIndex(owner, i))
+            );
             tokensListed.map(t => t.toNumber()).should.have.members([firstTokenId, secondTokenId]);
           });
         });

diff --git a/test/token/ERC721/ERC721Full.test.js b/test/token/ERC721/ERC721Full.test.js
@@ -2,7 +2,6 @@ const { assertRevert } = require('../../helpers/assertRevert');
 const { shouldBehaveLikeERC721 } = require('./ERC721.behavior');
 const { shouldBehaveLikeMintAndBurnERC721 } = require('./ERC721MintBurn.behavior');
 const { shouldSupportInterfaces } = require('../../introspection/SupportsInterface.behavior');
-const _ = require('lodash');
 
 const BigNumber = web3.BigNumber;
 const ERC721FullMock = artifacts.require('ERC721FullMock.sol');
@@ -173,7 +172,9 @@ contract('ERC721Full', function ([
 
         it('returns correct token IDs for target', async function () {
           (await this.token.balanceOf(another)).toNumber().should.be.equal(2);
-          const tokensListed = await Promise.all(_.range(2).map(i => this.token.tokenOfOwnerByIndex(another, i)));
+          const tokensListed = await Promise.all(
+            [0, 1].map(i => this.token.tokenOfOwnerByIndex(another, i))
+          );
           tokensListed.map(t => t.toNumber()).should.have.members([firstTokenId, secondTokenId]);
         });
 
@@ -186,7 +187,9 @@ contract('ERC721Full', function ([
 
     describe('tokenByIndex', function () {
       it('should return all tokens', async function () {
-        const tokensListed = await Promise.all(_.range(2).map(i => this.token.tokenByIndex(i)));
+        const tokensListed = await Promise.all(
+          [0, 1].map(i => this.token.tokenByIndex(i))
+        );
         tokensListed.map(t => t.toNumber()).should.have.members([firstTokenId, secondTokenId]);
       });
 
@@ -205,9 +208,10 @@ contract('ERC721Full', function ([
 
           (await this.token.totalSupply()).toNumber().should.be.equal(3);
 
-          const tokensListed = await Promise.all(_.range(3).map(i => this.token.tokenByIndex(i)));
-          const expectedTokens = _.filter(
-            [firstTokenId, secondTokenId, newTokenId, anotherNewTokenId],
+          const tokensListed = await Promise.all(
+            [0, 1, 2].map(i => this.token.tokenByIndex(i))
+          );
+          const expectedTokens = [firstTokenId, secondTokenId, newTokenId, anotherNewTokenId].filter(
             x => (x !== tokenId)
           );
           tokensListed.map(t => t.toNumber()).should.have.members(expectedTokens);