Skip to content

Commit 5f018e4

Browse files
committed
replaced database.py with conftest file
1 parent 937e2b0 commit 5f018e4

File tree

2 files changed

+21
-3
lines changed

2 files changed

+21
-3
lines changed

test/database.py renamed to test/conftest.py

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,10 +8,12 @@
88
from app.config import settings
99
from app.database import get_db
1010
from app.database import Base
11+
from app.oauth2 import create_access_token
12+
from app import models
1113
from alembic import command
1214

1315

14-
SQLALCHEMY_DATABASE_URL = f'postgresql://{settings.database_username}:{settings.database_password}@{settings.database_hostname}:{settings.database_port}/{settings.database_name}_test'
16+
SQLALCHEMY_DATABASE_URL = f'postgresql://{settings.database_user}:{settings.database_password}@{settings.database_hostname}:{settings.database_port}/{settings.database_db_name}_test'
1517

1618

1719
engine = create_engine(SQLALCHEMY_DATABASE_URL)
@@ -31,5 +33,13 @@ def session():
3133
db.close()
3234

3335

34-
35-
36+
@pytest.fixture()
37+
def client(session):
38+
def override_get_db():
39+
40+
try:
41+
yield session
42+
finally:
43+
session.close()
44+
app.dependency_overrides[get_db] = override_get_db
45+
yield TestClient(app)

test/test_users.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,13 @@
11
import pytest
22
from jose import jwt
33
from app import schemas
4+
from app.config import settings
45

56

7+
def test_create_user(client):
8+
res = client.post(
9+
"/users/", json={"email": "hello123@gmail.com", "password": "password123"})
10+
11+
new_user = schemas.UserOut(**res.json())
12+
assert new_user.email == "hello123@gmail.com"
13+
assert res.status_code == 201

0 commit comments

Comments
 (0)