Added test_page_get_query_continue for #129

Author: siznax

tests/query_plcontinue.py

@@ -0,0 +1,38 @@
+# -*- coding:utf-8 -*-
+
+query = 'https://en.wikipedia.org/w/api.php?action=query&formatversion=2&pllimit=500&prop=links&list=random&titles=List_of_programming_languages'
+
+response = r"""
+{
+    "continue": {
+        "plcontinue": "144146|0|Perl_Data_Language",
+        "continue": "||"
+    },
+    "query": {
+        "pages": [
+            {
+                "pageid": 144146,
+                "ns": 0,
+                "title": "List of programming languages",
+                "links": [
+                    {
+                        "ns": 0,
+                        "title": "A++"
+                    }
+                ]
+            }
+        ],
+        "random": [
+            {
+                "id": 123,
+                "ns": 0,
+                "title": "RANDOM"
+            }
+        ]
+    }
+}
+"""
+
+cache = {'query': query,
+         'response': response,
+         'info': {'content-type': 'TEST', 'status': 200}}

tests/test_basic.py

@@ -23,6 +23,7 @@
 from . import parse_91
 from . import parse_109
 from . import query
+from . import query_plcontinue
 from . import querymore
 from . import querymore_blcontinue
 from . import random_query
@@ -500,8 +501,25 @@ def test_page_get_more(self):
 
         self.assertTrue('requests' not in page.data)
 
+    def test_page_get_query_continue(self):
+        page = wptools.page('TEST', skip=SKIP_FLAG, silent=SILENT_FLAG)
+
+        page.cache['query'] = query.cache
+        page._set_data('query')
+        self.assertEqual(len(page.data['links']), 384)
+
+        page.cache = {'query': query_plcontinue.cache}
+        page._set_data('query')
+        self.assertTrue('continue' in page.data)
+        self.assertEqual(page.data['continue'],
+                         {'plcontinue': u'144146|0|Perl_Data_Language'})
+        self.assertEqual(len(page.data['links']), 385)
+
+        qry = page._query('query', wptools.query.WPToolsQuery())
+        self.assertTrue(qry.endswith('&plcontinue=144146|0|Perl_Data_Language'))
+
     def test_page_get_more_continue(self):
-        page = wptools.page('TEST', silent=SILENT_FLAG)
+        page = wptools.page('TEST', skip=SKIP_FLAG, silent=SILENT_FLAG)
 
         page.cache['querymore'] = querymore.cache
         page._set_data('querymore')