jenkins-bot submitted this change.

View Change

[tests] Add harvest_templates_tests

Change-Id: I74905df161e4a6ea18c8a0f023df680443de64ff
---
A docs/tests_ref/harvest_templates_tests.rst
M docs/tests_ref/index.rst
M tests/__init__.py
A tests/harvest_templates_tests.py
4 files changed, 63 insertions(+), 0 deletions(-)

diff --git a/docs/tests_ref/harvest_templates_tests.rst b/docs/tests_ref/harvest_templates_tests.rst
new file mode 100644
index 0000000..873c7a3
--- /dev/null
+++ b/docs/tests_ref/harvest_templates_tests.rst
@@ -0,0 +1,7 @@
+tests.harvest\_templates\_tests module
+======================================
+
+.. automodule:: tests.harvest_templates_tests
+    :members:
+    :undoc-members:
+    :show-inheritance:
diff --git a/docs/tests_ref/index.rst b/docs/tests_ref/index.rst
index 695a506..457b2c5 100644
--- a/docs/tests_ref/index.rst
+++ b/docs/tests_ref/index.rst
@@ -102,6 +102,7 @@
     fixing_redirects<./fixing_redirects_tests>
     generate_family_file<./generate_family_file_tests>
     generate_user_files<./generate_user_files_tests>
+    harvest_templates<./harvest_templates_tests>
     interwikidata<./interwikidata_tests>
     noreferences<./noreferences_tests>
     patrolbot<./patrolbot_tests>
diff --git a/tests/__init__.py b/tests/__init__.py
index 28835bf..854a029 100644
--- a/tests/__init__.py
+++ b/tests/__init__.py
@@ -148,6 +148,7 @@
     'fixing_redirects',
     'generate_family_file',
     'generate_user_files',
+    'harvest_templates',
     'interwikidata',
     'l10n',
     'patrolbot',
diff --git a/tests/harvest_templates_tests.py b/tests/harvest_templates_tests.py
new file mode 100644
index 0000000..b0804ee
--- /dev/null
+++ b/tests/harvest_templates_tests.py
@@ -0,0 +1,54 @@
+#!/usr/bin/python3
+"""Tests for scripts/harvest_template.py."""
+#
+# (C) Pywikibot team, 2022
+#
+# Distributed under the terms of the MIT license.
+#
+import unittest
+from contextlib import suppress
+
+from pywikibot import ItemPage
+from scripts.harvest_template import HarvestRobot
+
+from tests.aspects import ScriptMainTestCase
+
+
+class DummyBot:
+
+    """Dummy HarvestRobot."""
+
+    class current_page:  # noqa: D106, N801
+        pass
+
+    def __init__(self, site):  # noqa: D107
+        """Add a site object to current_page attribute."""
+        self.current_page.site = site
+
+
+class TestHarvestRobot(ScriptMainTestCase):
+
+    """Test HarvestRobot."""
+
+    family = 'wikipedia'
+    code = 'cs'
+
+    def test_template_link_target(self):
+        """Test template_link_target static method."""
+        tests = [
+            ('Pes', 'Q144'),
+            ('Imaginární číslo', 'Q9165172'),
+            ('Sequana', 'Q472766'),
+        ]
+        for link, item in tests:
+            with self.subTest(link=link, item=item):
+                dummy_item = ItemPage(self.site.data_repository(), 'Q1')
+                target = HarvestRobot.template_link_target(
+                    DummyBot(self.site), dummy_item, link)
+                self.assertIsInstance(target, ItemPage)
+                self.assertEqual(target.title(), item)
+
+
+if __name__ == '__main__':  # pragma: no cover
+    with suppress(SystemExit):
+        unittest.main()

To view, visit change 811911. To unsubscribe, or for help writing mail filters, visit settings.