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
|
#!/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 pytest
from telegram import Bot, File, PassportElementError, PassportFile
from telegram._utils.datetime import UTC, to_timestamp
from tests.auxil.bot_method_checks import (
check_defaults_handling,
check_shortcut_call,
check_shortcut_signature,
)
from tests.auxil.slots import mro_slots
@pytest.fixture(scope="class")
def passport_file(bot):
pf = PassportFile(
file_id=PassportFileTestBase.file_id,
file_unique_id=PassportFileTestBase.file_unique_id,
file_size=PassportFileTestBase.file_size,
file_date=PassportFileTestBase.file_date,
)
pf.set_bot(bot)
return pf
class PassportFileTestBase:
file_id = "data"
file_unique_id = "adc3145fd2e84d95b64d68eaa22aa33e"
file_size = 50
file_date = dtm.datetime.now(tz=UTC).replace(microsecond=0)
class TestPassportFileWithoutRequest(PassportFileTestBase):
def test_slot_behaviour(self, passport_file):
inst = passport_file
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_expected_values(self, passport_file):
assert passport_file.file_id == self.file_id
assert passport_file.file_unique_id == self.file_unique_id
assert passport_file.file_size == self.file_size
assert passport_file.file_date == self.file_date
def test_to_dict(self, passport_file):
passport_file_dict = passport_file.to_dict()
assert isinstance(passport_file_dict, dict)
assert passport_file_dict["file_id"] == passport_file.file_id
assert passport_file_dict["file_unique_id"] == passport_file.file_unique_id
assert passport_file_dict["file_size"] == passport_file.file_size
assert passport_file_dict["file_date"] == to_timestamp(passport_file.file_date)
def test_de_json_localization(self, passport_file, tz_bot, offline_bot, raw_bot):
json_dict = {
"file_id": self.file_id,
"file_unique_id": self.file_unique_id,
"file_size": self.file_size,
"file_date": to_timestamp(self.file_date),
}
pf = PassportFile.de_json(json_dict, offline_bot)
pf_raw = PassportFile.de_json(json_dict, raw_bot)
pf_tz = PassportFile.de_json(json_dict, tz_bot)
# comparing utcoffsets because comparing timezones is unpredicatable
date_offset = pf_tz.file_date.utcoffset()
tz_bot_offset = tz_bot.defaults.tzinfo.utcoffset(pf_tz.file_date.replace(tzinfo=None))
assert pf_raw.file_date.tzinfo == UTC
assert pf.file_date.tzinfo == UTC
assert date_offset == tz_bot_offset
def test_equality(self):
a = PassportFile(self.file_id, self.file_unique_id, self.file_size, self.file_date)
b = PassportFile("", self.file_unique_id, self.file_size, self.file_date)
c = PassportFile(self.file_id, self.file_unique_id, "", "")
d = PassportFile("", "", self.file_size, self.file_date)
e = PassportElementError("source", "type", "message")
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)
async def test_get_file_instance_method(self, monkeypatch, passport_file):
async def make_assertion(*_, **kwargs):
result = kwargs["file_id"] == passport_file.file_id
# we need to be a bit hacky here, b/c PF.get_file needs Bot.get_file to return a File
return File(file_id=result, file_unique_id=result)
assert check_shortcut_signature(PassportFile.get_file, Bot.get_file, ["file_id"], [])
assert await check_shortcut_call(
passport_file.get_file, passport_file.get_bot(), "get_file"
)
assert await check_defaults_handling(passport_file.get_file, passport_file.get_bot())
monkeypatch.setattr(passport_file.get_bot(), "get_file", make_assertion)
assert (await passport_file.get_file()).file_id == "True"
|
