소스 검색

Add contextmanager methods to testing Pipeline

Justin Patrin 9 년 전
부모
커밋
0fc2d1048d
1개의 변경된 파일6개의 추가작업 그리고 0개의 파일을 삭제
  1. 6 0
      celery/tests/backends/test_redis.py

+ 6 - 0
celery/tests/backends/test_redis.py

@@ -37,6 +37,12 @@ class Pipeline(object):
             return self
         return add_step
 
+    def __enter__(self):
+        return self
+
+    def __exit__(self, type, value, traceback):
+        pass
+
     def execute(self):
         return [step(*a, **kw) for step, a, kw in self.steps]