diff --git a/tests/unit/test_ap-gateway.py b/tests/unit/test_ap-gateway.py index 7e84b64d..7ecf023c 100644 --- a/tests/unit/test_ap-gateway.py +++ b/tests/unit/test_ap-gateway.py @@ -36,6 +36,7 @@ from wokkel import pubsub, rsm from libervia.backend.core import exceptions from libervia.backend.core.constants import Const as C +from libervia.backend.core.xmpp import SatXMPPComponent from libervia.backend.memory.sqla_mapping import SubscriptionState from libervia.backend.plugins import plugin_comp_ap_gateway from libervia.backend.plugins.plugin_comp_ap_gateway import constants as ap_const @@ -386,6 +387,7 @@ def mock_client(jid): client.host = "test.example" client._ap_storage.get = AsyncMock() client._ap_storage.aset = AsyncMock() + client.is_local = lambda jid_: SatXMPPComponent.is_local(client, jid_) return client diff --git a/tests/unit/test_plugin_xep_0215.py b/tests/unit/test_plugin_xep_0215.py index 1d687bbb..799a880e 100644 --- a/tests/unit/test_plugin_xep_0215.py +++ b/tests/unit/test_plugin_xep_0215.py @@ -18,7 +18,7 @@ from twisted.internet import defer from pytest_twisted import ensureDeferred as ed -from unittest.mock import MagicMock +from unittest.mock import AsyncMock, MagicMock from libervia.backend.plugins.plugin_xep_0215 import XEP_0215 from libervia.backend.tools import xml_tools from twisted.words.protocols.jabber import jid @@ -28,6 +28,9 @@ class TestXEP0215: def test_parse_services(self, host): """Services are parsed correctly""" xep_0215 = XEP_0215(host) + host.memory.disco.has_feature = AsyncMock() + host.memory.disco.has_feature.return_value = True + services_elt = xml_tools.parse( """