Add attachment tests
This commit is contained in:
Mads Marquart
199
tests/test_attachment.py
Normal file
199
tests/test_attachment.py
Normal file
@@ -0,0 +1,199 @@
|
|||||||
|
import pytest
|
||||||
|
import fbchat
|
||||||
|
from fbchat._attachment import UnsentMessage, ShareAttachment
|
||||||
|
|
||||||
|
|
||||||
|
def test_parse_unsent_message():
|
||||||
|
data = {
|
||||||
|
"legacy_attachment_id": "ee.mid.$xyz",
|
||||||
|
"story_attachment": {
|
||||||
|
"description": {"text": "You removed a message"},
|
||||||
|
"media": None,
|
||||||
|
"source": None,
|
||||||
|
"style_list": ["globally_deleted_message_placeholder", "fallback"],
|
||||||
|
"title_with_entities": {"text": ""},
|
||||||
|
"properties": [],
|
||||||
|
"url": None,
|
||||||
|
"deduplication_key": "deadbeef123",
|
||||||
|
"action_links": [],
|
||||||
|
"messaging_attribution": None,
|
||||||
|
"messenger_call_to_actions": [],
|
||||||
|
"xma_layout_info": None,
|
||||||
|
"target": None,
|
||||||
|
"subattachments": [],
|
||||||
|
},
|
||||||
|
"genie_attachment": {"genie_message": None},
|
||||||
|
}
|
||||||
|
assert UnsentMessage(
|
||||||
|
uid="ee.mid.$xyz"
|
||||||
|
) == fbchat._message.graphql_to_extensible_attachment(data)
|
||||||
|
|
||||||
|
|
||||||
|
def test_share_from_graphql_minimal():
|
||||||
|
data = {
|
||||||
|
"target": {},
|
||||||
|
"url": "a.com",
|
||||||
|
"title_with_entities": {"text": "a.com"},
|
||||||
|
"subattachments": [],
|
||||||
|
}
|
||||||
|
assert ShareAttachment(
|
||||||
|
url="a.com", original_url="a.com", title="a.com"
|
||||||
|
) == ShareAttachment._from_graphql(data)
|
||||||
|
|
||||||
|
|
||||||
|
def test_share_from_graphql_link():
|
||||||
|
data = {
|
||||||
|
"description": {"text": ""},
|
||||||
|
"media": {
|
||||||
|
"animated_image": None,
|
||||||
|
"image": None,
|
||||||
|
"playable_duration_in_ms": 0,
|
||||||
|
"is_playable": False,
|
||||||
|
"playable_url": None,
|
||||||
|
},
|
||||||
|
"source": {"text": "a.com"},
|
||||||
|
"style_list": ["share", "fallback"],
|
||||||
|
"title_with_entities": {"text": "a.com"},
|
||||||
|
"properties": [],
|
||||||
|
"url": "http://l.facebook.com/l.php?u=http%3A%2F%2Fa.com%2F&h=def&s=1",
|
||||||
|
"deduplication_key": "ee.mid.$xyz",
|
||||||
|
"action_links": [{"title": "About this website", "url": None}],
|
||||||
|
"messaging_attribution": None,
|
||||||
|
"messenger_call_to_actions": [],
|
||||||
|
"xma_layout_info": None,
|
||||||
|
"target": {"__typename": "ExternalUrl"},
|
||||||
|
"subattachments": [],
|
||||||
|
}
|
||||||
|
assert ShareAttachment(
|
||||||
|
author=None,
|
||||||
|
url="http://l.facebook.com/l.php?u=http%3A%2F%2Fa.com%2F&h=def&s=1",
|
||||||
|
original_url="http://a.com/",
|
||||||
|
title="a.com",
|
||||||
|
description="",
|
||||||
|
source="a.com",
|
||||||
|
image_url=None,
|
||||||
|
original_image_url=None,
|
||||||
|
image_width=None,
|
||||||
|
image_height=None,
|
||||||
|
attachments=[],
|
||||||
|
uid="ee.mid.$xyz",
|
||||||
|
) == ShareAttachment._from_graphql(data)
|
||||||
|
|
||||||
|
|
||||||
|
def test_share_from_graphql_link_with_image():
|
||||||
|
data = {
|
||||||
|
"description": {
|
||||||
|
"text": (
|
||||||
|
"Create an account or log in to Facebook."
|
||||||
|
" Connect with friends, family and other people you know."
|
||||||
|
" Share photos and videos, send messages and get updates."
|
||||||
|
)
|
||||||
|
},
|
||||||
|
"media": {
|
||||||
|
"animated_image": None,
|
||||||
|
"image": {
|
||||||
|
"uri": "https://www.facebook.com/rsrc.php/v3/x.png",
|
||||||
|
"height": 325,
|
||||||
|
"width": 325,
|
||||||
|
},
|
||||||
|
"playable_duration_in_ms": 0,
|
||||||
|
"is_playable": False,
|
||||||
|
"playable_url": None,
|
||||||
|
},
|
||||||
|
"source": None,
|
||||||
|
"style_list": ["share", "fallback"],
|
||||||
|
"title_with_entities": {"text": "Facebook – log in or sign up"},
|
||||||
|
"properties": [],
|
||||||
|
"url": "http://facebook.com/",
|
||||||
|
"deduplication_key": "deadbeef123",
|
||||||
|
"action_links": [],
|
||||||
|
"messaging_attribution": None,
|
||||||
|
"messenger_call_to_actions": [],
|
||||||
|
"xma_layout_info": None,
|
||||||
|
"target": {"__typename": "ExternalUrl"},
|
||||||
|
"subattachments": [],
|
||||||
|
}
|
||||||
|
assert ShareAttachment(
|
||||||
|
author=None,
|
||||||
|
url="http://facebook.com/",
|
||||||
|
original_url="http://facebook.com/",
|
||||||
|
title="Facebook – log in or sign up",
|
||||||
|
description=(
|
||||||
|
"Create an account or log in to Facebook."
|
||||||
|
" Connect with friends, family and other people you know."
|
||||||
|
" Share photos and videos, send messages and get updates."
|
||||||
|
),
|
||||||
|
source=None,
|
||||||
|
image_url="https://www.facebook.com/rsrc.php/v3/x.png",
|
||||||
|
original_image_url="https://www.facebook.com/rsrc.php/v3/x.png",
|
||||||
|
image_width=325,
|
||||||
|
image_height=325,
|
||||||
|
attachments=[],
|
||||||
|
uid="deadbeef123",
|
||||||
|
) == ShareAttachment._from_graphql(data)
|
||||||
|
|
||||||
|
|
||||||
|
def test_share_from_graphql_video():
|
||||||
|
data = {
|
||||||
|
"description": {
|
||||||
|
"text": (
|
||||||
|
"Rick Astley's official music video for “Never Gonna Give You Up”"
|
||||||
|
" Listen to Rick Astley: https://RickAstley.lnk.to/_listenYD"
|
||||||
|
" Subscribe to the official Rick As..."
|
||||||
|
)
|
||||||
|
},
|
||||||
|
"media": {
|
||||||
|
"animated_image": None,
|
||||||
|
"image": {
|
||||||
|
"uri": (
|
||||||
|
"https://external-arn2-1.xx.fbcdn.net/safe_image.php?d=xyz123"
|
||||||
|
"&w=960&h=540&url=https%3A%2F%2Fi.ytimg.com%2Fvi%2FdQw4w9WgXcQ"
|
||||||
|
"%2Fmaxresdefault.jpg&sx=0&sy=0&sw=1280&sh=720&_nc_hash=abc123"
|
||||||
|
),
|
||||||
|
"height": 540,
|
||||||
|
"width": 960,
|
||||||
|
},
|
||||||
|
"playable_duration_in_ms": 0,
|
||||||
|
"is_playable": True,
|
||||||
|
"playable_url": "https://www.youtube.com/embed/dQw4w9WgXcQ?autoplay=1",
|
||||||
|
},
|
||||||
|
"source": {"text": "youtube.com"},
|
||||||
|
"style_list": ["share", "fallback"],
|
||||||
|
"title_with_entities": {
|
||||||
|
"text": "Rick Astley - Never Gonna Give You Up (Video)"
|
||||||
|
},
|
||||||
|
"properties": [
|
||||||
|
{"key": "width", "value": {"text": "1280"}},
|
||||||
|
{"key": "height", "value": {"text": "720"}},
|
||||||
|
],
|
||||||
|
"url": "https://l.facebook.com/l.php?u=https%3A%2F%2Fyoutu.be%2FdQw4w9WgXcQ",
|
||||||
|
"deduplication_key": "ee.mid.$gAAT4Sw1WSGhzQ9uRWVtEpZHZ8ZPV",
|
||||||
|
"action_links": [{"title": "About this website", "url": None}],
|
||||||
|
"messaging_attribution": None,
|
||||||
|
"messenger_call_to_actions": [],
|
||||||
|
"xma_layout_info": None,
|
||||||
|
"target": {"__typename": "ExternalUrl"},
|
||||||
|
"subattachments": [],
|
||||||
|
}
|
||||||
|
assert ShareAttachment(
|
||||||
|
author=None,
|
||||||
|
url="https://l.facebook.com/l.php?u=https%3A%2F%2Fyoutu.be%2FdQw4w9WgXcQ",
|
||||||
|
original_url="https://youtu.be/dQw4w9WgXcQ",
|
||||||
|
title="Rick Astley - Never Gonna Give You Up (Video)",
|
||||||
|
description=(
|
||||||
|
"Rick Astley's official music video for “Never Gonna Give You Up”"
|
||||||
|
" Listen to Rick Astley: https://RickAstley.lnk.to/_listenYD"
|
||||||
|
" Subscribe to the official Rick As..."
|
||||||
|
),
|
||||||
|
source="youtube.com",
|
||||||
|
image_url=(
|
||||||
|
"https://external-arn2-1.xx.fbcdn.net/safe_image.php?d=xyz123"
|
||||||
|
"&w=960&h=540&url=https%3A%2F%2Fi.ytimg.com%2Fvi%2FdQw4w9WgXcQ"
|
||||||
|
"%2Fmaxresdefault.jpg&sx=0&sy=0&sw=1280&sh=720&_nc_hash=abc123"
|
||||||
|
),
|
||||||
|
original_image_url="https://i.ytimg.com/vi/dQw4w9WgXcQ/maxresdefault.jpg",
|
||||||
|
image_width=960,
|
||||||
|
image_height=540,
|
||||||
|
attachments=[],
|
||||||
|
uid="ee.mid.$gAAT4Sw1WSGhzQ9uRWVtEpZHZ8ZPV",
|
||||||
|
) == ShareAttachment._from_graphql(data)
|
||||||
Reference in New Issue
Block a user