15427c96c0
- Change file mode from 644 to 755 for all project files - Add chain_id parameter to get_balance RPC endpoint with default "ait-devnet" - Rename Miner.extra_meta_data to extra_metadata for consistency
78 lines
2.7 KiB
Python
Executable File
78 lines
2.7 KiB
Python
Executable File
"""Tests for multi-chain management CLI commands"""
|
|
|
|
import pytest
|
|
from click.testing import CliRunner
|
|
from unittest.mock import Mock, patch
|
|
from aitbc_cli.commands.chain import chain
|
|
|
|
@pytest.fixture
|
|
def runner():
|
|
"""Create CLI runner"""
|
|
return CliRunner()
|
|
|
|
@pytest.fixture
|
|
def mock_chain_manager():
|
|
"""Mock ChainManager"""
|
|
with patch('aitbc_cli.commands.chain.ChainManager') as mock:
|
|
yield mock.return_value
|
|
|
|
@pytest.fixture
|
|
def mock_config():
|
|
"""Mock configuration loader"""
|
|
with patch('aitbc_cli.commands.chain.load_multichain_config') as mock:
|
|
yield mock
|
|
|
|
class TestChainAddCommand:
|
|
"""Test chain add command"""
|
|
|
|
def test_add_chain_success(self, runner, mock_config, mock_chain_manager):
|
|
"""Test successful addition of a chain to a node"""
|
|
# Setup mock
|
|
mock_chain_manager.add_chain_to_node.return_value = True
|
|
|
|
# Run command
|
|
result = runner.invoke(chain, ['add', 'chain-123', 'node-456'])
|
|
|
|
# Assertions
|
|
assert result.exit_code == 0
|
|
assert "added to node" in result.output
|
|
mock_chain_manager.add_chain_to_node.assert_called_once_with('chain-123', 'node-456')
|
|
|
|
def test_add_chain_failure(self, runner, mock_config, mock_chain_manager):
|
|
"""Test failure when adding a chain to a node"""
|
|
# Setup mock
|
|
mock_chain_manager.add_chain_to_node.return_value = False
|
|
|
|
# Run command
|
|
result = runner.invoke(chain, ['add', 'chain-123', 'node-456'])
|
|
|
|
# Assertions
|
|
assert result.exit_code != 0
|
|
assert "Failed to add chain" in result.output
|
|
mock_chain_manager.add_chain_to_node.assert_called_once_with('chain-123', 'node-456')
|
|
|
|
def test_add_chain_exception(self, runner, mock_config, mock_chain_manager):
|
|
"""Test exception handling during chain addition"""
|
|
# Setup mock
|
|
mock_chain_manager.add_chain_to_node.side_effect = Exception("Connection error")
|
|
|
|
# Run command
|
|
result = runner.invoke(chain, ['add', 'chain-123', 'node-456'])
|
|
|
|
# Assertions
|
|
assert result.exit_code != 0
|
|
assert "Error adding chain to node: Connection error" in result.output
|
|
mock_chain_manager.add_chain_to_node.assert_called_once_with('chain-123', 'node-456')
|
|
|
|
def test_add_chain_missing_args(self, runner):
|
|
"""Test command with missing arguments"""
|
|
# Missing node_id
|
|
result = runner.invoke(chain, ['add', 'chain-123'])
|
|
assert result.exit_code != 0
|
|
assert "Missing argument" in result.output
|
|
|
|
# Missing chain_id and node_id
|
|
result = runner.invoke(chain, ['add'])
|
|
assert result.exit_code != 0
|
|
assert "Missing argument" in result.output
|