Skip to content

Commit ab07962

Browse files
committed
combine revisions & test profile url shows up in db
1 parent 2889cae commit ab07962

File tree

3 files changed

+7
-36
lines changed

3 files changed

+7
-36
lines changed

src/alembic/versions/166f3772fce7_auth_officer_init.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ def upgrade() -> None:
2727
sa.Column("computing_id", sa.String(32), primary_key=True),
2828
sa.Column("first_logged_in", sa.DateTime, nullable=False, default=datetime(2024, 6, 16)),
2929
sa.Column("last_logged_in", sa.DateTime, nullable=False, default=datetime(2024, 6, 16)),
30+
sa.Column("profile_picture_url", sa.Text(), nullable=True),
3031
)
3132
op.create_table(
3233
"user_session",

src/alembic/versions/e74292feb010_add_profile_picture_url_to_site_user.py

Lines changed: 0 additions & 31 deletions
This file was deleted.

src/load_test_db.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
import sqlalchemy
99
from sqlalchemy.ext.asyncio import AsyncSession
1010

11-
from auth.crud import create_user_session
11+
from auth.crud import create_user_session, update_site_user
1212
from database import SQLALCHEMY_TEST_DATABASE_URL, Base, DatabaseSessionManager
1313
from officers.constants import OfficerPosition
1414
from officers.crud import create_new_officer_info, create_new_officer_term, update_officer_info, update_officer_term
@@ -57,8 +57,9 @@ async def reset_db(engine):
5757
else:
5858
print(f"new tables: {table_list}")
5959

60-
async def load_test_auth_data():
61-
pass
60+
async def load_test_auth_data(db_session: AsyncSession):
61+
await create_user_session(db_session, "temp_id_314", "abc314")
62+
await update_site_user(db_session, "temp_id_314", "www.my_profile_picture_url.ca/test")
6263

6364
async def load_test_officers_data(db_session: AsyncSession):
6465
print("login the 3 users, putting them in the site users table")
@@ -216,7 +217,7 @@ async def load_sysadmin(db_session: AsyncSession):
216217
# put your computing id here for testing purposes
217218
SYSADMIN_COMPUTING_ID = "gsa92"
218219

219-
await create_user_session(db_session, "temp_id_4", "gsa92")
220+
await create_user_session(db_session, "temp_id_4", SYSADMIN_COMPUTING_ID)
220221
await create_new_officer_info(db_session, OfficerInfo(
221222
legal_name="Gabe Schulz",
222223
discord_id=None,
@@ -250,7 +251,7 @@ async def load_sysadmin(db_session: AsyncSession):
250251
async def async_main(sessionmanager):
251252
await reset_db(sessionmanager._engine)
252253
async with sessionmanager.session() as db_session:
253-
# load_test_auth_data
254+
await load_test_auth_data(db_session)
254255
await load_test_officers_data(db_session)
255256
await load_sysadmin(db_session)
256257

0 commit comments

Comments
 (0)