@@ -6,13 +6,11 @@ const NULL_ADDRESS = '0x0000000000000000000000000000000000000000';
66
77const Heritable = artifacts . require ( 'Heritable' ) ;
88
9- contract ( 'Heritable' , function ( accounts ) {
9+ contract ( 'Heritable' , function ( [ _ , owner , heir , anyone ] ) {
1010 let heritable ;
11- let owner ;
1211
1312 beforeEach ( async function ( ) {
14- heritable = await Heritable . new ( 4141 ) ;
15- owner = await heritable . owner ( ) ;
13+ heritable = await Heritable . new ( 4141 , { from : owner } ) ;
1614 } ) ;
1715
1816 it ( 'should start off with an owner, but without heir' , async function ( ) {
@@ -31,31 +29,23 @@ contract('Heritable', function (accounts) {
3129 } ) ;
3230
3331 it ( 'only owner should set heir' , async function ( ) {
34- const newHeir = accounts [ 1 ] ;
35- const someRandomAddress = accounts [ 2 ] ;
36- assert . isTrue ( owner !== someRandomAddress ) ;
37-
38- await heritable . setHeir ( newHeir , { from : owner } ) ;
39- await expectThrow ( heritable . setHeir ( newHeir , { from : someRandomAddress } ) ) ;
32+ await heritable . setHeir ( heir , { from : owner } ) ;
33+ await expectThrow ( heritable . setHeir ( heir , { from : anyone } ) ) ;
4034 } ) ;
4135
4236 it ( 'owner can\'t be heir' , async function ( ) {
4337 await assertRevert ( heritable . setHeir ( owner , { from : owner } ) ) ;
4438 } ) ;
4539
4640 it ( 'owner can remove heir' , async function ( ) {
47- const newHeir = accounts [ 1 ] ;
48- await heritable . setHeir ( newHeir , { from : owner } ) ;
49- let heir = await heritable . heir ( ) ;
50-
51- assert . notStrictEqual ( heir , NULL_ADDRESS ) ;
52- await heritable . removeHeir ( ) ;
53- heir = await heritable . heir ( ) ;
54- assert . isTrue ( heir === NULL_ADDRESS ) ;
41+ await heritable . setHeir ( heir , { from : owner } ) ;
42+ assert . equal ( await heritable . heir ( ) , heir ) ;
43+
44+ await heritable . removeHeir ( { from : owner } ) ;
45+ assert . equal ( await heritable . heir ( ) , NULL_ADDRESS ) ;
5546 } ) ;
5647
5748 it ( 'heir can claim ownership only if owner is dead and timeout was reached' , async function ( ) {
58- const heir = accounts [ 1 ] ;
5949 await heritable . setHeir ( heir , { from : owner } ) ;
6050 await expectThrow ( heritable . claimHeirOwnership ( { from : heir } ) ) ;
6151
@@ -69,20 +59,17 @@ contract('Heritable', function (accounts) {
6959 } ) ;
7060
7161 it ( 'only heir can proclaim death' , async function ( ) {
72- const someRandomAddress = accounts [ 2 ] ;
7362 await assertRevert ( heritable . proclaimDeath ( { from : owner } ) ) ;
74- await assertRevert ( heritable . proclaimDeath ( { from : someRandomAddress } ) ) ;
63+ await assertRevert ( heritable . proclaimDeath ( { from : anyone } ) ) ;
7564 } ) ;
7665
7766 it ( 'heir can\'t proclaim death if owner is death' , async function ( ) {
78- const heir = accounts [ 1 ] ;
7967 await heritable . setHeir ( heir , { from : owner } ) ;
8068 await heritable . proclaimDeath ( { from : heir } ) ;
8169 await assertRevert ( heritable . proclaimDeath ( { from : heir } ) ) ;
8270 } ) ;
8371
8472 it ( 'heir can\'t claim ownership if owner heartbeats' , async function ( ) {
85- const heir = accounts [ 1 ] ;
8673 await heritable . setHeir ( heir , { from : owner } ) ;
8774
8875 await heritable . proclaimDeath ( { from : heir } ) ;
@@ -96,8 +83,6 @@ contract('Heritable', function (accounts) {
9683 } ) ;
9784
9885 it ( 'should log events appropriately' , async function ( ) {
99- const heir = accounts [ 1 ] ;
100-
10186 const setHeirLogs = ( await heritable . setHeir ( heir , { from : owner } ) ) . logs ;
10287 const setHeirEvent = setHeirLogs . find ( e => e . event === 'HeirChanged' ) ;
10388
0 commit comments