5120861e17
- Add GPU fields (model, memory, count, CUDA version, price, region) to MarketplaceOffer model - Create new marketplace_gpu router for GPU-specific operations - Update offer sync to populate GPU fields from miner capabilities - Move GPU attributes from generic attributes dict to dedicated fields - Update MarketplaceOfferView schema with GPU fields - Expand CLI README with comprehensive documentation and
245 lines
8.4 KiB
Python
245 lines
8.4 KiB
Python
"""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
|