Verify authsource during MongoDB authentication test

Relates to celery@4923175
enkidulan · May 26, 2019 · 7eb6e30 · 7eb6e30
1 parent 8d77fc9
commit 7eb6e30
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/celery/backends/mongodb.py b/celery/backends/mongodb.py
@@ -271,8 +271,9 @@ def _get_database(self):
         conn = self._get_connection()
         db = conn[self.database_name]
         if self.user and self.password:
-            source = self.options.get('authsource', 
-                        self.database_name or 'admin'
+            source = self.options.get(
+                'authsource',
+                self.database_name or 'admin'
             )
             if not db.authenticate(self.user, self.password, source=source):
                 raise ImproperlyConfigured(

diff --git a/t/unit/backends/test_mongodb.py b/t/unit/backends/test_mongodb.py
@@ -454,7 +454,8 @@ def test_get_database_authfailure(self):
         x.password = 'cere4l'
         with pytest.raises(ImproperlyConfigured):
             x._get_database()
-        db.authenticate.assert_called_with('jerry', 'cere4l')
+        db.authenticate.assert_called_with('jerry', 'cere4l',
+                                           source=x.database_name)
 
     def test_prepare_client_options(self):
         with patch('pymongo.version_tuple', new=(3, 0, 3)):