#!/usr/bin/env python # # A library that provides a Python interface to the Telegram Bot API # Copyright (C) 2015-2022 # Leandro Toledo de Souza # # This program is free software: you can redistribute it and/or modify # it under the terms of the GNU Lesser Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Lesser Public License for more details. # # You should have received a copy of the GNU Lesser Public License # along with this program. If not, see [http://www.gnu.org/licenses/]. from pathlib import Path import pytest from telegram.error import TelegramError from telegram.request import Request from tests.conftest import data_file def test_slot_behaviour(mro_slots): inst = Request() for attr in inst.__slots__: assert getattr(inst, attr, 'err') != 'err', f"got extra slot '{attr}'" assert len(mro_slots(inst)) == len(set(mro_slots(inst))), "duplicate slot" def test_replaced_unprintable_char(): """ Clients can send arbitrary bytes in callback data. Make sure the correct error is raised in this case. """ server_response = b'{"invalid utf-8": "\x80", "result": "KUKU"}' assert Request._parse(server_response) == 'KUKU' def test_parse_illegal_json(): """ Clients can send arbitrary bytes in callback data. Make sure the correct error is raised in this case. """ server_response = b'{"invalid utf-8": "\x80", result: "KUKU"}' with pytest.raises(TelegramError, match='Invalid server response'): Request._parse(server_response) @pytest.mark.parametrize( "destination_path_type", [str, Path], ids=['str destination_path', 'pathlib.Path destination_path'], ) def test_download(destination_path_type): destination_filepath = data_file('downloaded_request.txt') request = Request() request.download("http://google.com", destination_path_type(destination_filepath)) assert destination_filepath.is_file() destination_filepath.unlink()