mirror of
https://github.com/sigp/lighthouse.git
synced 2026-03-15 10:52:43 +00:00
76 lines
1.6 KiB
Python
76 lines
1.6 KiB
Python
import os
|
|
import shutil
|
|
import subprocess
|
|
|
|
NUM_VALIDATORS=3
|
|
TEST_MNEMONIC = "test test test test test test test test test test test waste"
|
|
WALLET_NAME="test_wallet"
|
|
|
|
tmp_dir = os.path.join(".", "tmp")
|
|
mnemonic_path = os.path.join(tmp_dir, "mnemonic.txt")
|
|
lh_dir = os.path.join(tmp_dir, "lh")
|
|
lh_json_path = os.path.join(lh_dir, "deposit-data.json")
|
|
lh_wallet_password_path = os.path.join(lh_dir, "wallet.pass")
|
|
sdc_dir = os.path.join(tmp_dir, "sdc")
|
|
|
|
|
|
def setup():
|
|
if os.path.exists(tmp_dir):
|
|
cleanup()
|
|
os.mkdir(tmp_dir)
|
|
with open(mnemonic_path, "x") as file:
|
|
file.write(TEST_MNEMONIC)
|
|
|
|
|
|
def cleanup():
|
|
shutil.rmtree(tmp_dir)
|
|
|
|
|
|
def lighthouse_generate(network):
|
|
result = subprocess.run([
|
|
"lighthouse",
|
|
"--network",
|
|
network,
|
|
"account",
|
|
"wallet",
|
|
"recover",
|
|
"--datadir",
|
|
str(lh_dir),
|
|
"--name",
|
|
WALLET_NAME,
|
|
"--mnemonic-path",
|
|
str(mnemonic_path),
|
|
"--password-file",
|
|
str(lh_wallet_password_path)
|
|
])
|
|
assert(result.returncode == 0)
|
|
|
|
result = subprocess.run([
|
|
"lighthouse",
|
|
"--network",
|
|
network,
|
|
"account",
|
|
"validator",
|
|
"create",
|
|
"--datadir",
|
|
str(lh_dir),
|
|
"--wallet-name",
|
|
WALLET_NAME,
|
|
"--wallet-password",
|
|
str(lh_wallet_password_path),
|
|
"--count",
|
|
str(NUM_VALIDATORS),
|
|
"--json-deposit-data-path",
|
|
str(lh_json_path)
|
|
])
|
|
assert(result.returncode == 0)
|
|
|
|
|
|
def test(network):
|
|
setup()
|
|
lighthouse_generate(network)
|
|
# cleanup()
|
|
|
|
|
|
test("mainnet")
|