def test_open_local_or_remote_file__directory(tmpdir):
"""
Test the `open_local_or_remote_file` raises a ValueError for a given `Link`
to a directory.
"""
link = Link(path_to_url(tmpdir.strpath))
session = Session()
with pytest.raises(ValueError, match="Cannot open directory for read"):
with open_local_or_remote_file(link, session):
pass # pragma: no cover
def test_open_local_or_remote_file__directory(tmpdir):
"""
Test the `open_local_or_remote_file` raises a ValueError for a given `Link`
to a directory.
"""
link = Link(path_to_url(tmpdir.strpath))
session = Session()
with pytest.raises(ValueError, match="Cannot open directory for read"):
with open_local_or_remote_file(link, session):
pass # pragma: no cover
def test_open_local_or_remote_file__local_file(tmp_path, content, content_length):
"""
Test the `open_local_or_remote_file` returns a context manager to a FileStream
for a given `Link` to a local file.
"""
local_file_path = tmp_path / "foo.txt"
local_file_path.write_bytes(content)
link = Link(local_file_path.as_uri())
session = Session()
with open_local_or_remote_file(link, session) as file_stream:
assert file_stream.stream.read() == content
assert file_stream.size == content_length
def test_open_local_or_remote_file__local_file(tmp_path, content, content_length):
"""
Test the `open_local_or_remote_file` returns a context manager to a FileStream
for a given `Link` to a local file.
"""
local_file_path = tmp_path / "foo.txt"
local_file_path.write_bytes(content)
link = Link(local_file_path.as_uri())
session = Session()
with open_local_or_remote_file(link, session) as file_stream:
assert file_stream.stream.read() == content
assert file_stream.size == content_length
def test_open_local_or_remote_file__remote_file(tmp_path, content,
content_length,
expected_content_length):
"""
Test the `open_local_or_remote_file` returns a context manager to a FileStream
for a given `Link` to a remote file.
"""
link = Link("https://example.com/foo.txt")
session = Session()
response_file_path = tmp_path / "foo.txt"
response_file_path.write_bytes(content)
mock_response = mock.Mock()
mock_response.raw = response_file_path.open("rb")
mock_response.headers = {"content-length": content_length}
with mock.patch.object(session, "get", return_value=mock_response):
with open_local_or_remote_file(link, session) as file_stream:
assert file_stream.stream.read() == content
assert file_stream.size == expected_content_length
mock_response.close.assert_called_once()
def test_open_local_or_remote_file__remote_file(
tmp_path, content, content_length, expected_content_length
):
"""
Test the `open_local_or_remote_file` returns a context manager to a FileStream
for a given `Link` to a remote file.
"""
link = Link("https://example.com/foo.txt")
session = Session()
response_file_path = tmp_path / "foo.txt"
response_file_path.write_bytes(content)
mock_response = mock.Mock()
mock_response.raw = response_file_path.open("rb")
mock_response.headers = {"content-length": content_length}
with mock.patch.object(session, "get", return_value=mock_response):
with open_local_or_remote_file(link, session) as file_stream:
assert file_stream.stream.read() == content
assert file_stream.size == expected_content_length
mock_response.close.assert_called_once()
