Skip to content
This repository has been archived by the owner on May 11, 2024. It is now read-only.

feat(test): fix suite.go #453

Merged
merged 2 commits into from
Nov 30, 2023
Merged

feat(test): fix suite.go #453

Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
8b5eaeb
test
RogerLamTd Nov 30, 2023
5f3de6e
remove unused code, fix comment
RogerLamTd Nov 30, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
64 changes: 30 additions & 34 deletions testutils/suite.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -79,14 +79,10 @@ func (s *ClientTestSuite) SetupTest() {
s.ProverEndpoints = []*url.URL{LocalRandomProverEndpoint()}
s.proverServer = NewTestProverServer(s, l1ProverPrivKey, capacity.New(1024), s.ProverEndpoints[0])

allowance, err := rpcCli.TaikoToken.Allowance(
nil,
crypto.PubkeyToAddress(l1ProverPrivKey.PublicKey),
common.HexToAddress("TAIKO_L1_ADDRESS"),
)
balance, err := rpcCli.TaikoToken.BalanceOf(nil, crypto.PubkeyToAddress(l1ProverPrivKey.PublicKey))
s.Nil(err)

if allowance.Cmp(common.Big0) == 0 {
if balance.Cmp(common.Big0) == 0 {
// Do not verify zk && sgx proofs in tests.
securityConcilPrivKey, err := crypto.ToECDSA(common.Hex2Bytes(os.Getenv("L1_SECURITY_COUNCIL_PRIVATE_KEY")))
s.Nil(err)
Expand All @@ -104,35 +100,35 @@ func (s *ClientTestSuite) SetupTest() {
opts, err := bind.NewKeyedTransactorWithChainID(ownerPrivKey, rpcCli.L1ChainID)
s.Nil(err)
proverBalance := new(big.Int).Div(balance, common.Big2)
if proverBalance.Cmp(common.Big0) == 1 {
tx, err := rpcCli.TaikoToken.Transfer(opts, crypto.PubkeyToAddress(l1ProverPrivKey.PublicKey), proverBalance)
s.Nil(err)
_, err = rpc.WaitReceipt(context.Background(), rpcCli.L1, tx)
s.Nil(err)

// Deposit taiko tokens for provers.
opts, err = bind.NewKeyedTransactorWithChainID(l1ProverPrivKey, rpcCli.L1ChainID)
s.Nil(err)

_, err = rpcCli.TaikoToken.Approve(
opts,
common.HexToAddress(os.Getenv("ASSIGNMENT_HOOK_ADDRESS")),
new(big.Int).Exp(big.NewInt(1_000_000_000), big.NewInt(18), nil),
)
s.Nil(err)

_, err = rpcCli.TaikoToken.Approve(
opts,
common.HexToAddress(os.Getenv("TAIKO_L1_ADDRESS")),
new(big.Int).Exp(big.NewInt(1_000_000_000), big.NewInt(18), nil),
)
s.Nil(err)

_, err = rpc.WaitReceipt(context.Background(), rpcCli.L1, tx)
s.Nil(err)
}
}
s.Greater(proverBalance.Cmp(common.Big0), 0)

tx, err := rpcCli.TaikoToken.Transfer(opts, crypto.PubkeyToAddress(l1ProverPrivKey.PublicKey), proverBalance)
s.Nil(err)
_, err = rpc.WaitReceipt(context.Background(), rpcCli.L1, tx)
s.Nil(err)

// Increase allowance for AssignmentHook and TaikoL1
opts, err = bind.NewKeyedTransactorWithChainID(l1ProverPrivKey, rpcCli.L1ChainID)
s.Nil(err)

bigInt := new(big.Int).Exp(big.NewInt(1_000_000_000), big.NewInt(18), nil)
_, err = rpcCli.TaikoToken.Approve(
opts,
common.HexToAddress(os.Getenv("ASSIGNMENT_HOOK_ADDRESS")),
bigInt,
)
s.Nil(err)

_, err = rpcCli.TaikoToken.Approve(
opts,
common.HexToAddress(os.Getenv("TAIKO_L1_ADDRESS")),
bigInt,
)
s.Nil(err)

_, err = rpc.WaitReceipt(context.Background(), rpcCli.L1, tx)
s.Nil(err)
}
s.Nil(rpcCli.L1RawRPC.CallContext(context.Background(), &s.testnetL1SnapshotID, "evm_snapshot"))
s.NotEmpty(s.testnetL1SnapshotID)
}
Expand Down