1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
|
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, external_file_access=True)
with files.open("file:///" + path) as image_file:
contents = image_file.read()
assert_equal(bytes, type(contents))
with open(path, "rb") as source_file:
assert_equal(source_file.read(), contents)
def test_can_open_files_with_relative_uri():
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))
with open(generate_test_path("tiny-picture.png"), "rb") as source_file:
assert_equal(source_file.read(), contents)
def test_given_base_is_not_set_when_opening_relative_uri_then_error_is_raised():
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"
)
assert_equal(expected_message, str(error))
def test_error_is_raised_if_relative_uri_cannot_be_opened():
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" +
"[Errno 2] No such file or directory: '/tmp/not-a-real-file.png'"
)
assert_equal(expected_message, str(error))
def test_error_is_raised_if_file_uri_cannot_be_opened():
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)
|
