Ghytro
118 lines
4.5 KiB
Python
118 lines
4.5 KiB
Python
import unittest
|
|
from msgprocessor import (
|
|
TrackerRemovalMsgProcessor,
|
|
TrackerRemoverFactory,
|
|
TrackerRemovalProcessorMessage,
|
|
TrackerRemovalResult,
|
|
)
|
|
|
|
|
|
class TestRemoverFactory(unittest.TestCase):
|
|
factory = TrackerRemoverFactory()
|
|
|
|
def test_remove_strategy_constructor(self):
|
|
test_case_data = [
|
|
{
|
|
"domain": "youtube.com",
|
|
"remover": self.factory.remove_yt_trackers
|
|
},
|
|
{
|
|
"domain": "lowerlevel.youtube.com",
|
|
"remover": self.factory.remove_yt_trackers
|
|
},
|
|
{
|
|
"domain": "youtu.be",
|
|
"remover": self.factory.remove_yt_trackers,
|
|
},
|
|
{
|
|
"domain": "something.youtu.be",
|
|
"remover": self.factory.remove_yt_trackers,
|
|
}
|
|
]
|
|
for test_case in test_case_data:
|
|
self.assertIs(
|
|
self.factory.make_remover(test_case["domain"]),
|
|
self.factory.remove_yt_trackers,
|
|
)
|
|
|
|
def test_remove_yt_si(self):
|
|
test_case_data = [
|
|
{
|
|
"url": "https://youtu.be/jNQXAC9IVRw?si=qLIZT1rvs99_jbgy",
|
|
"expected_url": "https://youtu.be/jNQXAC9IVRw"
|
|
},
|
|
{
|
|
"url": "https://youtu.be/jNQXAC9IVRw?si=qLIZT1rvs99_jbgy&t=16",
|
|
"expected_url": "https://youtu.be/jNQXAC9IVRw?t=16"
|
|
},
|
|
{
|
|
"url": "https://www.youtube.com/watch?v=jNQXAC9IVRw",
|
|
"expected_url": "https://www.youtube.com/watch?v=jNQXAC9IVRw"
|
|
},
|
|
{
|
|
"url": "http://www.youtube.com/watch?v=jNQXAC9IVRw&si=qLIZT1rvs99_jbgy&t=16",
|
|
"expected_url": "http://www.youtube.com/watch?v=jNQXAC9IVRw&t=16"
|
|
}
|
|
]
|
|
for test_case in test_case_data:
|
|
self.assertEqual(self.factory.remove_yt_trackers(test_case["url"]), test_case["expected_url"])
|
|
|
|
|
|
class TestRemovalMsgProcessor(unittest.TestCase):
|
|
def test_remove_links(self):
|
|
test_case_data = [
|
|
{
|
|
"msg_text": "https://youtu.be/jNQXAC9IVRw?si=qLIZT1rvs99_jbgy",
|
|
"sender_username": "Ghytro",
|
|
"bot_responded": True,
|
|
"bot_response": "Message from @Ghytro:\n\nhttps://youtu.be/jNQXAC9IVRw"
|
|
},
|
|
{
|
|
"msg_text": "чекай https://youtu.be/jNQXAC9IVRw?si=qLIZT1rvs99_jbgy\nнаш слон хд",
|
|
"sender_username": "OllyHearn",
|
|
"bot_responded": True,
|
|
"bot_response": "Message from @OllyHearn:\n\nчекай https://youtu.be/jNQXAC9IVRw\nнаш слон хд"
|
|
},
|
|
{
|
|
"msg_text": "а я такая нитакуся без si ссылки шлю сразу https://youtu.be/jNQXAC9IVRw и по нескольку штук\nhttp://www.youtube.com/watch?v=jNQXAC9IVRw&si=qLIZT1rvs99_jbgy&t=16 дада",
|
|
"sender_username": "OllyHearn",
|
|
"bot_responded": True,
|
|
"bot_response": "Message from @OllyHearn:\n\nа я такая нитакуся без si ссылки шлю сразу https://youtu.be/jNQXAC9IVRw и по нескольку штук\nhttp://www.youtube.com/watch?v=jNQXAC9IVRw&t=16 дада"
|
|
},
|
|
{
|
|
"msg_text": "asdasdasdasdasdasdasd asdasd asdasd asdad sasa dadsas",
|
|
"sender_username": "Ghytro",
|
|
"bot_responded": False,
|
|
"bot_response": ""
|
|
}
|
|
]
|
|
for test_case in test_case_data:
|
|
result = TrackerRemovalMsgProcessor(
|
|
TrackerRemovalProcessorMessage(
|
|
fromUsername=test_case["sender_username"],
|
|
text=test_case["msg_text"]
|
|
)
|
|
).process()
|
|
self.assertEqual(
|
|
result,
|
|
TrackerRemovalResult(
|
|
needsToReply=test_case["bot_responded"],
|
|
text=test_case["bot_response"]
|
|
)
|
|
)
|
|
|
|
if __name__ == "__main__":
|
|
test_classes_to_run = [TestRemoverFactory, TestRemovalMsgProcessor]
|
|
|
|
loader = unittest.TestLoader()
|
|
|
|
suites_list = []
|
|
for test_class in test_classes_to_run:
|
|
suite = loader.loadTestsFromTestCase(test_class)
|
|
suites_list.append(suite)
|
|
|
|
big_suite = unittest.TestSuite(suites_list)
|
|
|
|
runner = unittest.TextTestRunner()
|
|
results = runner.run(big_suite)
|