1+ """
2+ Additional coverage tests for payments and MCP modules.
3+ """
4+
5+ import pytest
6+ from decimal import Decimal
7+ from solana_ai_registries .payments import PaymentManager
8+ from solana_ai_registries .mcp import McpServerRegistry
9+ from solana_ai_registries .agent import AgentRegistry
10+
11+
12+ class TestPaymentManagerBasics :
13+ """Basic tests for PaymentManager to improve coverage."""
14+
15+ def test_payment_manager_init_devnet (self ):
16+ """Test PaymentManager initialization for devnet."""
17+ from solana_ai_registries .client import SolanaAIRegistriesClient
18+ client = SolanaAIRegistriesClient ()
19+ manager = PaymentManager (client , use_mainnet = False )
20+
21+ assert manager .client == client
22+
23+ def test_payment_manager_init_mainnet (self ):
24+ """Test PaymentManager initialization for mainnet."""
25+ from solana_ai_registries .client import SolanaAIRegistriesClient
26+ client = SolanaAIRegistriesClient ()
27+ manager = PaymentManager (client , use_mainnet = True )
28+
29+ assert manager .client == client
30+
31+ def test_get_token_mint_devnet (self ):
32+ """Test getting token mint for devnet."""
33+ from solana_ai_registries .client import SolanaAIRegistriesClient
34+ client = SolanaAIRegistriesClient ()
35+ manager = PaymentManager (client , use_mainnet = False )
36+
37+ assert manager .token_mint is not None
38+
39+ def test_get_token_mint_mainnet (self ):
40+ """Test getting token mint for mainnet."""
41+ from solana_ai_registries .client import SolanaAIRegistriesClient
42+ client = SolanaAIRegistriesClient ()
43+ manager = PaymentManager (client , use_mainnet = True )
44+
45+ assert manager .token_mint is not None
46+
47+
48+ class TestMcpServerRegistryBasics :
49+ """Basic tests for McpServerRegistry to improve coverage."""
50+
51+ def test_mcp_registry_init (self ):
52+ """Test McpServerRegistry initialization."""
53+ from solana_ai_registries .client import SolanaAIRegistriesClient
54+ client = SolanaAIRegistriesClient ()
55+ registry = McpServerRegistry (client )
56+
57+ assert registry .client == client
58+
59+
60+ class TestAgentRegistryBasics :
61+ """Basic tests for AgentRegistry to improve coverage."""
62+
63+ def test_agent_registry_init (self ):
64+ """Test AgentRegistry initialization."""
65+ from solana_ai_registries .client import SolanaAIRegistriesClient
66+ client = SolanaAIRegistriesClient ()
67+ registry = AgentRegistry (client )
68+
69+ assert registry .client == client
70+
71+
72+ class TestConstantsCoverage :
73+ """Test constants module for additional coverage."""
74+
75+ def test_import_constants (self ):
76+ """Test importing constants."""
77+ from solana_ai_registries .constants import (
78+ MAX_AGENT_ID_LEN ,
79+ MAX_AGENT_NAME_LEN ,
80+ A2AMPL_BASE_UNIT
81+ )
82+
83+ assert MAX_AGENT_ID_LEN > 0
84+ assert MAX_AGENT_NAME_LEN > 0
85+ assert A2AMPL_BASE_UNIT > 0
86+
87+ def test_validation_functions (self ):
88+ """Test validation functions."""
89+ from solana_ai_registries .constants import validate_string_length , validate_url
90+
91+ # These should not raise
92+ validate_string_length ("test" , 10 , "test field" )
93+ validate_url ("https://example.com" , "test url" )
94+
95+
96+ class TestExceptionsCoverage :
97+ """Test exceptions module for additional coverage."""
98+
99+ def test_exception_creation (self ):
100+ """Test creating exceptions."""
101+ from solana_ai_registries .exceptions import (
102+ SolanaAIRegistriesError ,
103+ AgentNotFoundError ,
104+ McpServerNotFoundError
105+ )
106+
107+ base_error = SolanaAIRegistriesError ("Base error" )
108+ assert str (base_error ) == "Base error"
109+
110+ agent_error = AgentNotFoundError ("test_agent" , "test_owner" )
111+ assert "test_agent" in str (agent_error )
112+
113+ mcp_error = McpServerNotFoundError ("test_server" , "test_owner" )
114+ assert "test_server" in str (mcp_error )
0 commit comments