Migrate to the new contract interfaces (#29)

* Fix compile warning & Search contract by name automatically

Migrate to the new contract interfaces

* Fix compile

* Fix lint
This commit is contained in:
Chenxing Li 2024-03-14 13:56:17 +08:00 committed by GitHub
parent 2ed6bca446
commit 306c43c9dc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
9 changed files with 79 additions and 103 deletions

@ -1 +1 @@
Subproject commit 951c02992e2a03eb47e41eb840392ec824639640
Subproject commit 1f3f759236b48dd8682deba91011292cb5966f4c

View File

@ -30,3 +30,6 @@ resolver = "2"
discv5 = { path = "version-meld/discv5" }
eth2_ssz = { path = "version-meld/eth2_ssz" }
enr = { path = "version-meld/enr" }
[profile.bench.package.'storage']
debug = true

View File

@ -1,30 +1,35 @@
// use std::process::Command;
use std::process::Command;
// const INSTALL_ERROR_MESSAGE: &str =
// "Install dependencies for contract fail, try to run `yarn` in folder '0g-storage-contracts'";
// const COMPILE_ERROR_MESSAGE: &str =
// "Compile solidity contracts fail, try to run `yarn compile` in folder '0g-storage-contracts'";
const INSTALL_ERROR_MESSAGE: &str =
"Install dependencies for contract fail, try to run `yarn` in folder '0g-storage-contracts'";
const COMPILE_ERROR_MESSAGE: &str =
"Compile solidity contracts fail, try to run `yarn compile` in folder '0g-storage-contracts'";
#[allow(dead_code)]
fn compile_contracts() {
println!("cargo:rerun-if-changed=../../0g-storage-contracts/contracts/");
println!("cargo:rerun-if-changed=../../0g-storage-contracts/hardhat.config.ts");
let output = Command::new("yarn")
.arg("--cwd")
.arg("../../0g-storage-contracts")
.status()
.expect(INSTALL_ERROR_MESSAGE);
assert!(output.success(), "{}", INSTALL_ERROR_MESSAGE);
let output = Command::new("yarn")
.arg("--cwd")
.arg("../../0g-storage-contracts")
.arg("compile")
.status()
.expect(COMPILE_ERROR_MESSAGE);
assert!(output.success(), "{}", COMPILE_ERROR_MESSAGE);
}
fn main() {
// if cfg!(feature = "compile-contracts") {
// println!("cargo:rerun-if-changed=../../0g-storage-contracts/contracts/");
// println!("cargo:rerun-if-changed=../../0g-storage-contracts/hardhat.config.ts");
// let output = Command::new("yarn")
// .arg("--cwd")
// .arg("../../0g-storage-contracts")
// .status()
// .expect(INSTALL_ERROR_MESSAGE);
// assert!(output.success(), "{}", INSTALL_ERROR_MESSAGE);
// let output = Command::new("yarn")
// .arg("--cwd")
// .arg("../../0g-storage-contracts")
// .arg("compile")
// .status()
// .expect(COMPILE_ERROR_MESSAGE);
// assert!(output.success(), "{}", COMPILE_ERROR_MESSAGE);
// } else {
// println!("cargo:rerun-if-changed=../../0g-storage-contracts/artifacts/");
// }
if cfg!(feature = "compile-contracts") {
// compile_contracts();
// return;
}
println!("cargo:rerun-if-changed=../../0g-storage-contracts/artifacts/");
}

View File

@ -36,6 +36,3 @@ criterion = "0.4"
[[bench]]
name = "benchmark"
harness = false
[profile.bench]
debug = true

View File

@ -3,8 +3,9 @@ import os
import subprocess
import tempfile
import time
import rlp
from eth_utils import decode_hex
from eth_utils import decode_hex, keccak
from web3 import Web3, HTTPProvider
from web3.middleware import construct_sign_and_send_raw_middleware
from enum import Enum, unique
@ -22,6 +23,7 @@ from utility.utils import (
initialize_config,
wait_until,
)
from test_framework.contracts import load_contract_metadata
@unique
@ -218,15 +220,11 @@ class BlockchainNode(TestNode):
binary,
local_conf,
contract_path,
token_contract_path,
mine_contract_path,
log,
blockchain_node_type,
rpc_timeout=10,
):
self.contract_path = contract_path
self.token_contract_path = token_contract_path
self.mine_contract_path = mine_contract_path
self.blockchain_node_type = blockchain_node_type
@ -261,14 +259,15 @@ class BlockchainNode(TestNode):
# account = w3.eth.account.from_key(GENESIS_PRIV_KEY1)
# w3.middleware_onion.add(construct_sign_and_send_raw_middleware(account))
def deploy_contract(path, args=None):
def deploy_contract(name, args=None):
if args is None:
args = []
contract_interface = json.load(open(path, "r"))
contract_interface = load_contract_metadata(base_path=self.contract_path, name=name)
contract = w3.eth.contract(
abi=contract_interface["abi"],
bytecode=contract_interface["bytecode"],
)
tx_params = TX_PARAMS.copy()
del tx_params["gas"]
tx_hash = contract.constructor(*args).transact(tx_params)
@ -279,37 +278,35 @@ class BlockchainNode(TestNode):
)
return contract, tx_hash
def predict_contract_address(offset):
nonce = w3.eth.get_transaction_count(account1.address)
rlp_encoded = rlp.encode([decode_hex(account1.address), nonce + offset])
contract_address_bytes = keccak(rlp_encoded)[-20:]
contract_address = '0x' + contract_address_bytes.hex()
return Web3.to_checksum_address(contract_address)
flowAddress = predict_contract_address(1)
mineAddress = predict_contract_address(2)
ZERO = "0x0000000000000000000000000000000000000000"
self.log.debug("Start deploy contracts")
token_contract, _ = deploy_contract(self.token_contract_path)
self.log.debug("ERC20 deployed")
book, _ = deploy_contract("AddressBook", [flowAddress, ZERO, ZERO, mineAddress]);
self.log.debug("AddressBook deployed")
flow_contract, flow_contract_hash = deploy_contract(
self.contract_path, ["0x0000000000000000000000000000000000000000", 100, 0]
"Flow", [book.address, 100, 0]
)
self.log.debug("Flow deployed")
mine_contract, _ = deploy_contract(
self.mine_contract_path,
[flow_contract.address, "0x0000000000000000000000000000000000000000", 7],
"PoraMineTest",
[book.address, 3],
)
self.log.debug("Mine deployed")
self.log.info("All contracts deployed")
tx_hash = token_contract.functions.approve(
flow_contract.address, int(1e9)
).transact(TX_PARAMS)
self.wait_for_transaction_receipt(w3, tx_hash)
# setup second account
amount = int(1e8)
tx_hash = token_contract.functions.transfer(account2.address, amount).transact(
TX_PARAMS
)
self.wait_for_transaction_receipt(w3, tx_hash)
tx_hash = token_contract.functions.approve(
flow_contract.address, amount
).transact(TX_PARAMS1)
self.wait_for_transaction_receipt(w3, tx_hash)
tx_hash = mine_contract.functions.setMiner(decode_hex(MINER_ID)).transact(TX_PARAMS)
self.wait_for_transaction_receipt(w3, tx_hash)
@ -324,7 +321,7 @@ class BlockchainNode(TestNode):
construct_sign_and_send_raw_middleware([account1, account2])
)
contract_interface = json.load(open(self.contract_path, "r"))
contract_interface = load_contract_metadata(self.contract_path, "Flow")
return w3.eth.contract(address=contract_address, abi=contract_interface["abi"])
def wait_for_transaction(self, tx_hash):

