Method setUp

tests/test_fetch_node_timeout.py:23–55

Set up test fixtures.

(self)

    """Test suite for FetchNode timeout configuration and usage."""

    def setUp(self):
        """Set up test fixtures."""
        # Mock all the heavy external dependencies at import time
        self.mock_modules = {}
        for module in ['langchain_core', 'langchain_core.documents',
                       'langchain_community', 'langchain_community.document_loaders',
                       'langchain_openai', 'minify_html', 'pydantic',
                       'langchain', 'langchain.prompts']:
            if module not in sys.modules:
                sys.modules[module] = MagicMock()

        # Create mock Document class
        class MockDocument:
            def __init__(self, page_content, metadata=None):
                self.page_content = page_content
                self.metadata = metadata or {}

        sys.modules['langchain_core.documents'].Document = MockDocument

        # Create mock PyPDFLoader
        class MockPyPDFLoader:
            def __init__(self, source):
                self.source = source

            def load(self):
                time.sleep(0.1)  # Simulate some work
                return [MockDocument(page_content=f"PDF content from {self.source}")]

        sys.modules['langchain_community.document_loaders'].PyPDFLoader = MockPyPDFLoader

        # Now import FetchNode
        from scrapegraphai.nodes.fetch_node import FetchNode
        self.FetchNode = FetchNode

    def tearDown(self):
        """Clean up after tests."""