blob: 09e85c1d6adaa5fbdf84856a9d4dec52450ae3a2 (
plain)
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
|
try:
import aiohttp
except ImportError:
aiohttp = None
import unittest
from io import BytesIO
from slixmpp.test.integration import SlixIntegration
class TestHTTPUpload(SlixIntegration):
async def asyncSetUp(self):
await super().asyncSetUp()
self.add_client(
self.envjid('CI_ACCOUNT1'),
self.envstr('CI_ACCOUNT1_PASSWORD'),
)
self.register_plugins(['xep_0363'])
# Minimal data, we do not want to clutter the remote server
self.data = b'tototo'
await self.connect_clients()
@unittest.skipIf(aiohttp is None, "aiohttp is not installed")
async def test_httpupload(self):
"""Check we can upload a file properly."""
url = await self.clients[0]['xep_0363'].upload_file(
'toto.txt',
input_file=BytesIO(self.data),
size=len(self.data),
)
async with aiohttp.ClientSession() as session:
async with session.get(url) as resp:
text = await resp.text()
self.assertEqual(text.encode('utf-8'), self.data)
suite = unittest.TestLoader().loadTestsFromTestCase(TestHTTPUpload)
|
