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

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

Browse source 
🔧 fix(tests/conftest.py): remove unused fixture 'client'
🔧 fix(tests/conftest.py): remove unused import of 'langflow.main.create_app'
🔧 fix(tests/conftest.py): remove unused fixture 'client_fixture'
🔧 fix(tests/conftest.py): remove unused import of 'langflow.services.database.manager.DatabaseManager'
🔧 fix(tests/conftest.py): remove unused import of 'typer.testing.CliRunner'
🔧 fix(tests/conftest.py): remove unused import of 'sqlmodel.SQLModel'
🔧 fix(tests/conftest.py): remove unused import of 'sqlmodel.Session'
🔧 fix(tests/conftest.py): remove unused import of 'sqlmodel.create_engine'
🔧 fix(tests/conftest.py): remove unused import of 'sqlmodel.pool.StaticPool'
🔧 fix(tests/conftest.py): remove unused import of 'tempfile'
🔧 fix(tests/conftest.py): remove unused import of 'Path'
🔧 fix(tests/conftest.py): remove unused import of 'contextmanager'
🔧 fix(tests/test_custom_component.py): remove unused argument 'session_getter' from test functions
🔧 fix(tests/test_custom_component.py): remove unused import of 'db'
🔧 fix(tests/test_custom_component.py): remove unused import of 'Flow'
🔧 fix(tests/test_setup_superuser.py): remove unused import of 'DEFAULT_SUPERUSER_PASSWORD'
🔧 fix(tests/test_setup_superuser.py): remove unused import of 'Mock'
🔧 fix(tests/test_setup_superuser.py): remove unused import of 'User'
🔧 fix(tests/test_setup_superuser.py): remove unused import of 'reset_mock_credentials'
🔧 fix(tests/test_setup_superuser.py): remove unused import of 'mock_create_super_user'
🔧 fix(tests/test_setup_superuser.py): remove unused import of 'setup_superuser'
🔧 fix(tests/test_setup_superuser.py): remove unused import of 'mock_user'
🔧 fix(tests/test_setup_superuser.py): remove unused import of 'mock_get_session'
🔧 fix(tests/test_setup_superuser.py): remove unused import of 'mock_get_settings_manager'
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-09-20 14:32:14 -03:00
commit 82469752f8
3 changed files with 32 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

27
tests/conftest.py
View file

@ -15,6 +15,9 @@ from sqlmodel import SQLModel, Session, create_engine
from sqlmodel.pool import StaticPool
from typer.testing import CliRunner
# we need to import tmpdir
import tempfile
if TYPE_CHECKING:
from langflow.services.database.manager import DatabaseManager
@ -46,14 +49,14 @@ async def async_client() -> AsyncGenerator:
# Create client fixture for FastAPI
@pytest.fixture(scope="module", autouse=True)
def client():
from langflow.main import create_app
# @pytest.fixture(scope="module", autouse=True)
# def client():
# from langflow.main import create_app
app = create_app()
# app = create_app()
with TestClient(app) as client:
yield client
# with TestClient(app) as client:
# yield client
def get_graph(_type="basic"):
@ -111,8 +114,13 @@ def session_fixture():
yield session
@pytest.fixture(name="client")
def client_fixture(session: Session):
@pytest.fixture(name="client", autouse=True)
def client_fixture(session: Session, monkeypatch):
# Set the database url to a test database
db_dir = tempfile.mkdtemp()
db_path = Path(db_dir) / "test.db"
monkeypatch.setenv("LANGFLOW_DATABASE_URL", f"sqlite:///{db_path}")
def get_session_override():
return session
@ -124,6 +132,9 @@ def client_fixture(session: Session):
with TestClient(app) as client:
yield client
app.dependency_overrides.clear()
monkeypatch.undo()
# clear the temp db
db_path.unlink()
# @contextmanager