libervia-backend: tests/unit/test_ap-gateway.py comparison

comparison tests/unit/test_ap-gateway.py @ 3783:fedbf7aade11

tests (unit/ap_gateway): fix tests

author	Goffi <goffi@goffi.org>
date	Tue, 24 May 2022 17:49:14 +0200
parents	f31113777881
children	0b54be42d0aa

comparison

equal deleted inserted replaced

-:580df5b557be
+:fedbf7aade11
 from urllib import parse
 from functools import partial
 import pytest
 from pytest_twisted import ensureDeferred as ed
+from twisted.internet import defer
 from twisted.words.protocols.jabber import jid
 from twisted.web.server import Request
+from twisted.words.xish import domish
 from wokkel import rsm, pubsub
 from sat.core import exceptions
+from sat.core.constants import Const as C
 from sat.plugins import plugin_comp_ap_gateway
+from sat.plugins.plugin_comp_ap_gateway import constants as ap_const
 from sat.plugins.plugin_comp_ap_gateway.http_server import HTTPServer
+from sat.plugins.plugin_xep_0277 import NS_ATOM
 from sat.plugins.plugin_xep_0465 import NS_PPS
 from sat.tools.utils import xmpp_date
 from sat.tools import xml_tools
 from sat.plugins.plugin_comp_ap_gateway import TYPE_ACTOR
 TEST_BASE_URL = "https://example.org"
 TEST_USER = "test_user"
 TEST_AP_ACCOUNT = f"{TEST_USER}@example.org"
+TEST_AP_ACTOR_ID = f"{TEST_BASE_URL}/users/{TEST_USER}"
 PUBLIC_URL = "test.example"
+TEST_JID = jid.JID(f"some_user@{PUBLIC_URL}")
 AP_REQUESTS = {
 f"{TEST_BASE_URL}/.well-known/webfinger?"
 f"resource=acct:{parse.quote(TEST_AP_ACCOUNT)}": {
 "aliases": [
 count=len(ret_items)
 )
 return ret_items, {"rsm": rsm_resp.toDict(), "complete": True}
+def getVirtualClient(jid):
+client = MagicMock()
+client.jid = jid
+return client
 @pytest.fixture(scope="session")
 def ap_gateway(host):
 gateway = plugin_comp_ap_gateway.APGateway(host)
 gateway.initialised = True
 gateway.isPubsub = AsyncMock()
 gateway.isPubsub.return_value = False
 client = MagicMock()
 client.jid = jid.JID("ap.test.example")
 client.host = "test.example"
+client.getVirtualClient = getVirtualClient
 gateway.client = client
 gateway.local_only = True
 gateway.public_url = PUBLIC_URL
 gateway.ap_path = '_ap'
 gateway.base_ap_url = parse.urljoin(
 return gateway
 class TestActivityPubGateway:
+def getTitleXHTML(self, item_elt: domish.Element) -> domish.Element:
+return next(
+t for t in item_elt.entry.elements(NS_ATOM, "title")
+if t.getAttribute("type") == "xhtml"
+)
 @ed
 async def test_jid_and_node_convert_to_ap_handle(self, ap_gateway):
 """JID and pubsub node are converted correctly to an AP actor handle"""
 get_account = ap_gateway.getAPAccountFromJidAndNode
 assert rsm_resp.count == 4
 assert rsm_resp.index == 0
 assert rsm_resp.first == "https://example.org/users/test_user/statuses/4"
 assert rsm_resp.last == "https://example.org/users/test_user/statuses/3"
-assert items[0].entry.title.toXml() == (
+title_xhtml = self.getTitleXHTML(items[0])
+assert title_xhtml.toXml() == (
 "<title xmlns='http://www.w3.org/2005/Atom' type='xhtml'>"
 "<div xmlns='http://www.w3.org/1999/xhtml'><p>test message 4</p></div>"
 "</title>"
 )
 author_uri = str(
 [e for e in items[0].entry.author.elements() if e.name == "uri"][0]
 )
 assert author_uri == "xmpp:test_user\\40example.org@ap.test.example"
 assert str(items[0].entry.published) == "2021-12-16T17:25:03Z"
-assert items[1].entry.title.toXml() == (
+title_xhtml = self.getTitleXHTML(items[1])
+assert title_xhtml.toXml() == (
 "<title xmlns='http://www.w3.org/2005/Atom' type='xhtml'>"
 "<div xmlns='http://www.w3.org/1999/xhtml'><p>test message 3</p></div>"
 "</title>"
 )
 author_uri = str(
 assert rsm_resp.count == 4
 assert rsm_resp.index == 2
 assert rsm_resp.first == "https://example.org/users/test_user/statuses/2"
 assert rsm_resp.last == "https://example.org/users/test_user/statuses/1"
-assert items[0].entry.title.toXml() == (
+title_xhtml = self.getTitleXHTML(items[0])
+assert title_xhtml.toXml() == (
 "<title xmlns='http://www.w3.org/2005/Atom' type='xhtml'>"
 "<div xmlns='http://www.w3.org/1999/xhtml'><p>test message 2</p></div>"
 "</title>"
 )
 author_uri = str(
 [e for e in items[0].entry.author.elements() if e.name == "uri"][0]
 )
 assert author_uri == "xmpp:test_user\\40example.org@ap.test.example"
 assert str(items[0].entry.published) == "2021-12-16T17:27:03Z"
-assert items[1].entry.title.toXml() == (
+title_xhtml = self.getTitleXHTML(items[1])
+assert title_xhtml.toXml() == (
 "<title xmlns='http://www.w3.org/2005/Atom' type='xhtml'>"
 "<div xmlns='http://www.w3.org/1999/xhtml'><p>test message 1</p></div>"
 "</title>"
 )
 author_uri = str(
 assert rsm_resp.index == 2
 assert rsm_resp.first == "https://example.org/users/test_user/statuses/2"
 assert rsm_resp.last == "https://example.org/users/test_user/statuses/2"
 assert len(items) == 1
-assert items[0].entry.title.toXml() == (
+title_xhtml = self.getTitleXHTML(items[0])
+assert title_xhtml.toXml() == (
 "<title xmlns='http://www.w3.org/2005/Atom' type='xhtml'>"
 "<div xmlns='http://www.w3.org/1999/xhtml'><p>test message 2</p></div>"
 "</title>"
 )
 assert str(items[0].entry.published) == "2021-12-16T17:27:03Z"
 assert rsm_resp.count == 4
 assert rsm_resp.index == 1
 assert rsm_resp.first == "https://example.org/users/test_user/statuses/3"
 assert rsm_resp.last == "https://example.org/users/test_user/statuses/1"
 assert len(items) == 3
-assert items[0].entry.title.toXml() == (
+title_xhtml = self.getTitleXHTML(items[0])
+assert title_xhtml.toXml() == (
 "<title xmlns='http://www.w3.org/2005/Atom' type='xhtml'>"
 "<div xmlns='http://www.w3.org/1999/xhtml'><p>test message 3</p></div>"
 "</title>"
 )
-assert items[2].entry.title.toXml() == (
+title_xhtml = self.getTitleXHTML(items[2])
+assert title_xhtml.toXml() == (
 "<title xmlns='http://www.w3.org/2005/Atom' type='xhtml'>"
 "<div xmlns='http://www.w3.org/1999/xhtml'><p>test message 1</p></div>"
 "</title>"
 )

Mercurial > libervia-backend

comparison tests/unit/test_ap-gateway.py @ 3783:fedbf7aade11