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
|
########################################################################
# File name: test_e2e.py
# This file is part of: aioxmpp
#
# LICENSE
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General 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 General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
# License along with this program. If not, see
# <http://www.gnu.org/licenses/>.
#
########################################################################
import asyncio
from aioxmpp.utils import namespaces
import aioxmpp.ping
from aioxmpp.e2etest import (
TestCase,
blocking,
blocking_timed,
)
class TestPing(TestCase):
@blocking
async def setUp(self):
self.source, self.unimplemented, self.implemented = await asyncio.gather(
self.provisioner.get_connected_client(
services=[aioxmpp.ping.PingService],
),
self.provisioner.get_connected_client(
services=[aioxmpp.DiscoClient],
),
self.provisioner.get_connected_client(
services=[aioxmpp.ping.PingService],
),
)
@blocking_timed
async def test_ping_raises_error_condition(self):
ping_svc = self.source.summon(aioxmpp.ping.PingService)
with self.assertRaisesRegex(aioxmpp.XMPPCancelError,
"service-unavailable"):
await ping_svc.ping(self.unimplemented.local_jid)
@blocking_timed
async def test_ping_server(self):
ping_svc = self.source.summon(aioxmpp.ping.PingService)
await ping_svc.ping(self.unimplemented.local_jid.replace(
localpart=None,
resource=None,
))
@blocking_timed
async def test_ping_works_with_peer_with_ping_implementation(self):
ping_svc = self.source.summon(aioxmpp.ping.PingService)
self.assertIsNone(await ping_svc.ping(self.implemented.local_jid))
@blocking_timed
async def test_ping_service_exports_feature(self):
info = await self.unimplemented.summon(
aioxmpp.DiscoClient
).query_info(
self.source.local_jid,
)
self.assertIn(
namespaces.xep0199_ping,
info.features,
)
@blocking_timed
async def test_ping_service_replies_to_ping(self):
req = aioxmpp.IQ(
type_=aioxmpp.IQType.GET,
to=self.source.local_jid,
payload=aioxmpp.ping.Ping(),
)
resp = await self.unimplemented.send(req)
self.assertIsInstance(resp, aioxmpp.ping.Ping)
|