forked from python-telegram-bot/python-telegram-bot
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest_inputfile.py
More file actions
133 lines (114 loc) · 5.16 KB
/
test_inputfile.py
File metadata and controls
133 lines (114 loc) · 5.16 KB
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
#!/usr/bin/env python
#
# A library that provides a Python interface to the Telegram Bot API
# Copyright (C) 2015-2021
# Leandro Toledo de Souza <[email protected]>
#
# 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 logging
import os
import subprocess
import sys
from io import BytesIO
from telegram import InputFile
class TestInputFile:
png = os.path.join("tests", "data", "game.png")
def test_slot_behaviour(self, mro_slots):
inst = InputFile(BytesIO(b"blah"), filename="tg.jpg")
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_subprocess_pipe(self):
if sys.platform == "win32":
cmd = ["type", self.png]
else:
cmd = ["cat", self.png]
proc = subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=(sys.platform == "win32"))
in_file = InputFile(proc.stdout)
assert in_file.input_file_content == open(self.png, "rb").read()
assert in_file.mimetype == "image/png"
assert in_file.filename == "image.png"
try:
proc.kill()
except ProcessLookupError:
# This exception may be thrown if the process has finished before we had the chance
# to kill it.
pass
def test_mimetypes(self, caplog):
# Only test a few to make sure logic works okay
assert InputFile(open("tests/data/telegram.jpg", "rb")).mimetype == "image/jpeg"
assert InputFile(open("tests/data/telegram.webp", "rb")).mimetype == "image/webp"
assert InputFile(open("tests/data/telegram.mp3", "rb")).mimetype == "audio/mpeg"
# Test guess from file
assert InputFile(BytesIO(b"blah"), filename="tg.jpg").mimetype == "image/jpeg"
assert InputFile(BytesIO(b"blah"), filename="tg.mp3").mimetype == "audio/mpeg"
# Test fallback
assert (
InputFile(BytesIO(b"blah"), filename="tg.notaproperext").mimetype
== "application/octet-stream"
)
assert InputFile(BytesIO(b"blah")).mimetype == "application/octet-stream"
# Test string file
with caplog.at_level(logging.DEBUG):
assert InputFile(open("tests/data/text_file.txt")).mimetype == "text/plain"
assert len(caplog.records) == 1
assert caplog.records[0].getMessage().startswith("Could not parse file content")
def test_filenames(self):
assert InputFile(open("tests/data/telegram.jpg", "rb")).filename == "telegram.jpg"
assert InputFile(open("tests/data/telegram.jpg", "rb"), filename="blah").filename == "blah"
assert (
InputFile(open("tests/data/telegram.jpg", "rb"), filename="blah.jpg").filename
== "blah.jpg"
)
assert InputFile(open("tests/data/telegram", "rb")).filename == "telegram"
assert InputFile(open("tests/data/telegram", "rb"), filename="blah").filename == "blah"
assert (
InputFile(open("tests/data/telegram", "rb"), filename="blah.jpg").filename
== "blah.jpg"
)
class MockedFileobject:
# A open(?, 'rb') without a .name
def __init__(self, f):
self.f = open(f, "rb")
def read(self):
return self.f.read()
assert InputFile(MockedFileobject("tests/data/telegram.jpg")).filename == "image.jpeg"
assert (
InputFile(MockedFileobject("tests/data/telegram.jpg"), filename="blah").filename
== "blah"
)
assert (
InputFile(MockedFileobject("tests/data/telegram.jpg"), filename="blah.jpg").filename
== "blah.jpg"
)
assert (
InputFile(MockedFileobject("tests/data/telegram")).filename
== "application.octet-stream"
)
assert (
InputFile(MockedFileobject("tests/data/telegram"), filename="blah").filename == "blah"
)
assert (
InputFile(MockedFileobject("tests/data/telegram"), filename="blah.jpg").filename
== "blah.jpg"
)
def test_send_bytes(self, bot, chat_id):
# We test this here and not at the respective test modules because it's not worth
# duplicating the test for the different methods
with open("tests/data/text_file.txt", "rb") as file:
message = bot.send_document(chat_id, file.read())
out = BytesIO()
assert message.document.get_file().download(out=out)
out.seek(0)
assert out.read().decode("utf-8") == "PTB Rocks!"