forked from TrueCloudLab/frostfs-testcases
71 lines
2.1 KiB
Python
71 lines
2.1 KiB
Python
import os
|
|
import uuid
|
|
from dataclasses import dataclass
|
|
|
|
from frostfs_testlib.shell import Shell
|
|
from frostfs_testlib.utils import wallet_utils
|
|
|
|
from pytest_tests.helpers.cluster import Cluster, NodeBase
|
|
from pytest_tests.helpers.payment_neogo import deposit_gas, transfer_gas
|
|
from pytest_tests.resources.common import FREE_STORAGE, WALLET_CONFIG, WALLET_PASS
|
|
|
|
|
|
@dataclass
|
|
class WalletFile:
|
|
path: str
|
|
password: str = WALLET_PASS
|
|
config_path: str = WALLET_CONFIG
|
|
|
|
@staticmethod
|
|
def from_node(node: NodeBase):
|
|
return WalletFile(
|
|
node.get_wallet_path(), node.get_wallet_password(), node.get_wallet_config_path()
|
|
)
|
|
|
|
def get_address(self) -> str:
|
|
"""
|
|
Extracts the last address from wallet.
|
|
|
|
Returns:
|
|
The address of the wallet.
|
|
"""
|
|
return wallet_utils.get_last_address_from_wallet(self.path, self.password)
|
|
|
|
|
|
class WalletFactory:
|
|
def __init__(self, wallets_dir: str, shell: Shell, cluster: Cluster) -> None:
|
|
self.shell = shell
|
|
self.wallets_dir = wallets_dir
|
|
self.cluster = cluster
|
|
|
|
def create_wallet(self, password: str = WALLET_PASS) -> WalletFile:
|
|
"""
|
|
Creates new default wallet
|
|
Args:
|
|
password: wallet password
|
|
|
|
Returns:
|
|
WalletFile object of new wallet
|
|
"""
|
|
wallet_path = os.path.join(self.wallets_dir, f"{str(uuid.uuid4())}.json")
|
|
wallet_utils.init_wallet(wallet_path, password)
|
|
|
|
if not FREE_STORAGE:
|
|
main_chain = self.cluster.main_chain_nodes[0]
|
|
deposit = 30
|
|
transfer_gas(
|
|
shell=self.shell,
|
|
amount=deposit + 1,
|
|
main_chain=main_chain,
|
|
wallet_to_path=wallet_path,
|
|
wallet_to_password=password,
|
|
)
|
|
deposit_gas(
|
|
shell=self.shell,
|
|
amount=deposit,
|
|
main_chain=main_chain,
|
|
wallet_from_path=wallet_path,
|
|
wallet_from_password=password,
|
|
)
|
|
|
|
return WalletFile(wallet_path, password)
|