Test: Updated to async function testing for compatibility

langchain-ai · Dec 5, 2024 · 6f99252 · 6f99252
1 parent 78d0519
commit 6f99252
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/community/tests/unit_tests/document_loaders/test_web_base.py b/libs/community/tests/unit_tests/document_loaders/test_web_base.py
@@ -64,7 +64,7 @@ def test_lazy_load(mock_get: Any) -> None:
 
 @pytest.mark.requires("bs4")
 @patch("aiohttp.ClientSession.get")
-def test_aload(mock_get: Any) -> None:
+async def test_aload(mock_get: Any) -> None:
     async def mock_text() -> str:
         return "<html><body><p>Test content</p></body></html>"
 
@@ -76,7 +76,7 @@ async def mock_text() -> str:
         web_paths=["https://www.example.com"],
         header_template={"User-Agent": "test-user-agent"},
     )
-    results = loader.aload()
+    results = await loader.aload()
     assert len(results) == 1
     assert results[0].page_content == "Test content"
     mock_get.assert_called_with(