chore: update tests inline with fix for #100

Skelmis · Sep 1, 2022 · dd325f6 · dd325f6
1 parent 163199e
commit dd325f6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/tests/test_anti_spam_handler.py b/tests/test_anti_spam_handler.py
@@ -832,9 +832,7 @@ async def test_invalid_message(self, create_handler):
         message.attachments = True
 
         return_data = await create_handler.propagate(message)
-        assert return_data == {
-            "status": "Could not create a use-able message for the given message."
-        }
+        assert return_data == {"status": "We don't check against attachments"}
 
     def test_set_cache(self, create_handler):
         """Tests set_cache works as expected"""

diff --git a/tests/test_lib_dpy.py b/tests/test_lib_dpy.py
@@ -29,15 +29,15 @@ async def test_create_message_raises(self, create_dpy_lib_handler):
         message.embeds = None
         message.attachments = None
 
-        with pytest.raises(LogicError):
+        with pytest.raises(InvalidMessage):
             await create_dpy_lib_handler.create_message(message)
 
-        with pytest.raises(LogicError):
+        with pytest.raises(InvalidMessage):
             # Test embeds that ain't embeds raise
             message.embeds = [MockClass()]
             await create_dpy_lib_handler.create_message(message)
 
-        with pytest.raises(LogicError):
+        with pytest.raises(InvalidMessage):
             # test embed type raises
             embed = discord.Embed()
             embed.type = "Not rich"