1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185
|
#!/usr/bin/env python
#
# A library that provides a Python interface to the Telegram Bot API
# Copyright (C) 2015-2025
# Leandro Toledo de Souza <devs@python-telegram-bot.org>
#
# 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/].
import datetime as dtm
import time
import pytest
from telegram import LoginUrl, WebhookInfo
from telegram._utils.datetime import UTC, from_timestamp
from tests.auxil.slots import mro_slots
@pytest.fixture(scope="module")
def webhook_info():
return WebhookInfo(
url=WebhookInfoTestBase.url,
has_custom_certificate=WebhookInfoTestBase.has_custom_certificate,
pending_update_count=WebhookInfoTestBase.pending_update_count,
ip_address=WebhookInfoTestBase.ip_address,
last_error_date=WebhookInfoTestBase.last_error_date,
max_connections=WebhookInfoTestBase.max_connections,
allowed_updates=WebhookInfoTestBase.allowed_updates,
last_synchronization_error_date=WebhookInfoTestBase.last_synchronization_error_date,
)
class WebhookInfoTestBase:
url = "http://www.google.com"
has_custom_certificate = False
pending_update_count = 5
ip_address = "127.0.0.1"
last_error_date = time.time()
max_connections = 42
allowed_updates = ["type1", "type2"]
last_synchronization_error_date = time.time()
class TestWebhookInfoWithoutRequest(WebhookInfoTestBase):
def test_slot_behaviour(self, webhook_info):
for attr in webhook_info.__slots__:
assert getattr(webhook_info, attr, "err") != "err", f"got extra slot '{attr}'"
assert len(mro_slots(webhook_info)) == len(set(mro_slots(webhook_info))), "duplicate slot"
def test_to_dict(self, webhook_info):
webhook_info_dict = webhook_info.to_dict()
assert isinstance(webhook_info_dict, dict)
assert webhook_info_dict["url"] == self.url
assert webhook_info_dict["pending_update_count"] == self.pending_update_count
assert webhook_info_dict["last_error_date"] == self.last_error_date
assert webhook_info_dict["max_connections"] == self.max_connections
assert webhook_info_dict["allowed_updates"] == self.allowed_updates
assert webhook_info_dict["ip_address"] == self.ip_address
assert (
webhook_info_dict["last_synchronization_error_date"]
== self.last_synchronization_error_date
)
def test_de_json(self, offline_bot):
json_dict = {
"url": self.url,
"has_custom_certificate": self.has_custom_certificate,
"pending_update_count": self.pending_update_count,
"last_error_date": self.last_error_date,
"max_connections": self.max_connections,
"allowed_updates": self.allowed_updates,
"ip_address": self.ip_address,
"last_synchronization_error_date": self.last_synchronization_error_date,
}
webhook_info = WebhookInfo.de_json(json_dict, offline_bot)
assert webhook_info.api_kwargs == {}
assert webhook_info.url == self.url
assert webhook_info.has_custom_certificate == self.has_custom_certificate
assert webhook_info.pending_update_count == self.pending_update_count
assert isinstance(webhook_info.last_error_date, dtm.datetime)
assert webhook_info.last_error_date == from_timestamp(self.last_error_date)
assert webhook_info.max_connections == self.max_connections
assert webhook_info.allowed_updates == tuple(self.allowed_updates)
assert webhook_info.ip_address == self.ip_address
assert isinstance(webhook_info.last_synchronization_error_date, dtm.datetime)
assert webhook_info.last_synchronization_error_date == from_timestamp(
self.last_synchronization_error_date
)
def test_de_json_localization(self, offline_bot, raw_bot, tz_bot):
json_dict = {
"url": self.url,
"has_custom_certificate": self.has_custom_certificate,
"pending_update_count": self.pending_update_count,
"last_error_date": self.last_error_date,
"max_connections": self.max_connections,
"allowed_updates": self.allowed_updates,
"ip_address": self.ip_address,
"last_synchronization_error_date": self.last_synchronization_error_date,
}
webhook_info_bot = WebhookInfo.de_json(json_dict, offline_bot)
webhook_info_raw = WebhookInfo.de_json(json_dict, raw_bot)
webhook_info_tz = WebhookInfo.de_json(json_dict, tz_bot)
# comparing utcoffsets because comparing timezones is unpredicatable
last_error_date_offset = webhook_info_tz.last_error_date.utcoffset()
last_error_tz_bot_offset = tz_bot.defaults.tzinfo.utcoffset(
webhook_info_tz.last_error_date.replace(tzinfo=None)
)
sync_error_date_offset = webhook_info_tz.last_synchronization_error_date.utcoffset()
sync_error_date_tz_bot_offset = tz_bot.defaults.tzinfo.utcoffset(
webhook_info_tz.last_synchronization_error_date.replace(tzinfo=None)
)
assert webhook_info_raw.last_error_date.tzinfo == UTC
assert webhook_info_bot.last_error_date.tzinfo == UTC
assert last_error_date_offset == last_error_tz_bot_offset
assert webhook_info_raw.last_synchronization_error_date.tzinfo == UTC
assert webhook_info_bot.last_synchronization_error_date.tzinfo == UTC
assert sync_error_date_offset == sync_error_date_tz_bot_offset
def test_always_tuple_allowed_updates(self):
webhook_info = WebhookInfo(
self.url, self.has_custom_certificate, self.pending_update_count
)
assert webhook_info.allowed_updates == ()
def test_equality(self):
a = WebhookInfo(
url=self.url,
has_custom_certificate=self.has_custom_certificate,
pending_update_count=self.pending_update_count,
last_error_date=self.last_error_date,
max_connections=self.max_connections,
)
b = WebhookInfo(
url=self.url,
has_custom_certificate=self.has_custom_certificate,
pending_update_count=self.pending_update_count,
last_error_date=self.last_error_date,
max_connections=self.max_connections,
)
c = WebhookInfo(
url="http://github.com",
has_custom_certificate=True,
pending_update_count=78,
last_error_date=0,
max_connections=1,
)
d = WebhookInfo(
url="http://github.com",
has_custom_certificate=True,
pending_update_count=78,
last_error_date=0,
max_connections=1,
last_synchronization_error_date=123,
)
e = LoginUrl("text.com")
assert a == b
assert hash(a) == hash(b)
assert a is not b
assert a != c
assert hash(a) != hash(c)
assert a != d
assert hash(a) != hash(d)
assert a != e
assert hash(a) != hash(e)
|