Merge pull request #3 from ulamlabs/fixtest

fix test
ulamlabs · Oct 14, 2020 · a3944ad · a3944ad
2 parents 7609ad9 + d63b596
commit a3944ad
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -63,3 +63,6 @@ target/
 
 # IDE
 .idea/
+
+# MacOS
+.DS_Store
diff --git a/celery_mock/celery_mock.py b/celery_mock/celery_mock.py
@@ -75,7 +75,8 @@ def __enter__(self):
 
     def __exit__(self, *exc_info):
         self.stop()
-        _mocked_fn.stop()
+        if not _mocked_tasks:
+            _mocked_fn.stop()
 
 
 def _apply_async(selftask, a=None, kw=None, *args, **kwargs):

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "celery-mock"
-version = "1.1.5"
+version = "1.1.6"
 description = "celery-mock allows you to mock celery task and run them when you want"
 authors = ["Konrad Rotkiewicz <[email protected]>"]
 license = "MIT"