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
387 lines
14 KiB
Python
Executable File
387 lines
14 KiB
Python
Executable File
"""Tests for client CLI commands"""
|
|
|
|
import pytest
|
|
import json
|
|
from click.testing import CliRunner
|
|
from unittest.mock import Mock, patch
|
|
from aitbc_cli.commands.client import client
|
|
|
|
|
|
@pytest.fixture
|
|
def runner():
|
|
"""Create CLI runner"""
|
|
return CliRunner()
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_config():
|
|
"""Mock configuration"""
|
|
config = Mock()
|
|
config.coordinator_url = "http://test:8000"
|
|
config.api_key = "test_key"
|
|
return config
|
|
|
|
|
|
class TestClientCommands:
|
|
"""Test client command group"""
|
|
|
|
@patch('aitbc_cli.commands.client.httpx.Client')
|
|
def test_submit_job_success(self, mock_client_class, runner, mock_config):
|
|
"""Test successful job submission"""
|
|
# Setup mock
|
|
mock_client = Mock()
|
|
mock_client_class.return_value.__enter__.return_value = mock_client
|
|
mock_response = Mock()
|
|
mock_response.status_code = 201
|
|
mock_response.json.return_value = {"job_id": "test_job_123"}
|
|
mock_client.post.return_value = mock_response
|
|
|
|
# Run command
|
|
result = runner.invoke(client, [
|
|
'submit',
|
|
'--type', 'inference',
|
|
'--prompt', 'Test prompt',
|
|
'--model', 'test_model'
|
|
], obj={'config': mock_config, 'output_format': 'json'})
|
|
|
|
# Assertions
|
|
assert result.exit_code == 0
|
|
assert 'job_id' in result.output
|
|
|
|
# Verify API call
|
|
mock_client.post.assert_called_once()
|
|
call_args = mock_client.post.call_args
|
|
assert '/v1/jobs' in call_args[0][0]
|
|
assert call_args[1]['json']['payload']['type'] == 'inference'
|
|
assert call_args[1]['json']['payload']['prompt'] == 'Test prompt'
|
|
|
|
@patch('aitbc_cli.commands.client.httpx.Client')
|
|
def test_submit_job_from_file(self, mock_client_class, runner, mock_config, tmp_path):
|
|
"""Test job submission from file"""
|
|
# Create test job file
|
|
job_file = tmp_path / "test_job.json"
|
|
job_data = {
|
|
"type": "training",
|
|
"model": "gpt-3",
|
|
"dataset": "test_data"
|
|
}
|
|
job_file.write_text(json.dumps(job_data))
|
|
|
|
# Setup mock
|
|
mock_client = Mock()
|
|
mock_client_class.return_value.__enter__.return_value = mock_client
|
|
mock_response = Mock()
|
|
mock_response.status_code = 201
|
|
mock_response.json.return_value = {"job_id": "test_job_456"}
|
|
mock_client.post.return_value = mock_response
|
|
|
|
# Run command
|
|
result = runner.invoke(client, [
|
|
'submit',
|
|
'--file', str(job_file)
|
|
], obj={'config': mock_config, 'output_format': 'json'})
|
|
|
|
# Assertions
|
|
assert result.exit_code == 0
|
|
assert 'job_id' in result.output
|
|
|
|
# Verify API call used file data
|
|
call_args = mock_client.post.call_args
|
|
assert call_args[1]['json']['payload']['type'] == 'training'
|
|
assert call_args[1]['json']['payload']['model'] == 'gpt-3'
|
|
|
|
@patch('aitbc_cli.commands.client.httpx.Client')
|
|
def test_status_success(self, mock_client_class, runner, mock_config):
|
|
"""Test successful job status check"""
|
|
# Setup mock
|
|
mock_client = Mock()
|
|
mock_client_class.return_value.__enter__.return_value = mock_client
|
|
mock_response = Mock()
|
|
mock_response.status_code = 200
|
|
mock_response.json.return_value = {
|
|
"job_id": "test_job_123",
|
|
"state": "completed",
|
|
"result": "Test result"
|
|
}
|
|
mock_client.get.return_value = mock_response
|
|
|
|
# Run command
|
|
result = runner.invoke(client, [
|
|
'status',
|
|
'test_job_123'
|
|
], obj={'config': mock_config, 'output_format': 'json'})
|
|
|
|
# Assertions
|
|
assert result.exit_code == 0
|
|
assert 'completed' in result.output
|
|
assert 'test_job_123' in result.output
|
|
|
|
# Verify API call
|
|
mock_client.get.assert_called_once_with(
|
|
'http://test:8000/v1/jobs/test_job_123',
|
|
headers={"X-Api-Key": "test_key"}
|
|
)
|
|
|
|
@patch('aitbc_cli.commands.client.httpx.Client')
|
|
def test_cancel_job_success(self, mock_client_class, runner, mock_config):
|
|
"""Test successful job cancellation"""
|
|
# Setup mock
|
|
mock_client = Mock()
|
|
mock_client_class.return_value.__enter__.return_value = mock_client
|
|
mock_response = Mock()
|
|
mock_response.status_code = 200
|
|
mock_client.post.return_value = mock_response
|
|
|
|
# Run command
|
|
result = runner.invoke(client, [
|
|
'cancel',
|
|
'test_job_123'
|
|
], obj={'config': mock_config, 'output_format': 'json'})
|
|
|
|
# Assertions
|
|
assert result.exit_code == 0
|
|
|
|
# Verify API call
|
|
mock_client.post.assert_called_once_with(
|
|
'http://test:8000/v1/jobs/test_job_123/cancel',
|
|
headers={"X-Api-Key": "test_key"}
|
|
)
|
|
|
|
@patch('aitbc_cli.commands.client.httpx.Client')
|
|
def test_blocks_success(self, mock_client_class, runner, mock_config):
|
|
"""Test successful blocks listing"""
|
|
# Setup mock
|
|
mock_client = Mock()
|
|
mock_client_class.return_value.__enter__.return_value = mock_client
|
|
mock_response = Mock()
|
|
mock_response.status_code = 200
|
|
mock_response.json.return_value = {
|
|
"items": [
|
|
{"height": 100, "hash": "0x123"},
|
|
{"height": 101, "hash": "0x456"}
|
|
]
|
|
}
|
|
mock_client.get.return_value = mock_response
|
|
|
|
# Run command
|
|
result = runner.invoke(client, [
|
|
'blocks',
|
|
'--limit', '2'
|
|
], obj={'config': mock_config, 'output_format': 'json'})
|
|
|
|
# Assertions
|
|
assert result.exit_code == 0
|
|
assert 'items' in result.output
|
|
|
|
# Verify API call
|
|
mock_client.get.assert_called_once()
|
|
call_args = mock_client.get.call_args
|
|
assert '/v1/explorer/blocks' in call_args[0][0]
|
|
assert call_args[1]['params']['limit'] == 2
|
|
|
|
@patch('aitbc_cli.commands.client.httpx.Client')
|
|
def test_history_with_filters(self, mock_client_class, runner, mock_config):
|
|
"""Test job history with filters"""
|
|
# Setup mock
|
|
mock_client = Mock()
|
|
mock_client_class.return_value.__enter__.return_value = mock_client
|
|
mock_response = Mock()
|
|
mock_response.status_code = 200
|
|
mock_response.json.return_value = {
|
|
"jobs": [
|
|
{"id": "job1", "status": "completed"},
|
|
{"id": "job2", "status": "failed"}
|
|
]
|
|
}
|
|
mock_client.get.return_value = mock_response
|
|
|
|
# Run command with filters
|
|
result = runner.invoke(client, [
|
|
'history',
|
|
'--status', 'completed',
|
|
'--type', 'inference',
|
|
'--limit', '10'
|
|
], obj={'config': mock_config, 'output_format': 'json'})
|
|
|
|
# Assertions
|
|
assert result.exit_code == 0
|
|
|
|
# Verify API call with filters
|
|
mock_client.get.assert_called_once()
|
|
call_args = mock_client.get.call_args
|
|
assert call_args[1]['params']['status'] == 'completed'
|
|
assert call_args[1]['params']['type'] == 'inference'
|
|
assert call_args[1]['params']['limit'] == 10
|
|
|
|
@patch('aitbc_cli.commands.client.httpx.Client')
|
|
def test_api_error_handling(self, mock_client_class, runner, mock_config):
|
|
"""Test API error handling"""
|
|
# Setup mock for error response
|
|
mock_client = Mock()
|
|
mock_client_class.return_value.__enter__.return_value = mock_client
|
|
mock_response = Mock()
|
|
mock_response.status_code = 500
|
|
mock_response.text = "Internal Server Error"
|
|
mock_client.get.return_value = mock_response
|
|
|
|
# Run command
|
|
result = runner.invoke(client, [
|
|
'status',
|
|
'test_job_123'
|
|
], obj={'config': mock_config, 'output_format': 'json'})
|
|
|
|
# Assertions
|
|
assert result.exit_code != 0
|
|
assert 'Error' in result.output
|
|
|
|
def test_submit_missing_required_args(self, runner, mock_config):
|
|
"""Test submit command with missing required arguments"""
|
|
result = runner.invoke(client, [
|
|
'submit'
|
|
], obj={'config': mock_config, 'output_format': 'json'})
|
|
|
|
assert result.exit_code != 0
|
|
assert 'Error' in result.output
|
|
|
|
@patch('aitbc_cli.commands.client.httpx.Client')
|
|
def test_pay_command_success(self, mock_client_class, runner, mock_config):
|
|
"""Test creating a payment for a job"""
|
|
mock_client = Mock()
|
|
mock_client_class.return_value.__enter__.return_value = mock_client
|
|
mock_response = Mock()
|
|
mock_response.status_code = 201
|
|
mock_response.json.return_value = {
|
|
"job_id": "job_123",
|
|
"payment_id": "pay_abc",
|
|
"amount": 10.0,
|
|
"currency": "AITBC",
|
|
"status": "escrowed"
|
|
}
|
|
mock_client.post.return_value = mock_response
|
|
|
|
result = runner.invoke(client, [
|
|
'pay', 'job_123', '10.0',
|
|
'--currency', 'AITBC',
|
|
'--method', 'aitbc_token'
|
|
], obj={'config': mock_config, 'output_format': 'json'})
|
|
|
|
assert result.exit_code == 0
|
|
assert 'pay_abc' in result.output
|
|
|
|
@patch('aitbc_cli.commands.client.httpx.Client')
|
|
def test_pay_command_failure(self, mock_client_class, runner, mock_config):
|
|
"""Test payment creation failure"""
|
|
mock_client = Mock()
|
|
mock_client_class.return_value.__enter__.return_value = mock_client
|
|
mock_response = Mock()
|
|
mock_response.status_code = 400
|
|
mock_response.text = "Bad Request"
|
|
mock_client.post.return_value = mock_response
|
|
|
|
result = runner.invoke(client, [
|
|
'pay', 'job_123', '10.0'
|
|
], obj={'config': mock_config, 'output_format': 'json'})
|
|
|
|
assert result.exit_code != 0
|
|
assert 'Payment failed' in result.output
|
|
|
|
@patch('aitbc_cli.commands.client.httpx.Client')
|
|
def test_payment_status_success(self, mock_client_class, runner, mock_config):
|
|
"""Test getting payment status for a job"""
|
|
mock_client = Mock()
|
|
mock_client_class.return_value.__enter__.return_value = mock_client
|
|
mock_response = Mock()
|
|
mock_response.status_code = 200
|
|
mock_response.json.return_value = {
|
|
"job_id": "job_123",
|
|
"payment_id": "pay_abc",
|
|
"status": "escrowed",
|
|
"amount": 10.0
|
|
}
|
|
mock_client.get.return_value = mock_response
|
|
|
|
result = runner.invoke(client, [
|
|
'payment-status', 'job_123'
|
|
], obj={'config': mock_config, 'output_format': 'json'})
|
|
|
|
assert result.exit_code == 0
|
|
assert 'escrowed' in result.output
|
|
|
|
@patch('aitbc_cli.commands.client.httpx.Client')
|
|
def test_payment_status_not_found(self, mock_client_class, runner, mock_config):
|
|
"""Test payment status when no payment exists"""
|
|
mock_client = Mock()
|
|
mock_client_class.return_value.__enter__.return_value = mock_client
|
|
mock_response = Mock()
|
|
mock_response.status_code = 404
|
|
mock_client.get.return_value = mock_response
|
|
|
|
result = runner.invoke(client, [
|
|
'payment-status', 'job_999'
|
|
], obj={'config': mock_config, 'output_format': 'json'})
|
|
|
|
assert result.exit_code != 0
|
|
assert 'No payment found' in result.output
|
|
|
|
@patch('aitbc_cli.commands.client.httpx.Client')
|
|
def test_payment_receipt_success(self, mock_client_class, runner, mock_config):
|
|
"""Test getting a payment receipt"""
|
|
mock_client = Mock()
|
|
mock_client_class.return_value.__enter__.return_value = mock_client
|
|
mock_response = Mock()
|
|
mock_response.status_code = 200
|
|
mock_response.json.return_value = {
|
|
"payment_id": "pay_abc",
|
|
"job_id": "job_123",
|
|
"amount": 10.0,
|
|
"status": "released",
|
|
"transaction_hash": "0xabc123"
|
|
}
|
|
mock_client.get.return_value = mock_response
|
|
|
|
result = runner.invoke(client, [
|
|
'payment-receipt', 'pay_abc'
|
|
], obj={'config': mock_config, 'output_format': 'json'})
|
|
|
|
assert result.exit_code == 0
|
|
assert '0xabc123' in result.output
|
|
|
|
@patch('aitbc_cli.commands.client.httpx.Client')
|
|
def test_refund_success(self, mock_client_class, runner, mock_config):
|
|
"""Test requesting a refund"""
|
|
mock_client = Mock()
|
|
mock_client_class.return_value.__enter__.return_value = mock_client
|
|
mock_response = Mock()
|
|
mock_response.status_code = 200
|
|
mock_response.json.return_value = {
|
|
"status": "refunded",
|
|
"payment_id": "pay_abc"
|
|
}
|
|
mock_client.post.return_value = mock_response
|
|
|
|
result = runner.invoke(client, [
|
|
'refund', 'job_123', 'pay_abc',
|
|
'--reason', 'Job timed out'
|
|
], obj={'config': mock_config, 'output_format': 'json'})
|
|
|
|
assert result.exit_code == 0
|
|
assert 'refunded' in result.output
|
|
|
|
@patch('aitbc_cli.commands.client.httpx.Client')
|
|
def test_refund_failure(self, mock_client_class, runner, mock_config):
|
|
"""Test refund failure"""
|
|
mock_client = Mock()
|
|
mock_client_class.return_value.__enter__.return_value = mock_client
|
|
mock_response = Mock()
|
|
mock_response.status_code = 400
|
|
mock_response.text = "Cannot refund released payment"
|
|
mock_client.post.return_value = mock_response
|
|
|
|
result = runner.invoke(client, [
|
|
'refund', 'job_123', 'pay_abc',
|
|
'--reason', 'Changed mind'
|
|
], obj={'config': mock_config, 'output_format': 'json'})
|
|
|
|
assert result.exit_code != 0
|
|
assert 'Refund failed' in result.output
|