diff options
Diffstat (limited to 'tests/docx')
| -rw-r--r-- | tests/docx/files_tests.py | 21 |
1 files changed, 15 insertions, 6 deletions
diff --git a/tests/docx/files_tests.py b/tests/docx/files_tests.py index 6845cc2..b405373 100644 --- a/tests/docx/files_tests.py +++ b/tests/docx/files_tests.py @@ -1,10 +1,19 @@ -from mammoth.docx.files import Files, InvalidFileReferenceError +from mammoth.docx.files import ExternalFileAccessIsDisabledError, Files, InvalidFileReferenceError from ..testing import generate_test_path, assert_equal, assert_raises +def test_when_external_file_access_is_disabled_then_opening_file_raises_error(): + files = Files(None, external_file_access=False) + error = assert_raises(ExternalFileAccessIsDisabledError, lambda: files.open("/tmp/image.png")) + expected_message = ( + "could not open external image '/tmp/image.png', external file access is disabled" + ) + assert_equal(expected_message, str(error)) + + def test_can_open_files_with_file_uri(): path = generate_test_path("tiny-picture.png") - files = Files(None) + files = Files(None, external_file_access=True) with files.open("file:///" + path) as image_file: contents = image_file.read() assert_equal(bytes, type(contents)) @@ -13,7 +22,7 @@ def test_can_open_files_with_file_uri(): def test_can_open_files_with_relative_uri(): - files = Files(generate_test_path("")) + files = Files(generate_test_path(""), external_file_access=True) with files.open("tiny-picture.png") as image_file: contents = image_file.read() assert_equal(bytes, type(contents)) @@ -22,7 +31,7 @@ def test_can_open_files_with_relative_uri(): def test_given_base_is_not_set_when_opening_relative_uri_then_error_is_raised(): - files = Files(None) + files = Files(None, external_file_access=True) error = assert_raises(InvalidFileReferenceError, lambda: files.open("not-a-real-file.png")) expected_message = ( "could not find external image 'not-a-real-file.png', fileobj has no name" @@ -31,7 +40,7 @@ def test_given_base_is_not_set_when_opening_relative_uri_then_error_is_raised(): def test_error_is_raised_if_relative_uri_cannot_be_opened(): - files = Files("/tmp") + files = Files("/tmp", external_file_access=True) error = assert_raises(InvalidFileReferenceError, lambda: files.open("not-a-real-file.png")) expected_message = ( "could not open external image: 'not-a-real-file.png' (document directory: '/tmp')\n" + @@ -41,7 +50,7 @@ def test_error_is_raised_if_relative_uri_cannot_be_opened(): def test_error_is_raised_if_file_uri_cannot_be_opened(): - files = Files("/tmp") + files = Files("/tmp", external_file_access=True) error = assert_raises(InvalidFileReferenceError, lambda: files.open("file:///not-a-real-file.png")) expected_message = "could not open external image: 'file:///not-a-real-file.png' (document directory: '/tmp')\n" assert str(error).startswith(expected_message) |