Test updates

Author: davidhozic

src/daf/message/base.py

@@ -127,11 +127,12 @@ def __init__(self,
         self._created_at = datetime.now()
         self._data = data
         self._fbcdata = isinstance(data, _FunctionBaseCLASS)
+        self._deleted = False
         # Attributes created with this function will not be re-referenced to a different object
         # if the function is called again, ensuring safety (.update_method)
         misc._write_attr_once(self, "update_semaphore", asyncio.Semaphore(1))
         # For comparing copies of the object (prevents .update from overwriting)
-        misc._write_attr_once(self, "_id", id(self)) 
+        misc._write_attr_once(self, "_id", id(self))
 
     def __repr__(self) -> str:
         return f"{type(self).__name__}(data={self._data})"

testing/test_autogen.py

@@ -30,7 +30,7 @@ async def test_autoguild(guilds, accounts):
         assert guild_exclude not in found, "AutoGUILD included the guild that matches exclude pattern."
     finally:
         if auto_guild is not None:
-            daf.remove_object(auto_guild)
+            await daf.remove_object(auto_guild)
 
 @pytest.mark.asyncio
 async def test_autochannel(guilds, channels, accounts):
@@ -77,7 +77,7 @@ async def test_autochannel(guilds, channels, accounts):
         await auto_channel2.update()
     finally:
         if daf_guild is not None:
-            daf.remove_object(daf_guild)
+            await daf.remove_object(daf_guild)
 
 
 

testing/test_period_dynamic.py

@@ -105,9 +105,9 @@ async def dynamic_getter_async(items: list):
 
     finally:
         with suppress(ValueError):
-            daf.remove_object(guild)
+            await daf.remove_object(guild)
         with suppress(ValueError):
-            daf.remove_object(user)
+            await daf.remove_object(user)
 
 
 @pytest.mark.asyncio
@@ -153,4 +153,4 @@ def dynamic_getter(items: list):
 
     finally:
         with suppress(ValueError):
-            daf.remove_object(guild)
+            await daf.remove_object(guild)

testing/test_send_normal.py

@@ -47,7 +47,7 @@ async def test_text_message_send(channels: Tuple[daf.discord.ChannelType], guild
     assert len(result["channels"]["failed"]) == 0, "Failed to send to all channels"
 
     # DirectMESSAGE send
-    result = await direct_message._send()
+    result, _panic = await direct_message._send()
     sent_data_result = result["sent_data"]
 
     # Check results

testing/test_update.py

@@ -49,7 +49,7 @@ async def test_text_message_update(channels, guilds, accounts):
 
         # DirectMESSAGE send
         await direct_message.update(data=data)
-        result = await direct_message._send()
+        result, _panic = await direct_message._send()
         print(direct_message)
 
         # Check results