add unit tests for deleted revs & undelete

Author: fastily

requirements.txt

@@ -1 +1 @@
-requests==2.27.1
+requests==2.28.0

tests/res/deleted-revisions.json

@@ -0,0 +1,38 @@
+{
+    "batchcomplete": true,
+    "query": {
+        "pages": [
+            {
+                "ns": 2,
+                "title": "User:Fastily/SomePageThatWasDeleted",
+                "missing": true,
+                "deletedrevisions": [
+                    {
+                        "revid": 1094208014,
+                        "parentid": 1094207980,
+                        "minor": false,
+                        "user": "Fastily",
+                        "timestamp": "2022-06-21T09:47:20Z",
+                        "comment": "test again"
+                    },
+                    {
+                        "revid": 1094207980,
+                        "parentid": 1094207022,
+                        "minor": false,
+                        "user": "Fastily",
+                        "timestamp": "2022-06-21T09:47:07Z",
+                        "comment": "test 1"
+                    },
+                    {
+                        "revid": 1094207022,
+                        "parentid": 0,
+                        "minor": false,
+                        "user": "Fastily",
+                        "timestamp": "2022-06-21T09:38:57Z",
+                        "comment": "test"
+                    }
+                ]
+            }
+        ]
+    }
+}

tests/res/undelete.json

@@ -0,0 +1,8 @@
+{
+    "undelete": {
+        "title": "User:Fastily",
+        "revisions": 1,
+        "fileversions": 0,
+        "reason": "testing undeletion"
+    }
+}

tests/test_gquery.py

@@ -1,11 +1,12 @@
 from datetime import datetime, timezone
+from unittest.mock import Mock, patch
 
 from pwiki.gquery import GQuery
 from pwiki.ns import NS
 from pwiki.query_constants import MAX
 from pwiki.query_utils import flatten_generator
 
-from .base import WikiTestCase
+from .base import file_to_json, WikiTestCase
 
 
 class TestListCont(WikiTestCase):
@@ -116,6 +117,16 @@ class TestPropCont(WikiTestCase):
     def test_categories_on_page(self):
         self.assertCountEqual(["Category:Fastily Test", "Category:Fastily Test2"], next(GQuery.categories_on_page(self.wiki, "User:Fastily/Sandbox/Page/2", MAX)))
 
+    @patch("pwiki.query_utils.basic_query", return_value=file_to_json("deleted-revisions"))
+    def test_deleted_revisions(self, mock: Mock):
+        result = next(GQuery.deleted_revisions(self.wiki, "User:Fastily/SomePageThatWasDeleted"))
+
+        mock.assert_called_once()
+        self.assertEqual(3, len(result))
+        self.assertEqual("Fastily", result[0].user)
+        self.assertEqual("test 1", result[1].summary)
+        self.assertEqual(datetime(2022, 6, 21, 9, 38, 57, tzinfo=timezone.utc), result[2].timestamp)
+
     def test_revisions(self):
         # test 1 - base
         result = next(g := GQuery.revisions(self.wiki, "User:Fastily/Sandbox/RevisionTest", 2))

tests/test_wiki.py

@@ -27,6 +27,11 @@ def test_edit(self, mock: mock.Mock):
         with self.assertRaises(ValueError):
             self.wiki.edit("Foo")
 
+    def test_undelete(self, mock: mock.Mock):
+        mock.return_value = file_to_json("undelete")
+        self.assertTrue(self.wiki.undelete("User:Fastily", "testing undeletion"))
+        mock.assert_called_once()
+
 
 class TestReadOnlyWikiAction(WikiTestCase):
     """Tests Wiki actions which perform invisible/read-only updates the target wiki."""