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
