소스 검색

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):
 
         @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("grp.getgrall")