Karim shoair commited on
Commit ·
55c8b21
1
Parent(s): 135c17b
tests: update tests for mcp
Browse files- 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)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|