def test_binary_file_serializing(binary_gif_file, binary_img_file):
with open(f"tests/files/{binary_img_file['file_name']}", "rb") as f:
mfile = File.from_file(f)
assert File(**binary_img_file) == mfile
assert (mfile.media_type == binary_img_file["data"].split(
"data:", 1)[1].split(";", 1)[0])
with open(f"tests/files/{binary_gif_file['file_name']}", "rb") as f:
mfile = File.from_file(f)
assert File(**binary_gif_file) == mfile
assert (mfile.media_type == binary_gif_file["data"].split(
"data:", 1)[1].split(";", 1)[0])
def test_text_file_serializing(text_txt_file, text_json_file):
with open(f"tests/files/{text_txt_file['file_name']}", "r") as f:
mfile = File.from_file(f)
assert File(**text_txt_file) == mfile
assert (mfile.media_type == text_txt_file["data"].split("data:",
1)[1].split(
";", 1)[0])
with open(f"tests/files/{text_json_file['file_name']}", "r") as f:
mfile = File.from_file(f)
assert File(**text_json_file) == mfile
assert (mfile.media_type == text_json_file["data"].split(
"data:", 1)[1].split(";", 1)[0])
def test_creating_file_from_io_with_name(
io_cls,
file_data,
file_name,
explicit_file_name,
):
created_file = io_cls(file_data)
if not explicit_file_name:
created_file.name = file_name
assert File.from_file(created_file, filename=explicit_file_name) == File(
file_name=explicit_file_name or file_name,
data="data:text/plain;base64,dGVzdA==",
)
def test_file_creation_with_right_extension(extension):
File(file_name=f"tmp{extension}", data="")
def test_creating_file_from_string(file_data):
assert File.from_string(file_data, filename="test.txt") == File(
file_name="test.txt",
data="data:text/plain;base64,dGVzdA==",
)
def test_error_with_wrong_extension(extension):
with pytest.raises(ValidationError):
File(file_name="tmp{0}".format(extension), data="")
