def test_format_filename(self, mock_stack, includes, file_args):
"""Filenames should be formatted in short and long styles"""
with tempfile.NamedTemporaryFile(suffix=".json") as tfile:
includes["variables"]["tmpname"] = tfile.name
request_args = {"files": {"file1": tfile.name}}
file_spec = _get_file_arguments(request_args, mock_stack, includes)
file = file_spec["files"]["file1"]
assert file[0] == os.path.basename(tfile.name)
def test_a_file(self, mock_stack, includes):
"""Json file should have the correct mimetype etc."""
with tempfile.NamedTemporaryFile(suffix=".json") as tfile:
request_args = {"files": {"file1": tfile.name}}
file_spec = _get_file_arguments(request_args, mock_stack, includes)
file = file_spec["files"]["file1"]
assert file[0] == os.path.basename(tfile.name)
assert file[2] == "application/json"
def test_use_long_form_content_type(self, mock_stack, includes):
"""Use custom content type"""
with tempfile.NamedTemporaryFile(suffix=".json") as tfile:
request_args = {
"files": {
"file1": {
"file_path": tfile.name,
"content_type": "abc123",
"content_encoding": "def456",
}
}
}
file_spec = _get_file_arguments(request_args, mock_stack, includes)
file = file_spec["files"]["file1"]
assert file[0] == os.path.basename(tfile.name)
assert file[2] == "abc123"
assert file[3] == {"Content-Encoding": "def456"}
def test_get_empty_files_list(self, mock_stack, includes):
"""No specific files specified -> no files"""
request_args = {"files": {}}
assert _get_file_arguments(request_args, mock_stack, includes) == {}
def test_get_no_files(self, mock_stack, includes):
"""No files in request -> no files"""
request_args = {}
assert _get_file_arguments(request_args, mock_stack, includes) == {}
