mirror of
https://gitlab.com/pulsechaincom/staking-deposit-cli.git
synced 2025-01-10 13:01:22 +00:00
120 lines
3.8 KiB
Python
120 lines
3.8 KiB
Python
import asyncio
|
|
import os
|
|
|
|
import pytest
|
|
|
|
from click.testing import CliRunner
|
|
from eth2deposit.cli import new_mnemonic
|
|
from eth2deposit.deposit import cli
|
|
from eth2deposit.utils.constants import DEFAULT_VALIDATOR_KEYS_FOLDER_NAME
|
|
from eth2deposit.utils.intl import load_text
|
|
from .helpers import clean_key_folder, get_permissions, get_uuid
|
|
|
|
|
|
def test_new_mnemonic(monkeypatch) -> None:
|
|
# monkeypatch get_mnemonic
|
|
def mock_get_mnemonic(language, words_path, entropy=None) -> str:
|
|
return "fakephrase"
|
|
|
|
monkeypatch.setattr(new_mnemonic, "get_mnemonic", mock_get_mnemonic)
|
|
|
|
# Prepare folder
|
|
my_folder_path = os.path.join(os.getcwd(), 'TESTING_TEMP_FOLDER')
|
|
clean_key_folder(my_folder_path)
|
|
if not os.path.exists(my_folder_path):
|
|
os.mkdir(my_folder_path)
|
|
|
|
runner = CliRunner()
|
|
inputs = ['english', '1', 'mainnet', 'MyPassword', 'MyPassword', 'fakephrase']
|
|
data = '\n'.join(inputs)
|
|
result = runner.invoke(cli, ['new-mnemonic', '--folder', my_folder_path], input=data)
|
|
assert result.exit_code == 0
|
|
|
|
# Check files
|
|
validator_keys_folder_path = os.path.join(my_folder_path, DEFAULT_VALIDATOR_KEYS_FOLDER_NAME)
|
|
_, _, key_files = next(os.walk(validator_keys_folder_path))
|
|
|
|
all_uuid = [
|
|
get_uuid(validator_keys_folder_path + '/' + key_file)
|
|
for key_file in key_files
|
|
if key_file.startswith('keystore')
|
|
]
|
|
assert len(set(all_uuid)) == 1
|
|
|
|
# Verify file permissions
|
|
if os.name == 'posix':
|
|
for file_name in key_files:
|
|
assert get_permissions(validator_keys_folder_path, file_name) == '0o440'
|
|
|
|
# Clean up
|
|
clean_key_folder(my_folder_path)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_script() -> None:
|
|
my_folder_path = os.path.join(os.getcwd(), 'TESTING_TEMP_FOLDER')
|
|
if not os.path.exists(my_folder_path):
|
|
os.mkdir(my_folder_path)
|
|
|
|
if os.name == 'nt': # Windows
|
|
run_script_cmd = 'sh deposit.sh'
|
|
else: # Mac or Linux
|
|
run_script_cmd = './deposit.sh'
|
|
|
|
install_cmd = run_script_cmd + ' install'
|
|
proc = await asyncio.create_subprocess_shell(
|
|
install_cmd,
|
|
)
|
|
await proc.wait()
|
|
|
|
cmd_args = [
|
|
run_script_cmd + ' new-mnemonic',
|
|
'--num_validators', '5',
|
|
'--mnemonic_language', 'english',
|
|
'--chain', 'mainnet',
|
|
'--keystore_password', 'MyPassword',
|
|
'--folder', my_folder_path,
|
|
]
|
|
proc = await asyncio.create_subprocess_shell(
|
|
' '.join(cmd_args),
|
|
stdin=asyncio.subprocess.PIPE,
|
|
stdout=asyncio.subprocess.PIPE,
|
|
)
|
|
|
|
seed_phrase = ''
|
|
parsing = False
|
|
intl_file_path = os.path.join(os.getcwd(), 'eth2deposit/../eth2deposit/cli/new_mnemonic.json')
|
|
async for out in proc.stdout:
|
|
output = out.decode('utf-8').rstrip()
|
|
if output.startswith(load_text(['msg_mnemonic_presentation'], intl_file_path, 'new_mnemonic')):
|
|
parsing = True
|
|
elif output.startswith(load_text(['msg_mnemonic_retype_prompt'], intl_file_path, 'new_mnemonic')):
|
|
parsing = False
|
|
elif parsing:
|
|
seed_phrase += output
|
|
if len(seed_phrase) > 0:
|
|
encoded_phrase = seed_phrase.encode()
|
|
proc.stdin.write(encoded_phrase)
|
|
proc.stdin.write(b'\n')
|
|
|
|
assert len(seed_phrase) > 0
|
|
|
|
# Check files
|
|
validator_keys_folder_path = os.path.join(my_folder_path, DEFAULT_VALIDATOR_KEYS_FOLDER_NAME)
|
|
_, _, key_files = next(os.walk(validator_keys_folder_path))
|
|
|
|
all_uuid = [
|
|
get_uuid(validator_keys_folder_path + '/' + key_file)
|
|
for key_file in key_files
|
|
if key_file.startswith('keystore')
|
|
]
|
|
assert len(set(all_uuid)) == 5
|
|
|
|
# Verify file permissions
|
|
if os.name == 'posix':
|
|
for file_name in key_files:
|
|
assert get_permissions(validator_keys_folder_path, file_name) == '0o440'
|
|
|
|
# Clean up
|
|
clean_key_folder(my_folder_path)
|