mirror of
https://gitlab.com/chicken-riders/RcGcDw.git
synced 2025-02-22 00:14:10 +00:00
67 lines
3 KiB
Python
67 lines
3 KiB
Python
# This file is part of Recent changes Goat compatible Discord webhook (RcGcDw).
|
|
#
|
|
# RcGcDw is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# RcGcDw is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with RcGcDw. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import unittest
|
|
from typing import Tuple
|
|
|
|
from src.discord.queue import MessageQueue
|
|
from src.discord.message import DiscordMessage, DiscordMessageMetadata
|
|
|
|
|
|
def create_dummy(id: int = 0, **kwargs) -> Tuple[DiscordMessage, DiscordMessageMetadata]:
|
|
dm = DiscordMessage(event_type="log/{}".format(id), message_type="embed", webhook_url="https://example.com/")
|
|
dmm = DiscordMessageMetadata("POST", log_id=kwargs.get("log_id", int(id)*10), page_id=kwargs.get("page_id", int(id)),
|
|
rev_id=kwargs.get("rev_id", int(id)*10), webhook_url=kwargs.get("webhook_url", "https://example.com/"))
|
|
return dm, dmm
|
|
|
|
|
|
class TestQueue(unittest.TestCase):
|
|
def test_add_message(self):
|
|
queue = MessageQueue()
|
|
for _ in range(100):
|
|
queue.add_message(create_dummy())
|
|
self.assertEqual(len(queue), 100)
|
|
|
|
def test_cut_messages(self):
|
|
queue = MessageQueue()
|
|
for num in range(100):
|
|
queue.add_message(create_dummy(id=num))
|
|
queue.cut_messages(10)
|
|
self.assertEqual(list(queue)[0][1].page_id, 10)
|
|
|
|
def test_compare_message_to_dict(self):
|
|
queue = MessageQueue()
|
|
passing = [create_dummy(id=103, page_id=3928, rev_id=228848), create_dummy(id=108, page_id=3928, rev_id=228853)]
|
|
failing = [create_dummy(id=105, page_id=39, rev_id=2288), create_dummy(id=110, page_id=392, rev_id=228)]
|
|
for msg in passing:
|
|
with self.subTest():
|
|
self.assertTrue(queue.compare_message_to_dict(msg[1], {"page_id": 3928}))
|
|
for msg in failing:
|
|
with self.subTest():
|
|
self.assertFalse(queue.compare_message_to_dict(msg[1], {"page_id": 3928}))
|
|
|
|
def test_delete_all_with_matching_metadata(self):
|
|
queue = MessageQueue()
|
|
queue.add_message(create_dummy(id=103, page_id=500, rev_id=228844))
|
|
for num in range(100):
|
|
queue.add_message(create_dummy(id=num))
|
|
queue.add_message(create_dummy(id=105, page_id=3923, rev_id=228848))
|
|
queue_correct = MessageQueue()
|
|
for num in range(100):
|
|
queue_correct.add_message(create_dummy(id=num))
|
|
queue_correct.add_message(create_dummy(id=105, page_id=3923, rev_id=228848))
|
|
queue.delete_all_with_matching_metadata(page_id=500)
|
|
self.assertEqual(len(queue), len(queue_correct)) # Could be better but I'm running out of time
|