View File

@ -26,8 +26,6 @@ class BSCNode(BlockchainNode):
binary,
updated_config,
contract_path,
token_contract_path,
mine_contract_path,
log,
rpc_timeout=10,
):
@ -71,8 +69,6 @@ class BSCNode(BlockchainNode):
binary,
local_conf,
contract_path,
token_contract_path,
mine_contract_path,
log,
BlockChainNodeType.BSC,
rpc_timeout,

View File

@ -29,8 +29,6 @@ class ConfluxNode(BlockchainNode):
binary,
updated_config,
contract_path,
token_contract_path,
mine_contract_path,
log,
rpc_timeout=10,
):
@ -76,8 +74,6 @@ class ConfluxNode(BlockchainNode):
binary,
local_conf,
contract_path,
token_contract_path,
mine_contract_path,
log,
BlockChainNodeType.Conflux,
rpc_timeout,

View File

@ -0,0 +1,14 @@
from os.path import join
from pathlib import Path
import json
from web3 import Web3
def load_contract_metadata(base_path: str, name: str):
path = Path(join(base_path, "artifacts"))
try:
found_file = next(path.rglob(f"{name}.json"))
return json.loads(open(found_file, "r").read())
except StopIteration:
raise Exception(f"Cannot found contract {name}'s metadata")

View File

@ -74,8 +74,6 @@ class TestFramework:
self.blockchain_binary,
updated_config,
self.contract_path,
self.token_contract_path,
self.mine_contract_path,
self.log,
60,
)
@ -86,8 +84,6 @@ class TestFramework:
self.blockchain_binary,
updated_config,
self.contract_path,
self.token_contract_path,
self.mine_contract_path,
self.log,
)
else:
@ -212,27 +208,7 @@ class TestFramework:
dest="contract",
default=os.path.join(
__file_path__,
"../../0g-storage-contracts/artifacts/contracts/dataFlow/Flow.sol/Flow.json",
),
type=str,
)
parser.add_argument(
"--token-contract-path",
dest="token_contract",
default=os.path.join(
__file_path__,
"../config/MockToken.json",
),
type=str,
)
parser.add_argument(
"--mine-contract-path",
dest="mine_contract",
default=os.path.join(
__file_path__,
"../../0g-storage-contracts/artifacts/contracts/test/PoraMineTest.sol/PoraMineTest.json",
"../../0g-storage-contracts/",
),
type=str,
)
@ -414,18 +390,10 @@ class TestFramework:
self.zgs_binary = os.path.abspath(self.options.zerog_storage)
self.cli_binary = os.path.abspath(self.options.cli)
self.contract_path = os.path.abspath(self.options.contract)
self.token_contract_path = os.path.abspath(self.options.token_contract)
self.mine_contract_path = os.path.abspath(self.options.mine_contract)
assert os.path.exists(self.contract_path), (
"%s should be exist" % self.contract_path
)
assert os.path.exists(self.token_contract_path), (
"%s should be exist" % self.token_contract_path
)
assert os.path.exists(self.mine_contract_path), (
"%s should be exist" % self.mine_contract_path
)
if self.options.random_seed is not None:
random.seed(self.options.random_seed)