From 5798871dc2e0ff64fd8cd2e7e010063af39dadfe Mon Sep 17 00:00:00 2001 From: Nate Coraor Date: Thu, 18 Dec 2025 12:26:09 -0500 Subject: [PATCH] Fix mypy errors in test --- test/integration/test_history_archiving.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/test/integration/test_history_archiving.py b/test/integration/test_history_archiving.py index 2bbcb91ccb2..d8c23750561 100644 --- a/test/integration/test_history_archiving.py +++ b/test/integration/test_history_archiving.py @@ -295,6 +295,7 @@ class TestHistoryArchivingUserUpdateTime(PosixFileSourceSetup, IntegrationTestCa self.dataset_populator.wait_for_purge(history_id, dataset_id) user = sa_session.get(model.User, user_db_id) + assert user is not None assert user.update_time > old_update_time, ( f"User's update_time should be updated when archiving own history. " f"Got {user.update_time}, expected > {old_update_time}" @@ -336,6 +337,7 @@ class TestHistoryArchivingUserUpdateTime(PosixFileSourceSetup, IntegrationTestCa self.dataset_populator.wait_for_purge(history_id, dataset_id) owner = sa_session.get(model.User, owner_db_id) + assert owner is not None assert owner.update_time == old_update_time, ( f"Owner's update_time should NOT be updated when admin archives their history. " f"Got {owner.update_time}, expected {old_update_time}"