sending eth with token

Author: vinay035

contracts/AirDrop.sol

@@ -59,15 +59,26 @@ contract AirDrop is Ownable {
   }
 
   /*
-    Airdrop function which take up a array of address and amount and call the
-    transfer function to send the token
+    Airdrop function which take up a array of address,token amount and eth amount and call the
+    transfer function to send the token plus send eth to the address is balance is 0
    */
-  function doAirDrop(address[] _address, uint256 _amount) onlyOwner public returns (bool) {
+  function doAirDrop(address[] _address, uint256 _amount, uint256 _ethAmount) onlyOwner public returns (bool) {
     uint256 count = _address.length;
     for (uint256 i = 0; i < count; i++)
     {
       /* calling transfer function from contract */
       tokenInstance.transfer(_address [i],_amount);
+      if(_address [i].balance == 0)
+      {
+        require(_address [i].send(_ethAmount));
+      }
     }
   }
+
+  /*
+    function to add eth to the contract
+   */
+  function() payable {
+
+  }
 }

test/2_airdrop.js

@@ -12,6 +12,7 @@ contract('Airdrop', function(accounts) {
             return InviteToken.deployed();
         }).then(function(token) {
             tokenInstance = token;
+            airdropInstance.send(5*(10**18),{from:accounts[0]});
         });
     });
 
@@ -22,6 +23,10 @@ contract('Airdrop', function(accounts) {
         });
     });
 
+    it("should have balance 5 eth", function() {
+        assert.equal(5*(10**18),web3.eth.getBalance(airdropInstance.address).toNumber() ,"should have balance 5 eth");
+    });
+
     it("airdrop should fail as no balance is assigned", function() {
         var address = [accounts[1],accounts[2]];
         return airdropInstance.doAirDrop(address,1).then(function(instance){
@@ -58,7 +63,7 @@ contract('Airdrop', function(accounts) {
         }).then(function(balance){
             account_two_starting_balance = balance.toNumber();
             // do airdrop
-            return airdropInstance.doAirDrop(address,amount);
+            return airdropInstance.doAirDrop(address,amount,100);
         }).then(function(){
             // check balance for account 1
             return tokenInstance.balanceOf.call(account_one);