75 lines
2.9 KiB
Python
75 lines
2.9 KiB
Python
import unittest
|
|
|
|
from add_links.match import NoMatch, find_link_in_content
|
|
|
|
|
|
class FindLinkInContentTests(unittest.TestCase):
|
|
def test_links_first_of_two_adjacent_existing_links(self) -> None:
|
|
content = (
|
|
"'''''North Star''''' is a 1974 British [[thriller novel]] by "
|
|
"[[Hammond Innes]].<ref>Vinson & Kirkpatrick p.455</ref> "
|
|
"A man tries to prevent a plot to blow up a [[North Sea]] [[oil rig]]."
|
|
)
|
|
|
|
new_content, replacement, replaced_text = find_link_in_content(
|
|
"North Sea oil", content
|
|
)
|
|
|
|
self.assertEqual(
|
|
new_content,
|
|
"'''''North Star''''' is a 1974 British [[thriller novel]] by "
|
|
"[[Hammond Innes]].<ref>Vinson & Kirkpatrick p.455</ref> "
|
|
"A man tries to prevent a plot to blow up a "
|
|
"[[North Sea oil|North Sea]] [[oil rig]].",
|
|
)
|
|
self.assertEqual(replacement, "North Sea oil")
|
|
self.assertEqual(replaced_text, "North Sea]] [[oil")
|
|
|
|
def test_merges_existing_link_with_following_plain_text(self) -> None:
|
|
content = "[[anti-globalization]] movement"
|
|
|
|
new_content, replacement, replaced_text = find_link_in_content(
|
|
"anti-globalization movement", content
|
|
)
|
|
|
|
self.assertEqual(new_content, "[[anti-globalization movement]]")
|
|
self.assertEqual(replacement, "anti-globalization movement")
|
|
self.assertEqual(replaced_text, "anti-globalization]] movement")
|
|
|
|
def test_links_prefix_before_existing_link(self) -> None:
|
|
content = "cross-platform [[interchange station]]"
|
|
|
|
new_content, replacement, replaced_text = find_link_in_content(
|
|
"cross-platform interchange", content
|
|
)
|
|
|
|
self.assertEqual(
|
|
new_content,
|
|
"[[cross-platform interchange|cross-platform]] [[interchange station]]",
|
|
)
|
|
self.assertEqual(replacement, "cross-platform interchange")
|
|
self.assertEqual(replaced_text, "cross-platform [[interchange")
|
|
|
|
def test_lowercases_redirect_target_for_sentence_case_display_text(self) -> None:
|
|
content = (
|
|
"The absence of a voters' roll requirement was controversial, with "
|
|
"the NCA alleging potential rigging through ballot stuffing and "
|
|
"coercion in rural areas."
|
|
)
|
|
|
|
new_content, replacement, replaced_text = find_link_in_content(
|
|
"ballot stuffing", content, "Electoral fraud"
|
|
)
|
|
|
|
self.assertEqual(
|
|
new_content,
|
|
"The absence of a voters' roll requirement was controversial, with "
|
|
"the NCA alleging potential rigging through "
|
|
"[[electoral fraud|ballot stuffing]] and coercion in rural areas.",
|
|
)
|
|
self.assertEqual(replacement, "electoral fraud|ballot stuffing")
|
|
self.assertEqual(replaced_text, "ballot stuffing")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|