From c2225bf2fded9aeae5d193927137befa34014e1c Mon Sep 17 00:00:00 2001
From: Mads Marquart <madsmtm@gmail.com>
Date: Wed, 29 Aug 2018 14:07:44 +0200
Subject: [PATCH] Added more tests

---
 tests/conftest.py                 |  8 ++++++--
 tests/test_fetch.py               |  8 ++++----
 tests/test_message_management.py  | 11 +++++++++++
 tests/test_thread_interraction.py | 18 ++++++++++++++++--
 4 files changed, 37 insertions(+), 8 deletions(-)

diff --git a/tests/conftest.py b/tests/conftest.py
index 7c3025c..754e12f 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -20,9 +20,13 @@ def group(pytestconfig):
     return {"id": load_variable("group_id", pytestconfig.cache), "type": ThreadType.GROUP}
 
 
-@pytest.fixture(scope="session", params=["user", "group"])
+@pytest.fixture(params=["user", "group", pytest.mark.xfail("none")])
 def thread(request, user, group):
-    return user if request.param == "user" else group
+    return {
+        "user": user,
+        "group": group,
+        "none": {"id": "0", "type": ThreadType.GROUP}
+    }[request.param]
 
 
 @pytest.fixture(scope="session")
diff --git a/tests/test_fetch.py b/tests/test_fetch.py
index afcbdef..669de22 100644
--- a/tests/test_fetch.py
+++ b/tests/test_fetch.py
@@ -9,13 +9,13 @@ from fbchat.models import ThreadType, Message, Mention, EmojiSize, Sticker
 from utils import subset, STICKER_LIST, EMOJI_LIST
 
 
-def test_fetch_all_users(client):
-    users = client.fetchAllUsers()
+def test_fetch_all_users(client1):
+    users = client1.fetchAllUsers()
     assert len(users) > 0
 
 
-def test_fetch_thread_list(client):
-    threads = client.fetchThreadList(limit=2)
+def test_fetch_thread_list(client1):
+    threads = client1.fetchThreadList(limit=2)
     assert len(threads) == 2
 
 
diff --git a/tests/test_message_management.py b/tests/test_message_management.py
index ed6c447..00291ee 100644
--- a/tests/test_message_management.py
+++ b/tests/test_message_management.py
@@ -5,8 +5,19 @@ from __future__ import unicode_literals
 import pytest
 
 from fbchat.models import Message, MessageReaction
+from utils import subset
 
 
 def test_set_reaction(client):
     mid = client.send(Message(text="This message will be reacted to"))
     client.reactToMessage(mid, MessageReaction.LOVE)
+
+
+def test_delete_messages(client):
+    text1 = "This message will stay"
+    text2 = "This message will be removed"
+    mid1 = client.sendMessage(text1)
+    mid2 = client.sendMessage(text2)
+    client.deleteMessages(mid2)
+    message, = client.fetchThreadMessages(limit=1)
+    assert subset(vars(message), uid=mid1, author=client.uid, text=text1)
diff --git a/tests/test_thread_interraction.py b/tests/test_thread_interraction.py
index fa4215b..ce36202 100644
--- a/tests/test_thread_interraction.py
+++ b/tests/test_thread_interraction.py
@@ -50,10 +50,9 @@ def test_remove_from_and_add_admins_to_group(client1, client2, group, catch_even
 def test_change_title(client1, group, catch_event):
     title = random_hex()
     with catch_event("onTitleChange") as x:
-        mid = client1.changeThreadTitle(title, group["id"])
+        client1.changeThreadTitle(title, group["id"], thread_type=ThreadType.GROUP)
     assert subset(
         x.res,
-        mid=mid,
         author_id=client1.uid,
         new_title=title,
         thread_id=group["id"],
@@ -138,3 +137,18 @@ def test_change_approval_mode(client1, group, catch_event, require_admin_approva
         author_id=client1.uid,
         thread_id=group["id"],
     )
+
+@pytest.mark.parametrize("mute_time", [0, 10, 100, 1000, -1])
+def test_mute_thread(client, mute_time):
+    assert client.muteThread(mute_time)
+    assert client.unmuteThread()
+
+
+def test_mute_thread_reactions(client):
+    assert client.muteThreadReactions()
+    assert client.unmuteThreadReactions()
+
+
+def test_mute_thread_mentions(client):
+    assert client.muteThreadMentions()
+    assert client.unmuteThreadMentions()