Add first info to landing page
This commit is contained in:
@@ -1,30 +1,35 @@
|
||||
import pytest
|
||||
from fastapi import Request
|
||||
from fastapi.testclient import TestClient
|
||||
import os
|
||||
|
||||
from allmende_payment_system.app import create_app
|
||||
import pytest
|
||||
from fastapi.testclient import TestClient
|
||||
from sqlalchemy import create_engine
|
||||
from sqlalchemy.orm import sessionmaker
|
||||
|
||||
from allmende_payment_system.app import app
|
||||
from allmende_payment_system.models import Base
|
||||
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def client():
|
||||
app = create_app()
|
||||
|
||||
async def add_ynh_headers(request: Request, call_next):
|
||||
username = request.headers.get("APS-TEST-username", "test")
|
||||
# This seems to work although headers are immutable
|
||||
# If this ever turns out to be a problem, we can use request.state instead,
|
||||
# but will have to modify app.get_user
|
||||
request.headers._list.append((b"ynh_user", username.encode("utf-8")))
|
||||
|
||||
response = await call_next(request)
|
||||
return response
|
||||
|
||||
app.middleware("http")(add_ynh_headers)
|
||||
|
||||
os.environ["APS_username"] = "test"
|
||||
return TestClient(app)
|
||||
|
||||
|
||||
@pytest.fixture(scope="session")
|
||||
def unauthorized_client():
|
||||
app = create_app()
|
||||
os.environ.pop("APS_username", None)
|
||||
return TestClient(app)
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def test_db():
|
||||
engine = create_engine("sqlite:///:memory:")
|
||||
Base.metadata.create_all(bind=engine) # Create tables
|
||||
TestingSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
|
||||
|
||||
# Provide a session and the engine
|
||||
db = TestingSessionLocal()
|
||||
try:
|
||||
yield db
|
||||
finally:
|
||||
db.close()
|
||||
|
||||
Reference in New Issue
Block a user