async def test_removeOutdatedTimestamps(self):
ast = AntiSpamTracker(
AntiSpamHandler(MockedMember(mock_type="bot").to_mock()), 3, 50)
ast.update_cache(MockedMessage().to_mock(),
{"should_be_punished_this_message": True})
self.assertEqual(ast.get_user_count(MockedMessage().to_mock()), 1)
ast.remove_outdated_timestamps(123456789, 12345)
self.assertEqual(ast.get_user_count(MockedMessage().to_mock()), 1)
await asyncio.sleep(0.06)
ast.remove_outdated_timestamps(123456789, 12345)
self.assertEqual(ast.get_user_count(MockedMessage().to_mock()), 0)
async def test_removePunishment(self):
ast = AntiSpamTracker(
AntiSpamHandler(MockedMember(mock_type="bot").to_mock()), 5)
ast.remove_punishments(MockedMessage().to_mock())
ast.update_cache(MockedMessage().to_mock(),
{"should_be_punished_this_message": True})
self.assertEqual(1, ast.get_user_count(MockedMessage().to_mock()))
ast.remove_punishments(MockedMessage().to_mock())
with self.assertRaises(UserNotFound):
ast.get_user_count(MockedMessage().to_mock())
async def test_getUserCount(self):
ast = AntiSpamTracker(
AntiSpamHandler(MockedMember(mock_type="bot").to_mock()), 3)
ast.update_cache(MockedMessage().to_mock(),
{"should_be_punished_this_message": True})
user_count = ast.get_user_count(MockedMessage().to_mock())
self.assertEqual(user_count, 1)
for i in range(3):
ast.update_cache(MockedMessage().to_mock(),
{"should_be_punished_this_message": True})
user_count = ast.get_user_count(MockedMessage().to_mock())
self.assertEqual(user_count, 4)
with self.assertRaises(UserNotFound):
message = MockedMessage(guild_id=15).to_mock()
ast.get_user_count(message)
with self.assertRaises(UserNotFound):
message = MockedMessage(author_id=25).to_mock()
ast.get_user_count(message)