From 7198760d9dd1d662726b469cb5b8e57c0f77b617 Mon Sep 17 00:00:00 2001 From: Or Bin Date: Thu, 11 Apr 2019 14:20:26 +0300 Subject: [PATCH] #10 Added tests for _PresetDocumentResponse --- tests/responses/test_preset_responses.py | 36 ++++++++++++++++++++---- 1 file changed, 30 insertions(+), 6 deletions(-) diff --git a/tests/responses/test_preset_responses.py b/tests/responses/test_preset_responses.py index 315e927..027abec 100644 --- a/tests/responses/test_preset_responses.py +++ b/tests/responses/test_preset_responses.py @@ -47,13 +47,13 @@ def test_single_preset_message_with_some_payload(self, bot, bot_chat_id, generat self._test_single_preset_message(bot, bot_chat_id, generate_new_update, {'key': 'value'}) def test_multiple_preset_messages_with_none_payload(self, bot, bot_chat_id, generate_new_update): - self._test_single_preset_message(bot, bot_chat_id, generate_new_update, None) + self._test_multiple_preset_messages(bot, bot_chat_id, generate_new_update, None) def test_multiple_preset_messages_with_empty_payload(self, bot, bot_chat_id, generate_new_update): - self._test_single_preset_message(bot, bot_chat_id, generate_new_update, {}) + self._test_multiple_preset_messages(bot, bot_chat_id, generate_new_update, {}) def test_multiple_preset_messages_with_some_payload(self, bot, bot_chat_id, generate_new_update): - self._test_single_preset_message(bot, bot_chat_id, generate_new_update, {'key': 'value'}) + self._test_multiple_preset_messages(bot, bot_chat_id, generate_new_update, {'key': 'value'}) @flaky(max_runs=FLAKY_MAX_RUNS, min_passes=FLAKY_MIN_PASSES, rerun_filter=delay_rerun) @@ -93,10 +93,34 @@ def test_single_preset_reply_with_some_payload(self, bot, bot_chat_id, generate_ self._test_single_preset_reply(bot, bot_chat_id, generate_new_update, {'key': 'value'}) def test_multiple_preset_replies_with_none_payload(self, bot, bot_chat_id, generate_new_update): - self._test_single_preset_reply(bot, bot_chat_id, generate_new_update, None) + self._test_multiple_preset_replies(bot, bot_chat_id, generate_new_update, None) def test_multiple_preset_replies_with_empty_payload(self, bot, bot_chat_id, generate_new_update): - self._test_single_preset_reply(bot, bot_chat_id, generate_new_update, {}) + self._test_multiple_preset_replies(bot, bot_chat_id, generate_new_update, {}) def test_multiple_preset_replies_with_some_payload(self, bot, bot_chat_id, generate_new_update): - self._test_single_preset_reply(bot, bot_chat_id, generate_new_update, {'key': 'value'}) + self._test_multiple_preset_replies(bot, bot_chat_id, generate_new_update, {'key': 'value'}) + + +@flaky(max_runs=FLAKY_MAX_RUNS, min_passes=FLAKY_MIN_PASSES, rerun_filter=delay_rerun) +@pytest.mark.usefixtures('bot', 'bot_chat_id', 'generate_new_update') +class TestPresetDocumentResponse: + + DOCUMENT_URL = 'https://media.giphy.com/media/JIX9t2j0ZTN9S/giphy.gif' + + def _test_preset_document_url(self, bot, bot_chat_id, generate_new_update, payload): + response = _PresetDocumentResponse(self.DOCUMENT_URL) + update = generate_new_update(chat_id=bot_chat_id) + + message = response.respond(bot, update, payload) + assert message + assert message.document + + def test_single_preset_reply_with_none_payload(self, bot, bot_chat_id, generate_new_update): + self._test_preset_document_url(bot, bot_chat_id, generate_new_update, None) + + def test_single_preset_reply_with_empty_payload(self, bot, bot_chat_id, generate_new_update): + self._test_preset_document_url(bot, bot_chat_id, generate_new_update, {}) + + def test_single_preset_reply_with_some_payload(self, bot, bot_chat_id, generate_new_update): + self._test_preset_document_url(bot, bot_chat_id, generate_new_update, {'key': 'value'})