paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-04-25 10:49:30 -05:00
Code Issues Packages Projects Releases Wiki Activity

tests: wait for the consumer to exit before removing directories.

Browse Source
This commit is contained in:
jonaswinkler 2020-11-27 13:14:02 +01:00
parent 35b2033949
commit a4277706f2
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/documents/tests/test_management_consumer.py
View File

@ -38,6 +38,7 @@ class TestConsumer(TestCase):
sample_file = os.path.join(os.path.dirname(__file__), "samples", "simple.pdf")
def setUp(self) -> None:
self.t = None
patcher = mock.patch("documents.management.commands.document_consumer.async_task")
self.task_mock = patcher.start()
self.addCleanup(patcher.stop)
@ -53,7 +54,12 @@ class TestConsumer(TestCase):
def tearDown(self) -> None:
if self.t:
# set the stop flag
self.t.stop()
# wait for the consumer to exit.
self.t.join()
remove_dirs(self.dirs)
def wait_for_task_mock_call(self):
n = 0