You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
235 lines
10 KiB
235 lines
10 KiB
package eth
|
|
|
|
import (
|
|
"math/big"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
var wdelayerClient *WDelayerClient
|
|
var wdelayerClientTest *WDelayerClient
|
|
|
|
// var wdelayerClientKep *WDelayerClient
|
|
|
|
var initWithdrawalDelay = big.NewInt(60)
|
|
var newWithdrawalDelay = big.NewInt(79)
|
|
var maxEmergencyModeTime = time.Hour * 24 * 7 * 26
|
|
var maxWithdrawalDelay = time.Hour * 24 * 7 * 2
|
|
|
|
func TestWDelayerConstants(t *testing.T) {
|
|
wDelayerConstants, err := wdelayerClientTest.WDelayerConstants()
|
|
require.Nil(t, err)
|
|
assert.Equal(t, uint64(maxWithdrawalDelay.Seconds()), wDelayerConstants.MaxWithdrawalDelay)
|
|
assert.Equal(t, uint64(maxEmergencyModeTime.Seconds()), wDelayerConstants.MaxEmergencyModeTime)
|
|
assert.Equal(t, hermezRollupTestAddressConst, wDelayerConstants.HermezRollup)
|
|
}
|
|
|
|
func TestWDelayerGetHermezGovernanceDAOAddress(t *testing.T) {
|
|
governanceAddress, err := wdelayerClientTest.WDelayerGetHermezGovernanceDAOAddress()
|
|
require.Nil(t, err)
|
|
assert.Equal(t, &hermezGovernanceDAOAddressConst, governanceAddress)
|
|
}
|
|
|
|
func TestWDelayerSetHermezGovernanceDAOAddress(t *testing.T) {
|
|
wdelayerClientGov, err := NewWDelayerClient(ethereumClientGovDAO, wdelayerTestAddressConst)
|
|
require.Nil(t, err)
|
|
_, err = wdelayerClientGov.WDelayerSetHermezGovernanceDAOAddress(auxAddressConst)
|
|
require.Nil(t, err)
|
|
auxAddress, err := wdelayerClientTest.WDelayerGetHermezGovernanceDAOAddress()
|
|
require.Nil(t, err)
|
|
assert.Equal(t, &auxAddressConst, auxAddress)
|
|
currentBlockNum, err := wdelayerClientTest.client.EthLastBlock()
|
|
require.Nil(t, err)
|
|
wdelayerEvents, _, err := wdelayerClientTest.WDelayerEventsByBlock(currentBlockNum)
|
|
require.Nil(t, err)
|
|
assert.Equal(t, auxAddressConst, wdelayerEvents.NewHermezGovernanceDAOAddress[0].NewHermezGovernanceDAOAddress)
|
|
wdelayerClientAux, err := NewWDelayerClient(ethereumClientAux, wdelayerTestAddressConst)
|
|
require.Nil(t, err)
|
|
_, err = wdelayerClientAux.WDelayerSetHermezGovernanceDAOAddress(hermezGovernanceDAOAddressConst)
|
|
require.Nil(t, err)
|
|
}
|
|
|
|
func TestWDelayerGetHermezKeeperAddress(t *testing.T) {
|
|
keeperAddress, err := wdelayerClientTest.WDelayerGetHermezKeeperAddress()
|
|
require.Nil(t, err)
|
|
assert.Equal(t, &hermezKeeperAddressConst, keeperAddress)
|
|
}
|
|
|
|
func TestWDelayerSetHermezKeeperAddress(t *testing.T) {
|
|
wdelayerClientKep, err := NewWDelayerClient(ethereumClientKep, wdelayerTestAddressConst)
|
|
require.Nil(t, err)
|
|
_, err = wdelayerClientKep.WDelayerSetHermezKeeperAddress(auxAddressConst)
|
|
require.Nil(t, err)
|
|
auxAddress, err := wdelayerClientTest.WDelayerGetHermezKeeperAddress()
|
|
require.Nil(t, err)
|
|
assert.Equal(t, &auxAddressConst, auxAddress)
|
|
currentBlockNum, err := wdelayerClientTest.client.EthLastBlock()
|
|
require.Nil(t, err)
|
|
wdelayerEvents, _, err := wdelayerClientTest.WDelayerEventsByBlock(currentBlockNum)
|
|
require.Nil(t, err)
|
|
assert.Equal(t, auxAddressConst, wdelayerEvents.NewHermezKeeperAddress[0].NewHermezKeeperAddress)
|
|
wdelayerClientAux, err := NewWDelayerClient(ethereumClientAux, wdelayerTestAddressConst)
|
|
require.Nil(t, err)
|
|
_, err = wdelayerClientAux.WDelayerSetHermezKeeperAddress(hermezKeeperAddressConst)
|
|
require.Nil(t, err)
|
|
}
|
|
|
|
func TestWDelayerGetWhiteHackGroupAddress(t *testing.T) {
|
|
whiteHackGroupAddress, err := wdelayerClientTest.WDelayerGetWhiteHackGroupAddress()
|
|
require.Nil(t, err)
|
|
assert.Equal(t, &whiteHackGroupAddressConst, whiteHackGroupAddress)
|
|
}
|
|
|
|
func TestWDelayerSetWhiteHackGroupAddress(t *testing.T) {
|
|
wdelayerClientWhite, err := NewWDelayerClient(ethereumClientWhite, wdelayerTestAddressConst)
|
|
require.Nil(t, err)
|
|
_, err = wdelayerClientWhite.WDelayerSetWhiteHackGroupAddress(auxAddressConst)
|
|
require.Nil(t, err)
|
|
auxAddress, err := wdelayerClientTest.WDelayerGetWhiteHackGroupAddress()
|
|
require.Nil(t, err)
|
|
assert.Equal(t, &auxAddressConst, auxAddress)
|
|
currentBlockNum, err := wdelayerClientTest.client.EthLastBlock()
|
|
require.Nil(t, err)
|
|
wdelayerEvents, _, err := wdelayerClientTest.WDelayerEventsByBlock(currentBlockNum)
|
|
require.Nil(t, err)
|
|
assert.Equal(t, auxAddressConst, wdelayerEvents.NewWhiteHackGroupAddress[0].NewWhiteHackGroupAddress)
|
|
wdelayerClientAux, err := NewWDelayerClient(ethereumClientAux, wdelayerTestAddressConst)
|
|
require.Nil(t, err)
|
|
_, err = wdelayerClientAux.WDelayerSetWhiteHackGroupAddress(whiteHackGroupAddressConst)
|
|
require.Nil(t, err)
|
|
}
|
|
|
|
func TestWDelayerIsEmergencyMode(t *testing.T) {
|
|
emergencyMode, err := wdelayerClientTest.WDelayerIsEmergencyMode()
|
|
require.Nil(t, err)
|
|
assert.Equal(t, false, emergencyMode)
|
|
}
|
|
|
|
func TestWDelayerGetWithdrawalDelay(t *testing.T) {
|
|
withdrawalDelay, err := wdelayerClientTest.WDelayerGetWithdrawalDelay()
|
|
require.Nil(t, err)
|
|
assert.Equal(t, initWithdrawalDelay, withdrawalDelay)
|
|
}
|
|
|
|
func TestWDelayerChangeWithdrawalDelay(t *testing.T) {
|
|
wdelayerClientKep, err := NewWDelayerClient(ethereumClientKep, wdelayerTestAddressConst)
|
|
require.Nil(t, err)
|
|
_, err = wdelayerClientKep.WDelayerChangeWithdrawalDelay(newWithdrawalDelay.Uint64())
|
|
require.Nil(t, err)
|
|
withdrawalDelay, err := wdelayerClientTest.WDelayerGetWithdrawalDelay()
|
|
require.Nil(t, err)
|
|
assert.Equal(t, newWithdrawalDelay, withdrawalDelay)
|
|
currentBlockNum, err := wdelayerClientTest.client.EthLastBlock()
|
|
require.Nil(t, err)
|
|
wdelayerEvents, _, err := wdelayerClientTest.WDelayerEventsByBlock(currentBlockNum)
|
|
require.Nil(t, err)
|
|
assert.Equal(t, newWithdrawalDelay.Uint64(), wdelayerEvents.NewWithdrawalDelay[0].WithdrawalDelay)
|
|
}
|
|
|
|
func TestWDelayerDeposit(t *testing.T) {
|
|
amount := new(big.Int)
|
|
amount.SetString("1100000000000000000", 10)
|
|
wdelayerClientHermez, err := NewWDelayerClient(ethereumClientHermez, wdelayerTestAddressConst)
|
|
require.Nil(t, err)
|
|
_, err = wdelayerClientHermez.WDelayerDeposit(auxAddressConst, tokenHEZAddressConst, amount)
|
|
require.Nil(t, err)
|
|
currentBlockNum, err := wdelayerClientTest.client.EthLastBlock()
|
|
require.Nil(t, err)
|
|
wdelayerEvents, _, err := wdelayerClientTest.WDelayerEventsByBlock(currentBlockNum)
|
|
require.Nil(t, err)
|
|
assert.Equal(t, amount, wdelayerEvents.Deposit[0].Amount)
|
|
assert.Equal(t, auxAddressConst, wdelayerEvents.Deposit[0].Owner)
|
|
assert.Equal(t, tokenHEZAddressConst, wdelayerEvents.Deposit[0].Token)
|
|
}
|
|
|
|
func TestWDelayerDepositInfo(t *testing.T) {
|
|
amount := new(big.Int)
|
|
amount.SetString("1100000000000000000", 10)
|
|
state, err := wdelayerClientTest.WDelayerDepositInfo(auxAddressConst, tokenHEZAddressConst)
|
|
require.Nil(t, err)
|
|
assert.Equal(t, state.Amount, amount)
|
|
}
|
|
|
|
func TestWDelayerWithdrawal(t *testing.T) {
|
|
amount := new(big.Int)
|
|
amount.SetString("1100000000000000000", 10)
|
|
_, err := wdelayerClientTest.WDelayerWithdrawal(auxAddressConst, tokenHEZAddressConst)
|
|
require.Contains(t, err.Error(), "WITHDRAWAL_NOT_ALLOWED")
|
|
addTime(float64(newWithdrawalDelay.Int64()), ethClientDialURL)
|
|
addBlock(ethClientDialURL)
|
|
_, err = wdelayerClientTest.WDelayerWithdrawal(auxAddressConst, tokenHEZAddressConst)
|
|
require.Nil(t, err)
|
|
currentBlockNum, err := wdelayerClientTest.client.EthLastBlock()
|
|
require.Nil(t, err)
|
|
wdelayerEvents, _, err := wdelayerClientTest.WDelayerEventsByBlock(currentBlockNum)
|
|
require.Nil(t, err)
|
|
assert.Equal(t, amount, wdelayerEvents.Withdraw[0].Amount)
|
|
assert.Equal(t, auxAddressConst, wdelayerEvents.Withdraw[0].Owner)
|
|
assert.Equal(t, tokenHEZAddressConst, wdelayerEvents.Withdraw[0].Token)
|
|
}
|
|
|
|
func TestWDelayerSecondDeposit(t *testing.T) {
|
|
amount := new(big.Int)
|
|
amount.SetString("1100000000000000000", 10)
|
|
wdelayerClientHermez, err := NewWDelayerClient(ethereumClientHermez, wdelayerTestAddressConst)
|
|
require.Nil(t, err)
|
|
_, err = wdelayerClientHermez.WDelayerDeposit(auxAddressConst, tokenHEZAddressConst, amount)
|
|
require.Nil(t, err)
|
|
currentBlockNum, err := wdelayerClientTest.client.EthLastBlock()
|
|
require.Nil(t, err)
|
|
wdelayerEvents, _, err := wdelayerClientTest.WDelayerEventsByBlock(currentBlockNum)
|
|
require.Nil(t, err)
|
|
assert.Equal(t, amount, wdelayerEvents.Deposit[0].Amount)
|
|
assert.Equal(t, auxAddressConst, wdelayerEvents.Deposit[0].Owner)
|
|
assert.Equal(t, tokenHEZAddressConst, wdelayerEvents.Deposit[0].Token)
|
|
}
|
|
|
|
func TestWDelayerEnableEmergencyMode(t *testing.T) {
|
|
wdelayerClientKep, err := NewWDelayerClient(ethereumClientKep, wdelayerTestAddressConst)
|
|
require.Nil(t, err)
|
|
_, err = wdelayerClientKep.WDelayerEnableEmergencyMode()
|
|
require.Nil(t, err)
|
|
emergencyMode, err := wdelayerClientTest.WDelayerIsEmergencyMode()
|
|
require.Nil(t, err)
|
|
assert.Equal(t, true, emergencyMode)
|
|
currentBlockNum, err := wdelayerClientTest.client.EthLastBlock()
|
|
require.Nil(t, err)
|
|
wdelayerEvents, _, err := wdelayerClientTest.WDelayerEventsByBlock(currentBlockNum)
|
|
require.Nil(t, err)
|
|
auxEvent := new(WDelayerEventEmergencyModeEnabled)
|
|
assert.Equal(t, auxEvent, &wdelayerEvents.EmergencyModeEnabled[0])
|
|
}
|
|
|
|
func TestWDelayerGetEmergencyModeStartingTime(t *testing.T) {
|
|
emergencyModeStartingTime, err := wdelayerClientTest.WDelayerGetEmergencyModeStartingTime()
|
|
require.Nil(t, err)
|
|
// `emergencyModeStartingTime` is initialized to 0 in the smart
|
|
// contract construction. Since we called WDelayerEnableEmergencyMode
|
|
// previously, `emergencyModeStartingTime` is set to the time when the
|
|
// call was made, so it's > 0.
|
|
assert.True(t, emergencyModeStartingTime.Cmp(big.NewInt(0)) == 1)
|
|
}
|
|
|
|
func TestWDelayerEscapeHatchWithdrawal(t *testing.T) {
|
|
amount := new(big.Int)
|
|
amount.SetString("10000000000000000", 10)
|
|
wdelayerClientWhite, err := NewWDelayerClient(ethereumClientWhite, wdelayerTestAddressConst)
|
|
require.Nil(t, err)
|
|
_, err = wdelayerClientWhite.WDelayerEscapeHatchWithdrawal(governanceAddressConst, tokenHEZAddressConst, amount)
|
|
require.Contains(t, err.Error(), "NO_MAX_EMERGENCY_MODE_TIME")
|
|
seconds := maxEmergencyModeTime.Seconds()
|
|
addTime(seconds, ethClientDialURL)
|
|
_, err = wdelayerClientWhite.WDelayerEscapeHatchWithdrawal(governanceAddressConst, tokenHEZAddressConst, amount)
|
|
require.Nil(t, err)
|
|
currentBlockNum, err := wdelayerClientTest.client.EthLastBlock()
|
|
require.Nil(t, err)
|
|
wdelayerEvents, _, err := wdelayerClientTest.WDelayerEventsByBlock(currentBlockNum)
|
|
require.Nil(t, err)
|
|
assert.Equal(t, tokenHEZAddressConst, wdelayerEvents.EscapeHatchWithdrawal[0].Token)
|
|
assert.Equal(t, governanceAddressConst, wdelayerEvents.EscapeHatchWithdrawal[0].To)
|
|
assert.Equal(t, whiteHackGroupAddressConst, wdelayerEvents.EscapeHatchWithdrawal[0].Who)
|
|
assert.Equal(t, amount, wdelayerEvents.EscapeHatchWithdrawal[0].Amount)
|
|
}
|