this fix should be part of another PR

This commit is contained in:
Philipp 2024-06-19 22:10:17 +00:00
parent 67a82658ef
commit a1f6667bdd

View File

@ -3,7 +3,6 @@ from pathlib import Path
from unittest import mock from unittest import mock
import httpx import httpx
from django.conf import settings
from django.test import TestCase from django.test import TestCase
from documents.parsers import ParseError from documents.parsers import ParseError
@ -342,10 +341,7 @@ class TestTikaHtmlParse(HttpxMockMixin, BaseMailParserTestCase):
) )
parsed = self.parser.tika_parse(html) parsed = self.parser.tika_parse(html)
self.assertEqual(expected_text, parsed.strip()) self.assertEqual(expected_text, parsed.strip())
self.assertIn( self.assertIn("http://localhost:9998", str(self.httpx_mock.get_request().url))
f"{settings.TIKA_ENDPOINT}",
str(self.httpx_mock.get_request().url),
)
def test_tika_parse_exception(self): def test_tika_parse_exception(self):
""" """
@ -512,7 +508,7 @@ class TestParser(FileSystemAssertsMixin, HttpxMockMixin, BaseMailParserTestCase)
""" """
self.httpx_mock.add_response( self.httpx_mock.add_response(
url=f"{settings.TIKA_GOTENBERG_ENDPOINT}/forms/chromium/convert/html", url="http://localhost:3000/forms/chromium/convert/html",
method="POST", method="POST",
content=(self.SAMPLE_DIR / "simple_text.eml.pdf").read_bytes(), content=(self.SAMPLE_DIR / "simple_text.eml.pdf").read_bytes(),
) )
@ -533,7 +529,7 @@ class TestParser(FileSystemAssertsMixin, HttpxMockMixin, BaseMailParserTestCase)
- Archive file is generated - Archive file is generated
""" """
self.httpx_mock.add_response( self.httpx_mock.add_response(
url=f"{settings.TIKA_ENDPOINT}/tika/text", url="http://localhost:9998/tika/text",
method="PUT", method="PUT",
json={ json={
"Content-Type": "text/html", "Content-Type": "text/html",
@ -542,12 +538,12 @@ class TestParser(FileSystemAssertsMixin, HttpxMockMixin, BaseMailParserTestCase)
}, },
) )
self.httpx_mock.add_response( self.httpx_mock.add_response(
url=f"{settings.TIKA_GOTENBERG_ENDPOINT}/forms/chromium/convert/html", url="http://localhost:3000/forms/chromium/convert/html",
method="POST", method="POST",
content=(self.SAMPLE_DIR / "html.eml.pdf").read_bytes(), content=(self.SAMPLE_DIR / "html.eml.pdf").read_bytes(),
) )
self.httpx_mock.add_response( self.httpx_mock.add_response(
url=f"{settings.TIKA_GOTENBERG_ENDPOINT}/forms/pdfengines/merge", url="http://localhost:3000/forms/pdfengines/merge",
method="POST", method="POST",
content=b"Pretend merged PDF content", content=b"Pretend merged PDF content",
) )
@ -566,7 +562,7 @@ class TestParser(FileSystemAssertsMixin, HttpxMockMixin, BaseMailParserTestCase)
- ParseError is raised - ParseError is raised
""" """
self.httpx_mock.add_response( self.httpx_mock.add_response(
url=f"{settings.TIKA_ENDPOINT}/tika/text", url="http://localhost:9998/tika/text",
method="PUT", method="PUT",
json={ json={
"Content-Type": "text/html", "Content-Type": "text/html",
@ -575,12 +571,12 @@ class TestParser(FileSystemAssertsMixin, HttpxMockMixin, BaseMailParserTestCase)
}, },
) )
self.httpx_mock.add_response( self.httpx_mock.add_response(
url=f"{settings.TIKA_GOTENBERG_ENDPOINT}/forms/chromium/convert/html", url="http://localhost:3000/forms/chromium/convert/html",
method="POST", method="POST",
content=(self.SAMPLE_DIR / "html.eml.pdf").read_bytes(), content=(self.SAMPLE_DIR / "html.eml.pdf").read_bytes(),
) )
self.httpx_mock.add_response( self.httpx_mock.add_response(
url=f"{settings.TIKA_GOTENBERG_ENDPOINT}/forms/chromium/convert/html", url="http://localhost:3000/forms/chromium/convert/html",
method="POST", method="POST",
status_code=httpx.codes.INTERNAL_SERVER_ERROR, status_code=httpx.codes.INTERNAL_SERVER_ERROR,
) )
@ -598,7 +594,7 @@ class TestParser(FileSystemAssertsMixin, HttpxMockMixin, BaseMailParserTestCase)
- ParseError is raised - ParseError is raised
""" """
self.httpx_mock.add_response( self.httpx_mock.add_response(
url=f"{settings.TIKA_ENDPOINT}/tika/text", url="http://localhost:9998/tika/text",
method="PUT", method="PUT",
json={ json={
"Content-Type": "text/html", "Content-Type": "text/html",
@ -607,12 +603,12 @@ class TestParser(FileSystemAssertsMixin, HttpxMockMixin, BaseMailParserTestCase)
}, },
) )
self.httpx_mock.add_response( self.httpx_mock.add_response(
url=f"{settings.TIKA_GOTENBERG_ENDPOINT}/forms/chromium/convert/html", url="http://localhost:3000/forms/chromium/convert/html",
method="POST", method="POST",
content=(self.SAMPLE_DIR / "html.eml.pdf").read_bytes(), content=(self.SAMPLE_DIR / "html.eml.pdf").read_bytes(),
) )
self.httpx_mock.add_response( self.httpx_mock.add_response(
url=f"{settings.TIKA_GOTENBERG_ENDPOINT}/forms/pdfengines/merge", url="http://localhost:3000/forms/pdfengines/merge",
method="POST", method="POST",
status_code=httpx.codes.INTERNAL_SERVER_ERROR, status_code=httpx.codes.INTERNAL_SERVER_ERROR,
) )
@ -657,5 +653,5 @@ class TestParser(FileSystemAssertsMixin, HttpxMockMixin, BaseMailParserTestCase)
self.assertEqual( self.assertEqual(
str(request.url), str(request.url),
f"{settings.TIKA_GOTENBERG_ENDPOINT}/forms/chromium/convert/html", "http://localhost:3000/forms/chromium/convert/html",
) )