mirror of
				https://github.com/paperless-ngx/paperless-ngx.git
				synced 2025-10-30 03:56:23 -05:00 
			
		
		
		
	use tmp dir
This commit is contained in:
		| @@ -1,5 +1,6 @@ | ||||
| from django.test import TestCase | ||||
| from unittest import mock | ||||
| from tempfile import TemporaryDirectory | ||||
|  | ||||
| from ..consumer import Consumer | ||||
| from ..models import FileInfo | ||||
| @@ -31,16 +32,21 @@ class TestConsumer(TestCase): | ||||
|             (None, lambda _: {"weight": 0, "parser": DummyParser1}), | ||||
|             (None, lambda _: {"weight": 1, "parser": DummyParser2}), | ||||
|         ) | ||||
|  | ||||
|         self.assertEqual(Consumer(consume=".")._get_parser_class("doc.pdf"), | ||||
|                          DummyParser2) | ||||
|         with TemporaryDirectory() as tmpdir: | ||||
|             self.assertEqual( | ||||
|                 Consumer(consume=tmpdir)._get_parser_class("doc.pdf"), | ||||
|                 DummyParser2 | ||||
|             ) | ||||
|  | ||||
|     @mock.patch("documents.consumer.os.makedirs") | ||||
|     @mock.patch("documents.consumer.os.path.exists", return_value=True) | ||||
|     @mock.patch("documents.consumer.document_consumer_declaration.send") | ||||
|     def test__get_parser_class_0_parsers(self, m, *args): | ||||
|         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.path.exists", return_value=True) | ||||
| @@ -49,7 +55,8 @@ class TestConsumer(TestCase): | ||||
|         m.return_value = ( | ||||
|             (None, lambda _: {"weight": 0, "parser": self.DummyParser}), | ||||
|         ) | ||||
|         return Consumer(consume=".") | ||||
|         with TemporaryDirectory() as tmpdir: | ||||
|             return Consumer(consume=tmpdir) | ||||
|  | ||||
|  | ||||
| class TestAttributes(TestCase): | ||||
|   | ||||
		Reference in New Issue
	
	Block a user
	 Ovv
					Ovv