diff --git a/tests/providers/conftest.py b/tests/providers/conftest.py index df7bc38..949636c 100644 --- a/tests/providers/conftest.py +++ b/tests/providers/conftest.py @@ -3,7 +3,7 @@ import pytest from gevent import socket from web3.providers.ipc import IPCProvider -from web3.providers.rpc import TestRPCProvider, RPCProvider +from web3.providers.rpc import TestRPCProvider, RPCProvider, KeepAliveRPCProvider def get_open_port(): @@ -15,7 +15,7 @@ def get_open_port(): return port -@pytest.fixture(params=['tester', 'rpc', 'ipc']) +@pytest.fixture(params=['tester', 'rpc', 'ipc', 'keep-alive-rpc']) def disconnected_provider(request): """ Supply a Provider that's not connected to a node. @@ -31,6 +31,8 @@ def disconnected_provider(request): return provider elif request.param == 'rpc': return RPCProvider(port=9999) + elif request.param == 'rpc': + return KeepAliveRPCProvider(port=9999) elif request.param == 'ipc': return IPCProvider(ipc_path='nonexistent') else: diff --git a/web3/__init__.py b/web3/__init__.py index 0acc5d5..622f1b7 100644 --- a/web3/__init__.py +++ b/web3/__init__.py @@ -18,5 +18,6 @@ __all__ = [ "Web3", "RPCProvider", "TestRPCProvider", + "KeepAliveRPCProvider", "IPCProvider", ]