1
0

constants: rm set_testnet/set_mainnet fns with AbstractNet.set_as_network

This commit is contained in:
SomberNight
2024-09-16 15:27:45 +00:00
parent 778aea340e
commit 1257f21b1b
11 changed files with 28 additions and 47 deletions

View File

@@ -41,13 +41,13 @@ class ElectrumTestCase(unittest.IsolatedAsyncioTestCase, Logger):
def setUpClass(cls):
super().setUpClass()
if cls.TESTNET:
constants.set_testnet()
constants.BitcoinTestnet.set_as_network()
@classmethod
def tearDownClass(cls):
super().tearDownClass()
if cls.TESTNET:
constants.set_mainnet()
constants.BitcoinMainnet.set_as_network()
def setUp(self):
self._test_lock.acquire()
@@ -79,14 +79,14 @@ def as_testnet(func):
if asyncio.iscoroutinefunction(func):
async def run_test(*args, **kwargs):
try:
constants.set_testnet()
constants.BitcoinTestnet.set_as_network()
return await func(*args, **kwargs)
finally:
constants.net = old_net
else:
def run_test(*args, **kwargs):
try:
constants.set_testnet()
constants.BitcoinTestnet.set_as_network()
return func(*args, **kwargs)
finally:
constants.net = old_net