Fix test broken in force push

openzim · Jun 20, 2024 · 2763688 · 2763688
1 parent fbdeed6
commit 2763688
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/wp1/api_test.py b/wp1/api_test.py
@@ -6,10 +6,15 @@
 
 import wp1.api
 
+TEST_CREDS = {
+    'user': 'user',
+    'pass': 'password',
+}
+
 
 class ApiWithCredsTest(unittest.TestCase):
 
-  @patch('wp1.api.get_credentials')
+  @patch('wp1.api.get_credentials', return_value=TEST_CREDS)
   @patch('wp1.api.mwclient.Site')
   @patch('wp1.api.site')
   @patch('http.cookiejar.MozillaCookieJar')
@@ -21,7 +26,7 @@ def test_login(self, patched_cookies, patched_site, patched_mwsite,
     wp1.api.login()
     self.assertEqual(1, site.login.call_count)
 
-  @patch('wp1.api.get_credentials')
+  @patch('wp1.api.get_credentials', return_value=TEST_CREDS)
   @patch('wp1.api.mwclient.Site')
   @patch('http.cookiejar.MozillaCookieJar')
   def test_login_already_logged_in(self, patched_cookies, patched_mwsite,
@@ -31,7 +36,7 @@ def test_login_already_logged_in(self, patched_cookies, patched_mwsite,
     self.assertTrue(actual)
     self.assertEqual(0, site.login.call_count)
 
-  @patch('wp1.api.get_credentials')
+  @patch('wp1.api.get_credentials', return_value=TEST_CREDS)
   @patch('wp1.api.mwclient.Site')
   @patch('wp1.api.site')
   @patch('wp1.api.logger')