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

🔧 fix(test_store.py): fix import statement for CreateComponentResponse in test_search_components function

Browse source 
🚀 feat(test_store.py): update import statement for CreateComponentResponse in test_search_components function to reflect changes in schema
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-16 09:44:35 -03:00
commit 883bf3a86a
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
tests/test_store.py
View file

@ -1,7 +1,7 @@
# FILEPATH: /Users/ogabrielluiz/Projects/langflow2/tests/test_store_service.py
from datetime import datetime
from unittest.mock import patch, Mock
from unittest.mock import Mock, patch
from langflow.services.deps import get_store_service
@ -9,7 +9,7 @@ from langflow.services.deps import get_store_service
@patch("langflow.services.store.service.httpx")
def test_search_components(mock_httpx: Mock, client):
# Mock the response from the HTTP GET request
from langflow.services.store.schema import ComponentResponse
from langflow.services.store.schema import CreateComponentResponse
mock_response = Mock()
mock_response.json.return_value = {
@ -54,4 +54,4 @@ def test_search_components(mock_httpx: Mock, client):
# Assert that the search method returns a list of ComponentResponse objects
assert len(components) == 2
assert all(isinstance(component, ComponentResponse) for component in components)
assert all(isinstance(component, CreateComponentResponse) for component in components)