From 7d5acb697086c903d485679f3dc754777c8e6ef5 Mon Sep 17 00:00:00 2001 From: toofar Date: Sat, 2 Dec 2023 12:04:06 +1300 Subject: Add test for get_pdfjs_js_path() --- tests/unit/browser/test_pdfjs.py | 26 ++++++++++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/tests/unit/browser/test_pdfjs.py b/tests/unit/browser/test_pdfjs.py index fe2fea9a0..7bd9588b4 100644 --- a/tests/unit/browser/test_pdfjs.py +++ b/tests/unit/browser/test_pdfjs.py @@ -193,6 +193,32 @@ def test_is_available(available, mocker): assert pdfjs.is_available() == available +@pytest.mark.parametrize('found_file', [ + "build/pdf.js", + "build/pdf.mjs", + None, +]) +def test_get_pdfjs_js_path(found_file, mocker): + + def side_effect(requested): + if found_file and requested.endswith(found_file): + return + + raise pdfjs.PDFJSNotFound(requested) + + mock = mocker.patch.object(pdfjs, 'get_pdfjs_res', autospec=True) + mock.side_effect = side_effect + + if found_file is None: + with pytest.raises( + pdfjs.PDFJSNotFound, + match="Path 'build/pdf.js or build/pdf.mjs' not found" + ): + pdfjs.get_pdfjs_js_path() + else: + assert pdfjs.get_pdfjs_js_path() == found_file + + @pytest.mark.parametrize('mimetype, url, enabled, expected', [ # PDF files ('application/pdf', 'http://www.example.com', True, True), -- cgit v1.2.3-54-g00ecf