from test.lib.util import StanzaHandlerTest
from nbxmpp.modules.ogp import OpenGraph
from nbxmpp.protocol import Message
from nbxmpp.structs import MessageProperties
from nbxmpp.structs import OpenGraphData
class MockOpenGraph(OpenGraph):
def __init__(self) -> None:
pass
class OpenGraphTest(StanzaHandlerTest):
def test_parse_ogp(self) -> None:
msg = Message(node=_DATA)
self._parse_and_assert_values(msg)
def test_parse_wikipedia(self) -> None:
msg = Message(
node="""
Wikipédia, l'encyclopédie libre
https://fr.wikipedia.org/wiki/Wikip%C3%A9dia:Accueil_principal
https://fr.wikipedia.org
"""
)
props = self._parse(msg)
ogp = props["https://fr.wikipedia.org"]
self.assertEqual(ogp.title, "Wikipédia, l'encyclopédie libre")
self.assertEqual(
ogp.url, "https://fr.wikipedia.org/wiki/Wikip%C3%A9dia:Accueil_principal"
)
def test_parse_youtube(self) -> None:
msg = Message(
node="""
Pierre Lapointe - Je déteste ma vie (Paris tristesse) - YouTube
Paris tristesse en vente maintenant: http://smarturl.it/paristristesseRéalisation | montage : Pascal Grandmaison et Philippe CraigDirection photo: Pascal Gra...
https://www.youtube.com/watch?v=45V5qdjhbGI&t=1
YouTube
https://www.youtube.com/watch?v=45V5qdjhbGI&t=1
"""
)
props = self._parse(msg)
ogp = props["https://www.youtube.com/watch?v=45V5qdjhbGI&t=1"]
self.assertEqual(
ogp.title, "Pierre Lapointe - Je déteste ma vie (Paris tristesse) - YouTube"
)
self.assertEqual(
ogp.description,
"Paris tristesse en vente maintenant: http://smarturl.it/paristristesseRéalisation | montage : Pascal Grandmaison et Philippe CraigDirection photo: Pascal Gra...",
)
self.assertEqual(ogp.site_name, "YouTube")
def test_set_ogp(self) -> None:
msg1 = Message(node=_DATA)
msg2 = Message(to=msg1.getTo(), body=msg1.getBody())
msg2.addOpenGraph(
"https://the.link.example.com/what-was-linked-to",
OpenGraphData(
title="Page Title",
url="Canonical URL",
image="https://link.to.example.com/image.png",
type="website",
site_name="Some Website",
description="Page Description",
),
)
self._parse_and_assert_values(msg2)
def _parse(self, msg: Message) -> dict[str, OpenGraphData]:
properties = MessageProperties("whatever@whatever.com")
parser = MockOpenGraph()
parser._process_message_opengraph(None, msg, properties)
self.assertEqual(len(properties.open_graph), 1)
return properties.open_graph
def _parse_and_assert_values(self, msg: Message) -> None:
props = self._parse(msg)
ogp = props["https://the.link.example.com/what-was-linked-to"]
self.assertEqual(ogp.title, "Page Title")
self.assertEqual(ogp.url, "Canonical URL")
self.assertEqual(ogp.image, "https://link.to.example.com/image.png")
self.assertEqual(ogp.type, "website")
self.assertEqual(ogp.site_name, "Some Website")
self.assertEqual(ogp.description, "Page Description")
_DATA = """
I wanted to mention https://the.link.example.com/what-was-linked-to
Page Title
Page Description
Canonical URL
https://link.to.example.com/image.png
website
Some Website
"""