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)
|