def test_get_id_from_config(bounty_collector):
config_file_name = 'test_node_config'
node_index = 1
config_node = ConfigStorage(config_file_name)
config_node.update({'node_id': node_index})
node_id = bounty_collector.get_id_from_config(config_file_name)
assert node_id == node_index
def test_init_base_agent(skale):
print("Test agent init with given node id")
agent0 = BaseAgent(skale, 0)
assert agent0.id == 0
print("Test agent init without given node id - read id from file")
config_node = ConfigStorage(NODE_CONFIG_FILEPATH)
config_node.update({'node_id': 1})
agent1 = BaseAgent(skale)
assert agent1.id == 1
print("Test agent init with non-existing node id")
with pytest.raises(NodeNotFoundException):
BaseAgent(skale, 100)
print("Test agent init with non-integer node id")
config_node.update({'node_id': 'one'})
with pytest.raises(Exception):
BaseAgent(skale)
def prepare_wallets(count):
account_dict = {"address": "0x0", "private_key": "0x0"}
for i in range(count):
local_wallet_config = ConfigStorage(LOCAL_WALLET_FILEPATH + str(i))
local_wallet_config.update(account_dict)
def test_config_storage():
config_node1 = ConfigStorage(TEST_DIR1)
config_node1.update({'node_id': 1})
node_id = config_node1['node_id']
assert node_id == 1
node_id = config_node1.safe_get('node_id')
assert node_id == 1
node_id = config_node1.safe_get('wrong_node_id')
assert node_id is None
config_node2 = ConfigStorage(TEST_DIR2, {'node_id': 2})
node_id = config_node2['node_id']
assert node_id == 2
config_node2.update({'node_id': 3})
node_id = config_node2['node_id']
assert node_id == 3
config_node2['node_id'] = 4
node_id = config_node2['node_id']
assert node_id == 4
test_config = config_node2.get()
node_id = test_config['node_id']
assert node_id == 4