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

🔧 fix(conftest.py): remove unused imports and commented out code

Browse source 
✨ feat(conftest.py): add session_getter fixture to create a blank session for testing
🔧 fix(test_custom_component.py): pass session_getter fixture to list_flows function for testing
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-07-31 17:31:40 -03:00
commit ddafdf31b5
2 changed files with 34 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
tests/test_custom_component.py
View file

@ -517,13 +517,13 @@ def db(app):
app.db.drop_all()
def test_list_flows_return_type(component):
flows = component.list_flows()
def test_list_flows_return_type(component, session_getter):
flows = component.list_flows(get_session=session_getter)
assert isinstance(flows, list)
def test_list_flows_flow_objects(component):
flows = component.list_flows()
def test_list_flows_flow_objects(component, session_getter):
flows = component.list_flows(get_session=session_getter)
assert all(isinstance(flow, Flow) for flow in flows)