Updated test for RecaptchaSolver

ispras · Apr 26, 2024 · d9ed4e1 · d9ed4e1
1 parent 23cfb8d
commit d9ed4e1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/scrapypuppeteer/response.py b/scrapypuppeteer/response.py
@@ -122,7 +122,7 @@ def __init__(self, url, puppeteer_request, context_id, page_id, data, **kwargs):
         super().__init__(url, puppeteer_request, context_id, page_id, **kwargs)
 
 
-class PuppeteerRecaptchaSolverResponse(PuppeteerJsonResponse):
+class PuppeteerRecaptchaSolverResponse(PuppeteerJsonResponse):  # TODO: to inherit from HTMLResponse (or think about better solution to the problem!)
     """
     Response for RecaptchaSolver.
     Result is available via self.recaptcha_data and self.data["recaptcha_data"]

diff --git a/tests/mockserver.py b/tests/mockserver.py
@@ -137,9 +137,14 @@ def _form_response(self, page_id, context_id, close_page):
 
 class RecaptchaSolver(LeafResource):
     def _form_response(self, page_id, context_id, close_page):
+        html = """
+            <html> <head></head> <body>clicked</body>
+        """
         return {
             "contextId": context_id,
             "pageId": page_id,
+            "html": html,
+            "cookies": None,
             "recaptcha_data": {
                 "captchas": [1],  # 1 captcha
                 "some_other_fields": [],