aboutsummaryrefslogtreecommitdiffstats
path: root/tests
diff options
context:
space:
mode:
authorGravatar ionite34 <[email protected]>2022-08-06 13:57:51 -0400
committerGravatar Ionite <[email protected]>2022-08-16 16:38:26 -0400
commitd93afa3dbe16404a60e23acaf394affcda5aff89 (patch)
tree894eafe05d5b9ca48501f385c92a5a96379dba04 /tests
parentRefactored test to not use datetime patch (diff)
Updated previous tests to use subset method
Diffstat (limited to 'tests')
-rw-r--r--tests/bot/exts/moderation/infraction/test_utils.py12
1 files changed, 5 insertions, 7 deletions
diff --git a/tests/bot/exts/moderation/infraction/test_utils.py b/tests/bot/exts/moderation/infraction/test_utils.py
index 5ba0f4273..6c9af2555 100644
--- a/tests/bot/exts/moderation/infraction/test_utils.py
+++ b/tests/bot/exts/moderation/infraction/test_utils.py
@@ -310,7 +310,7 @@ class TestPostInfraction(unittest.IsolatedAsyncioTestCase):
async def test_normal_post_infraction(self):
"""Should return response from POST request if there are no errors."""
now = datetime.now()
- payload = {
+ expected = {
"actor": self.ctx.author.id,
"hidden": True,
"reason": "Test reason",
@@ -323,14 +323,12 @@ class TestPostInfraction(unittest.IsolatedAsyncioTestCase):
self.ctx.bot.api_client.post.return_value = "foo"
actual = await utils.post_infraction(self.ctx, self.member, "ban", "Test reason", now, True, False)
-
self.assertEqual(actual, "foo")
self.ctx.bot.api_client.post.assert_awaited_once()
- await_args = str(self.ctx.bot.api_client.post.await_args)
- # Check existing keys present, allow for additional keys (e.g. `last_applied`)
- for key, value in payload.items():
- self.assertTrue(key in await_args)
- self.assertTrue(str(value) in await_args)
+
+ # Since `last_applied` is based on current time, just check if expected is a subset of payload
+ payload: dict = self.ctx.bot.api_client.post.await_args_list[0].kwargs["json"]
+ self.assertEqual(payload, payload | expected)
async def test_unknown_error_post_infraction(self):
"""Should send an error message to chat when a non-400 error occurs."""