[QOLDEV-1011] mock out global object during auth test

qld-gov-au · Nov 15, 2024 · 1669f5a · 1669f5a
1 parent 57fb298
commit 1669f5a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/ckanext/datarequests/tests/test_auth.py b/ckanext/datarequests/tests/test_auth.py
@@ -58,9 +58,12 @@ class AuthTest(unittest.TestCase):
     def setUp(self):
         self._get_action = auth.get_action
         auth.get_action = MagicMock()
+        self._g = controller_functions.g
+        controller_functions.g = MagicMock()
 
     def tearDown(self):
         auth.get_action = self._get_action
+        controller_functions.g = self._g
 
     @parameterized.expand([
         # Data Requests

diff --git a/ckanext/datarequests/tests/test_helpers.py b/ckanext/datarequests/tests/test_helpers.py
@@ -32,14 +32,11 @@ def setUp(self):
         self.db_patch = patch('ckanext.datarequests.helpers.db')
         self.db_patch.start()
 
-        self._c = helpers.c
-        helpers.c = MagicMock()
-        helpers.c.userobj.id = '12345'
+        helpers.tk.g.userobj.id = '12345'
 
     def tearDown(self):
         self.tk_patch.stop()
         self.db_patch.stop()
-        helpers.c = self._c
 
     def test_get_comments_number(self):
         # Mocking