tests/conftest.py

from typing import Dict, Generator

import pytest
from fastapi.testclient import TestClient
from sqlalchemy.orm import Session

from core.config import settings
from core.db import SessionLocal
from misc.scripts.launch import app
from cruds import crud_user
from schemas import user as user_schemas
import datetime
import requests


headers = {
    "accept": "application/json",
    "Content-Type": "application/json",
}


@pytest.fixture(scope="session")
def db() -> Generator:
    yield SessionLocal()


@pytest.fixture(scope="session")
def super_user_client() -> Generator:
    db = SessionLocal()
    super_user = user_schemas.UserCreate(
        email="[email protected]",
        user_type=1,
        password="test",
        full_name="Test Super Admin",
        is_active=True,
        address="Test Land",
        contact_number="9841111111",
        dob=datetime.datetime(2021, 1, 1),
        join_year=2021,
    )
    super_user_obj = crud_user.create(db=db, obj_in=super_user)
    crud_user.verify_user(db=db, db_obj=super_user_obj)

    data = {
        "username": "[email protected]",
        "password": "test",
        "remember_me": True,
    }

    super_user_cookies = None
    with TestClient(app) as c:
        req = c.post(
            f"{settings.BACKEND_URL_BASE}{settings.API_V1_STR}/auth/web/",
            json=data,
        )
        super_user_cookies = req.cookies

    assert req.status_code == 200
    assert req.cookies.get("session"), "Cookie not returned!"

    with TestClient(app) as c:
        c.cookies = super_user_cookies
        yield c

    crud_user.remove(db=SessionLocal(), id=super_user_obj.id)


@pytest.fixture(scope="session")
def client() -> Generator:
    with TestClient(app) as c:
        yield c