@@ -169,9 +169,9 @@ def test_agent_registry_entry_edge_cases(self):
169169
170170 def test_service_endpoint_edge_cases (self ):
171171 """Test ServiceEndpoint with different protocols."""
172- # Test explicit protocol setting
172+ # Test explicit protocol setting with valid URL
173173 endpoint = ServiceEndpoint (
174- url = "custom ://example.com" , protocol = "custom" , auth_type = "bearer"
174+ url = "https ://example.com" , protocol = "custom" , auth_type = "bearer"
175175 )
176176 assert endpoint .protocol == "custom"
177177 assert endpoint .auth_type == "bearer"
@@ -222,7 +222,7 @@ def test_client_with_custom_commitment(self, mock_async_client):
222222class TestAgentRegistryAdditionalCoverage :
223223 """Additional tests for agent registry functionality."""
224224
225- @patch ("solana_ai_registries.agent .AsyncClient" )
225+ @patch ("solana_ai_registries.client .AsyncClient" )
226226 def test_agent_registry_initialization (self , mock_async_client ):
227227 """Test AgentRegistry initialization."""
228228 mock_client = Mock ()
@@ -233,7 +233,7 @@ def test_agent_registry_initialization(self, mock_async_client):
233233class TestMcpServerRegistryAdditionalCoverage :
234234 """Additional tests for MCP server registry functionality."""
235235
236- @patch ("solana_ai_registries.mcp .AsyncClient" )
236+ @patch ("solana_ai_registries.client .AsyncClient" )
237237 def test_mcp_registry_initialization (self , mock_async_client ):
238238 """Test McpServerRegistry initialization."""
239239 mock_client = Mock ()
@@ -244,7 +244,7 @@ def test_mcp_registry_initialization(self, mock_async_client):
244244class TestPaymentManagerAdditionalCoverage :
245245 """Additional tests for payment manager functionality."""
246246
247- @patch ("solana_ai_registries.payments .AsyncClient" )
247+ @patch ("solana_ai_registries.client .AsyncClient" )
248248 def test_payment_manager_initialization (self , mock_async_client ):
249249 """Test PaymentManager initialization."""
250250 mock_client = Mock ()
@@ -313,7 +313,7 @@ def test_all_validation_scenarios(self):
313313 # String length validation
314314 validate_string_length ("short" , 10 , "test" )
315315
316- with pytest .raises (ValueError , match = "test must be at most " ):
316+ with pytest .raises (ValueError , match = "test exceeds maximum length " ):
317317 validate_string_length ("very_long_string_that_exceeds_limit" , 5 , "test" )
318318
319319 # URL validation
@@ -335,7 +335,7 @@ def test_all_validation_scenarios(self):
335335 ]
336336
337337 for url in invalid_urls :
338- with pytest .raises (ValueError , match = "must be a valid URL " ):
338+ with pytest .raises (ValueError , match = "must start with one of " ):
339339 validate_url (url , "test_url" )
340340
341341 # Invalid schemes
0 commit comments