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

use tmp dir

Browse Source
This commit is contained in:
Ovv 2018-02-25 19:42:03 +01:00 committed by Daniel Quinn
parent 8fefafb844
commit f8c6c07bb7
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/documents/tests/test_consumer.py
View File

@ -1,5 +1,6 @@
from django.test import TestCase from django.test import TestCase
from unittest import mock from unittest import mock
from tempfile import TemporaryDirectory
from ..consumer import Consumer from ..consumer import Consumer
from ..models import FileInfo from ..models import FileInfo
@ -31,16 +32,21 @@ class TestConsumer(TestCase):
(None, lambda _: {"weight": 0, "parser": DummyParser1}), (None, lambda _: {"weight": 0, "parser": DummyParser1}),
(None, lambda _: {"weight": 1, "parser": DummyParser2}), (None, lambda _: {"weight": 1, "parser": DummyParser2}),
) )
with TemporaryDirectory() as tmpdir:
self.assertEqual(Consumer(consume=".")._get_parser_class("doc.pdf"), self.assertEqual(
DummyParser2) Consumer(consume=tmpdir)._get_parser_class("doc.pdf"),
DummyParser2
)
@mock.patch("documents.consumer.os.makedirs") @mock.patch("documents.consumer.os.makedirs")
@mock.patch("documents.consumer.os.path.exists", return_value=True) @mock.patch("documents.consumer.os.path.exists", return_value=True)
@mock.patch("documents.consumer.document_consumer_declaration.send") @mock.patch("documents.consumer.document_consumer_declaration.send")
def test__get_parser_class_0_parsers(self, m, *args): def test__get_parser_class_0_parsers(self, m, *args):
m.return_value = ((None, lambda _: None),) m.return_value = ((None, lambda _: None),)
self.assertIsNone(Consumer(consume=".")._get_parser_class("doc.pdf")) with TemporaryDirectory() as tmpdir:
self.assertIsNone(
Consumer(consume=tmpdir)._get_parser_class("doc.pdf")
)
@mock.patch("documents.consumer.os.makedirs") @mock.patch("documents.consumer.os.makedirs")
@mock.patch("documents.consumer.os.path.exists", return_value=True) @mock.patch("documents.consumer.os.path.exists", return_value=True)
@ -49,7 +55,8 @@ class TestConsumer(TestCase):
m.return_value = ( m.return_value = (
(None, lambda _: {"weight": 0, "parser": self.DummyParser}), (None, lambda _: {"weight": 0, "parser": self.DummyParser}),
) )
return Consumer(consume=".") with TemporaryDirectory() as tmpdir:
return Consumer(consume=tmpdir)
class TestAttributes(TestCase): class TestAttributes(TestCase):