paradizelost/paperless-ngx
1
0
Fork 0
mirror of https://github.com/paperless-ngx/paperless-ngx.git synced 2025-11-03 03:16:10 -06: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 b10c2c770c
commit 2262a6ec28
1 changed files with 12 additions and 5 deletions
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):