From 4eb23df6274912085e60ddc0879ce1b853addb93 Mon Sep 17 00:00:00 2001 From: ruthra kumar Date: Wed, 16 Jul 2025 14:34:12 +0530 Subject: [PATCH] refactor(test): set_user utility method --- erpnext/tests/utils.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/erpnext/tests/utils.py b/erpnext/tests/utils.py index 056a39b5faf..eedf022b828 100644 --- a/erpnext/tests/utils.py +++ b/erpnext/tests/utils.py @@ -2490,6 +2490,15 @@ class ERPNextTestSuite(unittest.TestCase): ) ) + @contextmanager + def set_user(self, user: str): + try: + old_user = frappe.session.user + frappe.set_user(user) + yield + finally: + frappe.set_user(old_user) + @ERPNextTestSuite.registerAs(staticmethod) @contextmanager