jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor test cases and add new test for URLComponent

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-03-08 18:08:08 -03:00
commit 2207b78fff
1 changed files with 21 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

26
tests/test_data_components.py
View file

@ -28,7 +28,9 @@ async def test_successful_get_request(api_request):
respx.get(url).mock(return_value=Response(200, json=mock_response)) respx.get(url).mock(return_value=Response(200, json=mock_response))
# Making the request # 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 # Assertions
assert result.data["status_code"] == 200 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)) respx.get(url).mock(return_value=Response(404))
# Making the request # 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 # Assertions
assert result.data["status_code"] == 404 assert result.data["status_code"] == 404
@ -56,10 +60,14 @@ async def test_timeout(api_request):
# Mocking a timeout # Mocking a timeout
url = "https://example.com/api/timeout" url = "https://example.com/api/timeout"
method = "GET" 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 # 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 # Assertions
assert result.data["status_code"] == 408 assert result.data["status_code"] == 408
@ -126,7 +134,9 @@ def test_directory_component_build_with_multithreading(
# Assert # Assert
mock_resolve_path.assert_called_once_with(path) 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_parallel_load_records.assert_called_once_with(
mock_retrieve_file_paths.return_value, silent_errors, max_concurrency 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) results = directory_component.build(str(docs_path), use_multithreading=False)
docs_files = list(docs_path.glob("*.mdx")) docs_files = list(docs_path.glob("*.mdx"))
assert len(results) == len(docs_files) 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")