浏览代码

Fixes test for os.initroups on Py2.6

Ask Solem 13 年之前
父节点
当前提交
2a72f8adad
共有 1 个文件被更改,包括 5 次插入8 次删除
  1. 5 8
      celery/tests/utilities/test_platforms.py

+ 5 - 8
celery/tests/utilities/test_platforms.py

@@ -256,14 +256,11 @@ if not current_app.IS_WINDOWS:
     class test_initgroups(Case):
     class test_initgroups(Case):
 
 
         @patch("pwd.getpwuid")
         @patch("pwd.getpwuid")
-        def test_with_initgroups(self, getpwuid):
-            prev, os.initgroups = os.initgroups, Mock()
-            try:
-                getpwuid.return_value = ["user"]
-                initgroups(5001, 50001)
-                os.initgroups.assert_called_with("user", 50001)
-            finally:
-                os.initgroups = prev
+        @patch("os.initgroups", create=True)
+        def test_with_initgroups(self, initgroups_, getpwuid):
+            getpwuid.return_value = ["user"]
+            initgroups(5001, 50001)
+            initgroups_.assert_called_with("user", 50001)
 
 
         @patch("celery.platforms.setgroups")
         @patch("celery.platforms.setgroups")
         @patch("grp.getgrall")
         @patch("grp.getgrall")