def test_unknown_type_submission(self, bot):
submission = SubmissionBuilder(file_ext="zzz", file_size=47453).build_full_submission()
chat_id = -9327622
message_id = 2873292
try:
submission.send_message(bot, chat_id, message_id)
assert False, "Should have thrown exception."
except CantSendFileType as e:
assert str(e) == "I'm sorry, I don't understand that file extension (zzz)."
def test_swf_submission(self, bot):
submission = SubmissionBuilder(file_ext="swf", file_size=47453).build_full_submission()
chat_id = -9327622
message_id = 2873292
try:
submission.send_message(bot, chat_id, message_id)
assert False, "Should have thrown exception."
except CantSendFileType as e:
assert str(e) == "I'm sorry, I can't neaten \".swf\" files."
def test_send_message__without_prefix(self, bot):
submission = SubmissionBuilder(file_ext="jpg", file_size=FASubmission.SIZE_LIMIT_IMAGE - 1)\
.build_full_submission()
chat_id = -9327622
message_id = 2873292
submission.send_message(bot, chat_id, message_id)
bot.send_photo.assert_called_once()
assert bot.send_photo.call_args[1]['chat_id'] == chat_id
assert bot.send_photo.call_args[1]['photo'] == submission.download_url
assert bot.send_photo.call_args[1]['caption'] == submission.link
assert bot.send_photo.call_args[1]['reply_to_message_id'] == message_id
def test_pdf_submission(self, bot):
submission = SubmissionBuilder(file_ext="gif", file_size=47453).build_full_submission()
chat_id = -9327622
message_id = 2873292
submission.send_message(bot, chat_id, message_id)
bot.send_photo.assert_not_called()
bot.send_document.assert_called_once()
assert bot.send_document.call_args[1]['chat_id'] == chat_id
assert bot.send_document.call_args[1]['document'] == submission.download_url
assert bot.send_document.call_args[1]['caption'] == submission.link
assert bot.send_document.call_args[1]['reply_to_message_id'] == message_id
def test_auto_doc_just_under_size_limit(self, bot):
submission = SubmissionBuilder(file_ext="gif", file_size=FASubmission.SIZE_LIMIT_DOCUMENT - 1)\
.build_full_submission()
chat_id = -9327622
message_id = 2873292
submission.send_message(bot, chat_id, message_id)
bot.send_document.assert_called_once()
bot.send_photo.assert_not_called()
bot.send_message.assert_not_called()
assert bot.send_document.call_args[1]['chat_id'] == chat_id
assert bot.send_document.call_args[1]['document'] == submission.download_url
assert bot.send_document.call_args[1]['caption'] == submission.link
assert bot.send_document.call_args[1]['reply_to_message_id'] == message_id
def test_image_over_document_size_limit(self, bot):
submission = SubmissionBuilder(file_ext="jpg", file_size=FASubmission.SIZE_LIMIT_DOCUMENT + 1)\
.build_full_submission()
chat_id = -9327622
message_id = 2873292
submission.send_message(bot, chat_id, message_id)
bot.send_photo.assert_called_once()
assert bot.send_photo.call_args[1]['chat_id'] == chat_id
assert bot.send_photo.call_args[1]['photo'] == submission.thumbnail_url
assert bot.send_photo.call_args[1]['caption'] == \
f"{submission.link}\n[Direct download]({submission.download_url})"
assert bot.send_photo.call_args[1]['reply_to_message_id'] == message_id
assert bot.send_photo.call_args[1]['parse_mode'] == telegram.ParseMode.MARKDOWN
def test_txt_submission(self, bot):
submission = SubmissionBuilder(file_ext="txt").build_full_submission()
chat_id = -9327622
message_id = 2873292
submission.send_message(bot, chat_id, message_id)
bot.send_message.assert_not_called()
bot.send_photo.assert_called_once()
bot.send_document.assert_not_called()
bot.send_audio.assert_not_called()
assert bot.send_photo.call_args[1]['chat_id'] == chat_id
assert bot.send_photo.call_args[1]['photo'] == submission.full_image_url
assert bot.send_photo.call_args[1]['caption'] == \
f"{submission.link}\n[Direct download]({submission.download_url})"
assert bot.send_photo.call_args[1]['reply_to_message_id'] == message_id
assert bot.send_photo.call_args[1]['parse_mode'] == telegram.ParseMode.MARKDOWN
