def test_get_all_attachments(len_of_attachments): builder = MessageBuilder() for _ in range(len_of_attachments): builder.document() assert len(builder.message.attachments.all_attachments) == len_of_attachments
def test_document_in_attachments(): builder = MessageBuilder() builder.image() builder.document() assert builder.message.attachments.document
def test_file_with_unsupported_extension(): builder = MessageBuilder() builder.document(file_name="test.py") assert builder.message.attachments.file