ref: Remove unused sync session_scope, with_session and engine (#5333)
Remove unused sync session_scope, with_session and engine
This commit is contained in:
parent
9bf372ee3e
commit
3f0e383135
20 changed files with 75 additions and 114 deletions
|
|
@ -615,14 +615,14 @@ async def test_read_only_starter_projects(client: AsyncClient, logged_in_headers
|
|||
assert len(response.json()) == len(starter_projects)
|
||||
|
||||
|
||||
def test_sqlite_pragmas():
|
||||
async def test_sqlite_pragmas():
|
||||
db_service = get_db_service()
|
||||
|
||||
with db_service.with_session() as session:
|
||||
async with db_service.with_session() as session:
|
||||
from sqlalchemy import text
|
||||
|
||||
assert session.exec(text("PRAGMA journal_mode;")).scalar() == "wal"
|
||||
assert session.exec(text("PRAGMA synchronous;")).scalar() == 1
|
||||
assert (await session.exec(text("PRAGMA journal_mode;"))).scalar() == "wal"
|
||||
assert (await session.exec(text("PRAGMA synchronous;"))).scalar() == 1
|
||||
|
||||
|
||||
@pytest.mark.usefixtures("active_user")
|
||||
|
|
|
|||
|
|
@ -11,7 +11,7 @@ from langflow.initial_setup.setup import (
|
|||
)
|
||||
from langflow.interface.types import aget_all_types_dict
|
||||
from langflow.services.database.models.folder.model import Folder
|
||||
from langflow.services.deps import async_session_scope
|
||||
from langflow.services.deps import session_scope
|
||||
from sqlalchemy.orm import selectinload
|
||||
from sqlmodel import select
|
||||
|
||||
|
|
@ -52,7 +52,7 @@ async def test_get_project_data():
|
|||
|
||||
@pytest.mark.usefixtures("client")
|
||||
async def test_create_or_update_starter_projects():
|
||||
async with async_session_scope() as session:
|
||||
async with session_scope() as session:
|
||||
# Get the number of projects returned by load_starter_projects
|
||||
num_projects = len(await load_starter_projects())
|
||||
|
||||
|
|
|
|||
|
|
@ -1,7 +1,7 @@
|
|||
import pytest
|
||||
from langflow.services.auth.utils import get_password_hash
|
||||
from langflow.services.database.models.user import User
|
||||
from langflow.services.deps import async_session_scope
|
||||
from langflow.services.deps import session_scope
|
||||
from sqlalchemy.exc import IntegrityError
|
||||
|
||||
|
||||
|
|
@ -18,7 +18,7 @@ def test_user():
|
|||
async def test_login_successful(client, test_user):
|
||||
# Adding the test user to the database
|
||||
try:
|
||||
async with async_session_scope() as session:
|
||||
async with session_scope() as session:
|
||||
session.add(test_user)
|
||||
await session.commit()
|
||||
except IntegrityError:
|
||||
|
|
|
|||
|
|
@ -22,13 +22,13 @@ from langflow.schema.properties import Properties, Source
|
|||
# Assuming you have these imports available
|
||||
from langflow.services.database.models.message import MessageCreate, MessageRead
|
||||
from langflow.services.database.models.message.model import MessageTable
|
||||
from langflow.services.deps import async_session_scope
|
||||
from langflow.services.deps import session_scope
|
||||
from langflow.services.tracing.utils import convert_to_langchain_type
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
async def created_message():
|
||||
async with async_session_scope() as session:
|
||||
async with session_scope() as session:
|
||||
message = MessageCreate(text="Test message", sender="User", sender_name="User", session_id="session_id")
|
||||
messagetable = MessageTable.model_validate(message, from_attributes=True)
|
||||
messagetables = await aadd_messagetables([messagetable], session)
|
||||
|
|
@ -37,7 +37,7 @@ async def created_message():
|
|||
|
||||
@pytest.fixture
|
||||
async def created_messages(async_session): # noqa: ARG001
|
||||
async with async_session_scope() as _session:
|
||||
async with session_scope() as _session:
|
||||
messages = [
|
||||
MessageCreate(text="Test message 1", sender="User", sender_name="User", session_id="session_id2"),
|
||||
MessageCreate(text="Test message 2", sender="User", sender_name="User", session_id="session_id2"),
|
||||
|
|
|
|||
|
|
@ -8,12 +8,12 @@ from langflow.memory import aadd_messagetables
|
|||
# Assuming you have these imports available
|
||||
from langflow.services.database.models.message import MessageCreate, MessageRead, MessageUpdate
|
||||
from langflow.services.database.models.message.model import MessageTable
|
||||
from langflow.services.deps import async_session_scope
|
||||
from langflow.services.deps import session_scope
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
async def created_message():
|
||||
async with async_session_scope() as session:
|
||||
async with session_scope() as session:
|
||||
message = MessageCreate(text="Test message", sender="User", sender_name="User", session_id="session_id")
|
||||
messagetable = MessageTable.model_validate(message, from_attributes=True)
|
||||
messagetables = await aadd_messagetables([messagetable], session)
|
||||
|
|
@ -22,7 +22,7 @@ async def created_message():
|
|||
|
||||
@pytest.fixture
|
||||
async def created_messages(session): # noqa: ARG001
|
||||
async with async_session_scope() as _session:
|
||||
async with session_scope() as _session:
|
||||
messages = [
|
||||
MessageCreate(text="Test message 1", sender="User", sender_name="User", session_id="session_id2"),
|
||||
MessageCreate(text="Test message 2", sender="User", sender_name="User", session_id="session_id2"),
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue