fix(core): retry query_selector on stale document ID (#235)

Handle `ProtocolException: Could not find node with given id` in
`query_selector` and `query_selector_all` when operating on the default
document. This ensures the method re-fetches the document and retries
exactly once instead of raising or incorrectly returning None.

Closes #234

Author: Avejack

CHANGELOG.md

@@ -9,6 +9,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Fixed
 
+- Fix race condition in `Tab.query_selector` and `Tab.query_selector_all` on stale document. @Avejack
+
 ### Added
 
 ### Changed

zendriver/core/tab.py

@@ -469,6 +469,13 @@ async def query_selector_all(
                     setattr(_node, "__last", True)
                     return await self.query_selector_all(selector, _node)
             else:
+                if e.message is not None and "could not find node" in e.message.lower():
+                    # The document node is stale; refetch and retry once
+                    doc = await self.send(cdp.dom.get_document(-1, True))
+                    # Prevent double-retry by marking this node as 'last attempt'
+                    setattr(doc, "__last", True)
+                    return await self.query_selector_all(selector, doc)
+
                 await self.disable_dom_agent()
                 raise
         if not node_ids:
@@ -531,7 +538,11 @@ async def query_selector(
                 and "could not find node" in e.message.lower()
                 and doc
             ):
-                return None
+                # The document node is stale; refetch and retry once
+                doc = await self.send(cdp.dom.get_document(-1, True))
+                # Prevent double-retry by marking this node as 'last attempt'
+                setattr(doc, "__last", True)
+                return await self.query_selector(selector, doc)
             else:
                 await self.disable_dom_agent()
                 raise