Add test for get_pdfjs_js_path()

1 files changed, 26 insertions, 0 deletions

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),