5273b1866f
- Update all agent command endpoints to remove /v1 prefix for API consistency - Rename `success` variable to `is_success` in chain.py to avoid conflict with success() function - Rename `output` parameter to `output_file` in genesis.py for clarity - Add admin command help tests to verify command structure - Update blockchain status endpoint from /status to /v1/health in tests - Mark admin help command as working
78 lines
2.7 KiB
Python
78 lines
2.7 KiB
Python
"""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
|