Karim shoair commited on
Commit
55c8b21
·
1 Parent(s): 135c17b

tests: update tests for mcp

Browse files
Files changed (1) hide show
  1. tests/ai/test_ai_mcp.py +0 -12
tests/ai/test_ai_mcp.py CHANGED
@@ -1,6 +1,5 @@
1
  import pytest
2
  import pytest_httpbin
3
- from unittest.mock import Mock, patch
4
 
5
  from scrapling.core.ai import ScraplingMCPServer, ResponseModel
6
 
@@ -17,11 +16,6 @@ class TestMCPServer:
17
  def server(self):
18
  return ScraplingMCPServer()
19
 
20
- def test_server_creation(self, server):
21
- """Test server instance creation"""
22
- assert server._server is not None
23
- assert server._server.name == "Scrapling"
24
-
25
  def test_get_tool(self, server, test_url):
26
  """Test the get tool method"""
27
  result = server.get(url=test_url, extraction_type="markdown")
@@ -62,9 +56,3 @@ class TestMCPServer:
62
  """Test the bulk_stealthy_fetch tool method"""
63
  result = await server.bulk_stealthy_fetch(urls=(test_url, test_url), headless=True)
64
  assert all(isinstance(r, ResponseModel) for r in result)
65
-
66
- def test_serve_method(self, server):
67
- """Test the serve method"""
68
- with patch.object(server._server, 'run') as mock_run:
69
- server.serve()
70
- mock_run.assert_called_once_with(transport="stdio")
 
1
  import pytest
2
  import pytest_httpbin
 
3
 
4
  from scrapling.core.ai import ScraplingMCPServer, ResponseModel
5
 
 
16
  def server(self):
17
  return ScraplingMCPServer()
18
 
 
 
 
 
 
19
  def test_get_tool(self, server, test_url):
20
  """Test the get tool method"""
21
  result = server.get(url=test_url, extraction_type="markdown")
 
56
  """Test the bulk_stealthy_fetch tool method"""
57
  result = await server.bulk_stealthy_fetch(urls=(test_url, test_url), headless=True)
58
  assert all(isinstance(r, ResponseModel) for r in result)