diff --git a/tests/test_data_components.py b/tests/test_data_components.py index cbaa39733..ff65f408c 100644 --- a/tests/test_data_components.py +++ b/tests/test_data_components.py @@ -28,7 +28,9 @@ async def test_successful_get_request(api_request): respx.get(url).mock(return_value=Response(200, json=mock_response)) # Making the request - result = await api_request.make_request(client=httpx.AsyncClient(), method=method, url=url) + result = await api_request.make_request( + client=httpx.AsyncClient(), method=method, url=url + ) # Assertions assert result.data["status_code"] == 200 @@ -44,7 +46,9 @@ async def test_failed_request(api_request): respx.get(url).mock(return_value=Response(404)) # Making the request - result = await api_request.make_request(client=httpx.AsyncClient(), method=method, url=url) + result = await api_request.make_request( + client=httpx.AsyncClient(), method=method, url=url + ) # Assertions assert result.data["status_code"] == 404 @@ -56,10 +60,14 @@ async def test_timeout(api_request): # Mocking a timeout url = "https://example.com/api/timeout" method = "GET" - respx.get(url).mock(side_effect=httpx.TimeoutException(message="Timeout", request=None)) + respx.get(url).mock( + side_effect=httpx.TimeoutException(message="Timeout", request=None) + ) # Making the request - result = await api_request.make_request(client=httpx.AsyncClient(), method=method, url=url, timeout=1) + result = await api_request.make_request( + client=httpx.AsyncClient(), method=method, url=url, timeout=1 + ) # Assertions assert result.data["status_code"] == 408 @@ -126,7 +134,9 @@ def test_directory_component_build_with_multithreading( # Assert mock_resolve_path.assert_called_once_with(path) - mock_retrieve_file_paths.assert_called_once_with(path, types, load_hidden, recursive, depth) + mock_retrieve_file_paths.assert_called_once_with( + path, types, load_hidden, recursive, depth + ) mock_parallel_load_records.assert_called_once_with( mock_retrieve_file_paths.return_value, silent_errors, max_concurrency ) @@ -157,3 +167,9 @@ def test_directory_without_mocks(): results = directory_component.build(str(docs_path), use_multithreading=False) docs_files = list(docs_path.glob("*.mdx")) assert len(results) == len(docs_files) + + +def test_url_component(): + url_component = data.URLComponent() + # the url component can be used to load the contents of a website + record = url_component.build("https://example.com")