151a114235
* Make TravisCI setup * Use pytest, move tests to seperate files * Added system to check if `onX` events were successfully executed
80 lines
2.3 KiB
Python
80 lines
2.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
from __future__ import unicode_literals
|
|
|
|
import pytest
|
|
|
|
from os import path
|
|
from fbchat.models import ThreadType, Message, Mention, EmojiSize, Sticker
|
|
from utils import subset
|
|
|
|
|
|
def test_fetch_all_users(client):
|
|
users = client.fetchAllUsers()
|
|
assert len(users) > 0
|
|
|
|
|
|
def test_fetch_thread_list(client):
|
|
threads = client.fetchThreadList(limit=2)
|
|
assert len(threads) == 2
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"emoji, emoji_size",
|
|
[
|
|
("😆", EmojiSize.SMALL),
|
|
("😆", EmojiSize.MEDIUM),
|
|
("😆", EmojiSize.LARGE),
|
|
# These fail because the emoji is made into a sticker
|
|
# This should be fixed
|
|
pytest.mark.xfail((None, EmojiSize.SMALL)),
|
|
pytest.mark.xfail((None, EmojiSize.MEDIUM)),
|
|
pytest.mark.xfail((None, EmojiSize.LARGE)),
|
|
],
|
|
)
|
|
def test_fetch_message_emoji(client, emoji, emoji_size):
|
|
mid = client.sendEmoji(emoji, emoji_size)
|
|
message, = client.fetchThreadMessages(limit=1)
|
|
|
|
assert subset(
|
|
vars(message), uid=mid, author=client.uid, text=emoji, emoji_size=emoji_size
|
|
)
|
|
|
|
|
|
def test_fetch_message_mentions(client):
|
|
text = "This is a test of fetchThreadMessages"
|
|
mentions = [Mention(client.uid, offset=10, length=4)]
|
|
|
|
mid = client.send(Message(text, mentions=mentions))
|
|
message, = client.fetchThreadMessages(limit=1)
|
|
|
|
assert subset(vars(message), uid=mid, author=client.uid, text=text)
|
|
for i, m in enumerate(mentions):
|
|
assert vars(message.mentions[i]) == vars(m)
|
|
|
|
|
|
@pytest.mark.parametrize("sticker_id", ["767334476626295"])
|
|
def test_fetch_message_sticker(client, sticker_id):
|
|
mid = client.send(Message(sticker=Sticker(sticker_id)))
|
|
message, = client.fetchThreadMessages(limit=1)
|
|
|
|
assert subset(vars(message), uid=mid, author=client.uid)
|
|
assert subset(vars(message.sticker), uid=sticker_id)
|
|
|
|
|
|
def test_fetch_info(client1, group):
|
|
info = client1.fetchUserInfo("4")["4"]
|
|
assert info.name == "Mark Zuckerberg"
|
|
|
|
info = client1.fetchGroupInfo(group["id"])[group["id"]]
|
|
assert info.type == ThreadType.GROUP
|
|
|
|
|
|
def test_fetch_image_url(client):
|
|
url = path.join(path.dirname(__file__), "image.png")
|
|
|
|
client.sendLocalImage(url)
|
|
message, = client.fetchThreadMessages(limit=1)
|
|
|
|
assert client.fetchImageUrl(message.attachments[0].uid)
|