Skip to content

Commit 3b52721

Browse files
committed
clean db after test
1 parent 79c6191 commit 3b52721

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

backend/app/tests/conftest.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -7,13 +7,7 @@
77
from app.core.config import settings
88
from app.core.db import engine, init_db
99
from app.main import app
10-
from app.models import (
11-
APIKey,
12-
Organization,
13-
Project,
14-
ProjectUser,
15-
User,
16-
)
10+
from app.models import APIKey, Organization, Project, ProjectUser, User, OpenAI_Thread
1711
from app.tests.utils.user import authentication_token_from_email
1812
from app.tests.utils.utils import get_superuser_token_headers
1913

@@ -29,6 +23,7 @@ def db() -> Generator[Session, None, None]:
2923
session.execute(delete(Organization))
3024
session.execute(delete(APIKey))
3125
session.execute(delete(User))
26+
session.execute(delete(OpenAI_Thread))
3227
session.commit()
3328

3429

0 commit comments

Comments
 (0)