@@ -3,32 +3,32 @@ const { expectThrow } = require('../helpers/expectThrow');
33const CanReclaimToken = artifacts . require ( 'CanReclaimToken' ) ;
44const BasicTokenMock = artifacts . require ( 'BasicTokenMock' ) ;
55
6- contract ( 'CanReclaimToken' , function ( accounts ) {
6+ contract ( 'CanReclaimToken' , function ( [ _ , owner , anyone ] ) {
77 let token = null ;
88 let canReclaimToken = null ;
99
1010 beforeEach ( async function ( ) {
1111 // Create contract and token
12- token = await BasicTokenMock . new ( accounts [ 0 ] , 100 ) ;
13- canReclaimToken = await CanReclaimToken . new ( ) ;
12+ token = await BasicTokenMock . new ( owner , 100 , { from : owner } ) ;
13+ canReclaimToken = await CanReclaimToken . new ( { from : owner } ) ;
1414 // Force token into contract
15- await token . transfer ( canReclaimToken . address , 10 ) ;
15+ await token . transfer ( canReclaimToken . address , 10 , { from : owner } ) ;
1616 const startBalance = await token . balanceOf ( canReclaimToken . address ) ;
1717 assert . equal ( startBalance , 10 ) ;
1818 } ) ;
1919
2020 it ( 'should allow owner to reclaim tokens' , async function ( ) {
21- const ownerStartBalance = await token . balanceOf ( accounts [ 0 ] ) ;
22- await canReclaimToken . reclaimToken ( token . address ) ;
23- const ownerFinalBalance = await token . balanceOf ( accounts [ 0 ] ) ;
21+ const ownerStartBalance = await token . balanceOf ( owner ) ;
22+ await canReclaimToken . reclaimToken ( token . address , { from : owner } ) ;
23+ const ownerFinalBalance = await token . balanceOf ( owner ) ;
2424 const finalBalance = await token . balanceOf ( canReclaimToken . address ) ;
2525 assert . equal ( finalBalance , 0 ) ;
2626 assert . equal ( ownerFinalBalance - ownerStartBalance , 10 ) ;
2727 } ) ;
2828
2929 it ( 'should allow only owner to reclaim tokens' , async function ( ) {
3030 await expectThrow (
31- canReclaimToken . reclaimToken ( token . address , { from : accounts [ 1 ] } ) ,
31+ canReclaimToken . reclaimToken ( token . address , { from : anyone } )
3232 ) ;
3333 } ) ;
3434} ) ;
0 commit comments