From 883bf3a86a309d72ae8ddfe33c375268d77faab4 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 16 Nov 2023 09:44:35 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20fix(test=5Fstore.py):=20fix=20im?= =?UTF-8?q?port=20statement=20for=20CreateComponentResponse=20in=20test=5F?= =?UTF-8?q?search=5Fcomponents=20function=20=F0=9F=9A=80=20feat(test=5Fsto?= =?UTF-8?q?re.py):=20update=20import=20statement=20for=20CreateComponentRe?= =?UTF-8?q?sponse=20in=20test=5Fsearch=5Fcomponents=20function=20to=20refl?= =?UTF-8?q?ect=20changes=20in=20schema?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tests/test_store.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_store.py b/tests/test_store.py index c4fbcca7e..d139537fa 100644 --- a/tests/test_store.py +++ b/tests/test_store.py @@ -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)