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])
async def my_handler(message: Message) -> None: with open("my_file.txt") as f: notification = SendingMessage( file=File.from_file(f), credentials=message.credentials, ) await bot.send(notification)
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_message_file_from_file( self, sending_message: SendingMessage) -> None: original_file = sending_message.file sending_message.add_file(File.from_file(original_file.file)) assert sending_message.file == original_file