Auto Provider batch requests (#3607)

* Get active provider from AutoProvider

* Add docs

* Add newsfragment

Author: kclowes

docs/web3.main.rst

@@ -72,6 +72,12 @@ Batch Requests
             responses = batch.execute()
             assert len(responses) == 3
 
+    .. note::
+
+       Within the batching context above, calls are suspended until
+       ``batch.execute()`` is called. Calling a method without
+       passing it to ``batch.add`` might result in unexpected behavior.
+
     Using the batch object directly:
 
     .. code-block:: python

newsfragments/3607.bugfix.rst

@@ -0,0 +1 @@
+Batching can now be used with the AutoProvider

tests/integration/go_ethereum/test_goethereum_http.py

@@ -76,6 +76,17 @@ def w3(geth_process, endpoint_uri):
     return Web3(Web3.HTTPProvider(endpoint_uri, request_kwargs={"timeout": 10}))
 
 
+@pytest.fixture(scope="module")
+def auto_w3(geth_process, endpoint_uri):
+    wait_for_http(endpoint_uri)
+
+    from web3.auto import (
+        w3,
+    )
+
+    return w3
+
+
 class TestGoEthereumWeb3ModuleTest(GoEthereumWeb3ModuleTest):
     pass
 
@@ -105,7 +116,15 @@ class TestGoEthereumDebugModuleTest(GoEthereumDebugModuleTest):
 
 
 class TestGoEthereumEthModuleTest(GoEthereumEthModuleTest):
-    pass
+    def test_auto_provider_batching(
+        self,
+        auto_w3: "Web3",
+        monkeypatch,
+        endpoint_uri,
+    ) -> None:
+        monkeypatch.setenv("WEB3_PROVIDER_URI", endpoint_uri)
+        # test that batch_requests doesn't error out when using the auto provider
+        auto_w3.batch_requests()
 
 
 class TestGoEthereumNetModuleTest(GoEthereumNetModuleTest):

tests/integration/go_ethereum/test_goethereum_ipc.py

@@ -53,6 +53,17 @@ def geth_ipc_path(datadir):
         os.remove(_geth_ipc_path)
 
 
+@pytest.fixture(scope="module")
+def auto_w3(geth_process, geth_ipc_path):
+    wait_for_socket(geth_ipc_path)
+
+    from web3.auto import (
+        w3,
+    )
+
+    return w3
+
+
 @pytest.fixture(scope="module")
 def w3(geth_process, geth_ipc_path):
     wait_for_socket(geth_ipc_path)
@@ -68,7 +79,15 @@ class TestGoEthereumDebugModuleTest(GoEthereumDebugModuleTest):
 
 
 class TestGoEthereumEthModuleTest(GoEthereumEthModuleTest):
-    pass
+    def test_auto_provider_batching(
+        self,
+        auto_w3: "Web3",
+        monkeypatch,
+        geth_ipc_path,
+    ) -> None:
+        monkeypatch.setenv("WEB3_PROVIDER_URI", f"file:///{geth_ipc_path}")
+        # test that batch_requests doesn't error out when using the auto provider
+        auto_w3.batch_requests()
 
 
 class TestGoEthereumNetModuleTest(GoEthereumNetModuleTest):

web3/manager.py

@@ -447,6 +447,8 @@ def _batch_requests(self) -> RequestBatcher[Method[Callable[..., Any]]]:
         """
         Context manager for making batch requests
         """
+        if isinstance(self.provider, AutoProvider):
+            self.provider = self.provider._get_active_provider(use_cache=True)
         if not isinstance(self.provider, (AsyncJSONBaseProvider, JSONBaseProvider)):
             raise Web3TypeError("Batch requests are not supported by this provider.")
         return RequestBatcher(self.w3)