Merge pull request #32859 from sagarvora/try-fin

fix: always restore old user

frappe/tests/classes/context_managers.py

@@ -46,12 +46,14 @@ def freeze_time(time_to_freeze: Any, is_utc: bool = False, *args: Any, **kwargs:
 
 @UnitTestCase.registerAs(staticmethod)
 @contextmanager
-def set_user(user: str) -> None:
+def set_user(user: str):
 	"""Temporarily: set the user."""
-	old_user = frappe.session.user
-	frappe.set_user(user)
-	yield
-	frappe.set_user(old_user)
+	try:
+		old_user = frappe.session.user
+		frappe.set_user(user)
+		yield
+	finally:
+		frappe.set_user(old_user)
 
 
 @UnitTestCase.registerAs(staticmethod)