Add message tests

tests/test_message.py

@@ -0,0 +1,140 @@
+import pytest
+import fbchat
+from fbchat._message import (
+    EmojiSize,
+    Mention,
+    Message,
+    graphql_to_extensible_attachment,
+)
+
+
+@pytest.mark.parametrize(
+    "tags,size",
+    [
+        (None, None),
+        (["hot_emoji_size:unknown"], None),
+        (["bunch", "of:different", "tags:large", "hot_emoji_size:s"], EmojiSize.SMALL),
+        (["hot_emoji_size:s"], EmojiSize.SMALL),
+        (["hot_emoji_size:m"], EmojiSize.MEDIUM),
+        (["hot_emoji_size:l"], EmojiSize.LARGE),
+        (["hot_emoji_size:small"], EmojiSize.SMALL),
+        (["hot_emoji_size:medium"], EmojiSize.MEDIUM),
+        (["hot_emoji_size:large"], EmojiSize.LARGE),
+    ],
+)
+def test_emojisize_from_tags(tags, size):
+    assert size is EmojiSize._from_tags(tags)
+
+
+def test_graphql_to_extensible_attachment_empty():
+    assert None is graphql_to_extensible_attachment({})
+
+
+@pytest.mark.parametrize(
+    "obj,type_",
+    [
+        # UnsentMessage testing is done in test_attachment.py
+        (fbchat.LocationAttachment, "MessageLocation"),
+        (fbchat.LiveLocationAttachment, "MessageLiveLocation"),
+        (fbchat.ShareAttachment, "ExternalUrl"),
+        (fbchat.ShareAttachment, "Story"),
+    ],
+)
+def test_graphql_to_extensible_attachment_dispatch(monkeypatch, obj, type_):
+    monkeypatch.setattr(obj, "_from_graphql", lambda data: True)
+    data = {"story_attachment": {"target": {"__typename": type_}}}
+    assert graphql_to_extensible_attachment(data)
+
+
+def test_message_format_mentions():
+    expected = Message(
+        text="Hey 'Peter'! My name is Michael",
+        mentions=[
+            Mention(thread_id="1234", offset=4, length=7),
+            Mention(thread_id="4321", offset=24, length=7),
+        ],
+    )
+    assert expected == Message.format_mentions(
+        "Hey {!r}! My name is {}", ("1234", "Peter"), ("4321", "Michael")
+    )
+    assert expected == Message.format_mentions(
+        "Hey {p!r}! My name is {}", ("4321", "Michael"), p=("1234", "Peter")
+    )
+
+
+def test_message_get_forwarded_from_tags():
+    assert not Message._get_forwarded_from_tags(None)
+    assert not Message._get_forwarded_from_tags(["hot_emoji_size:unknown"])
+    assert Message._get_forwarded_from_tags(
+        ["attachment:photo", "inbox", "sent", "source:chat:forward", "tq"]
+    )
+
+
+def test_message_to_send_data_minimal():
+    assert {"action_type": "ma-type:user-generated-message", "body": "Hey"} == Message(
+        text="Hey"
+    )._to_send_data()
+
+
+def test_message_to_send_data_mentions():
+    msg = Message(
+        text="Hey 'Peter'! My name is Michael",
+        mentions=[
+            Mention(thread_id="1234", offset=4, length=7),
+            Mention(thread_id="4321", offset=24, length=7),
+        ],
+    )
+    assert {
+        "action_type": "ma-type:user-generated-message",
+        "body": "Hey 'Peter'! My name is Michael",
+        "profile_xmd[0][id]": "1234",
+        "profile_xmd[0][length]": 7,
+        "profile_xmd[0][offset]": 4,
+        "profile_xmd[0][type]": "p",
+        "profile_xmd[1][id]": "4321",
+        "profile_xmd[1][length]": 7,
+        "profile_xmd[1][offset]": 24,
+        "profile_xmd[1][type]": "p",
+    } == msg._to_send_data()
+
+
+def test_message_to_send_data_sticker():
+    msg = Message(sticker=fbchat.Sticker(uid="123"))
+    assert {
+        "action_type": "ma-type:user-generated-message",
+        "sticker_id": "123",
+    } == msg._to_send_data()
+
+
+def test_message_to_send_data_emoji():
+    msg = Message(text="😀", emoji_size=EmojiSize.LARGE)
+    assert {
+        "action_type": "ma-type:user-generated-message",
+        "body": "😀",
+        "tags[0]": "hot_emoji_size:large",
+    } == msg._to_send_data()
+    msg = Message(emoji_size=EmojiSize.LARGE)
+    assert {
+        "action_type": "ma-type:user-generated-message",
+        "sticker_id": "369239383222810",
+    } == msg._to_send_data()
+
+
+@pytest.mark.skip(reason="need to be added")
+def test_message_to_send_data_quick_replies():
+    raise NotImplementedError
+
+
+@pytest.mark.skip(reason="need to gather test data")
+def test_message_from_graphql():
+    pass
+
+
+@pytest.mark.skip(reason="need to gather test data")
+def test_message_from_reply():
+    pass
+
+
+@pytest.mark.skip(reason="need to gather test data")
+def test_message_from_pull():
+    pass