def test_SERVICES():
setup_base_env()
e = Env()
assert e.services == []
# This has a blank entry between commas
os.environ['SERVICES'] = 'tcp://foo.bar:1234,,ws://1.2.3.4:567,rpc://[::1]:700'
e = Env()
assert e.services == [
Service('tcp', NetAddress('foo.bar', 1234)),
Service('ws', NetAddress('1.2.3.4', 567)),
Service('rpc', NetAddress('::1', 700)),
]
def test_EVENT_LOOP_POLICY():
e = Env()
assert e.loop_policy is None
os.environ['EVENT_LOOP_POLICY'] = 'foo'
with pytest.raises(Env.Error):
Env()
os.environ['EVENT_LOOP_POLICY'] = 'uvloop'
try:
Env()
except ImportError:
pass
del os.environ['EVENT_LOOP_POLICY']
def test_BANNER_FILE():
e = Env()
assert e.banner_file is None
assert e.tor_banner_file is None
os.environ['BANNER_FILE'] = 'banner_file'
e = Env()
assert e.banner_file == 'banner_file'
assert e.tor_banner_file == 'banner_file'
os.environ['TOR_BANNER_FILE'] = 'tor_banner_file'
e = Env()
assert e.banner_file == 'banner_file'
assert e.tor_banner_file == 'tor_banner_file'
def test_PEER_DISCOVERY():
e = Env()
assert e.peer_discovery == Env.PD_ON
os.environ['PEER_DISCOVERY'] = ' '
e = Env()
assert e.peer_discovery == Env.PD_OFF
os.environ['PEER_DISCOVERY'] = 'ON'
e = Env()
assert e.peer_discovery == Env.PD_ON
os.environ['PEER_DISCOVERY'] = 'self'
e = Env()
assert e.peer_discovery == Env.PD_SELF
def assert_integer(env_var, attr, default=''):
setup_base_env()
if default != '':
e = Env()
assert getattr(e, attr) == default
value = random.randrange(5, 2000)
os.environ[env_var] = str(value) + '.1'
with pytest.raises(Env.Error):
Env()
os.environ[env_var] = str(value)
e = Env()
assert getattr(e, attr) == value
def main():
'''Set up logging and run the server.'''
log_fmt = Env.default('LOG_FORMAT', '%(levelname)s:%(name)s:%(message)s')
logging.basicConfig(level=logging.INFO, format=log_fmt)
logging.info('LbryumX server starting')
try:
controller = Controller(Env(LBC))
controller.run()
except Exception:
traceback.print_exc()
logging.critical('LbryumX server terminated abnormally')
else:
logging.info('LbryumX server terminated normally')
def target():
loop = self.loop
asyncio.set_event_loop(loop)
env = Env()
env.loop_policy = asyncio.get_event_loop_policy()
self.controller = Controller(env)
logging.basicConfig(level=logging.DEBUG)
loop.run_until_complete(
asyncio.wait(
[self.controller.serve(self.evt),
self.evt.wait()],
return_when=asyncio.FIRST_COMPLETED))
loop.close()
def test_SERVICES_default_rpc():
# This has a blank entry between commas
os.environ['SERVICES'] = 'rpc://foo.bar'
e = Env()
assert e.services[0].host == 'foo.bar'
assert e.services[0].port == 8000
os.environ['SERVICES'] = 'rpc://:800'
e = Env()
assert e.services[0].host == 'localhost'
assert e.services[0].port == 800
os.environ['SERVICES'] = 'rpc://'
e = Env()
assert e.services[0].host == 'localhost'
assert e.services[0].port == 8000
def test_SSL_PORT():
# Requires both SSL_CERTFILE and SSL_KEYFILE to be set
os.environ['SSL_PORT'] = '50002'
os.environ['SSL_CERTFILE'] = 'certfile'
with pytest.raises(Env.Error):
Env()
os.environ.pop('SSL_CERTFILE')
os.environ['SSL_KEYFILE'] = 'keyfile'
with pytest.raises(Env.Error):
Env()
os.environ['SSL_CERTFILE'] = 'certfile'
Env()
os.environ.pop('SSL_PORT')
assert_integer('SSL_PORT', 'ssl_port', None)
def test_COIN_NET():
'''Test COIN and NET defaults and redirection.'''
setup_base_env()
e = Env()
assert e.coin == lib_coins.BitcoinCash
os.environ['NET'] = 'testnet'
e = Env()
assert e.coin == lib_coins.BitcoinCashTestnet
os.environ['NET'] = ' testnet '
e = Env()
assert e.coin == lib_coins.BitcoinCashTestnet
os.environ.pop('NET')
os.environ['COIN'] = ' Litecoin '
e = Env()
assert e.coin == lib_coins.Litecoin
os.environ['NET'] = 'testnet'
e = Env()
assert e.coin == lib_coins.LitecoinTestnet
os.environ.pop('NET')
os.environ['COIN'] = ' BitcoinGold '
e = Env()
assert e.coin == lib_coins.BitcoinGold
os.environ['NET'] = 'testnet'
e = Env()
assert e.coin == lib_coins.BitcoinGoldTestnet
os.environ['NET'] = 'regtest'
e = Env()
assert e.coin == lib_coins.BitcoinGoldRegtest
def test_HOST():
assert_default('HOST', 'host', 'localhost')
os.environ['HOST'] = ''
e = Env()
assert e.cs_host(for_rpc=False) == ''
os.environ['HOST'] = '192.168.0.1,23.45.67.89'
e = Env()
assert e.cs_host(for_rpc=False) == ['192.168.0.1', '23.45.67.89']
os.environ['HOST'] = '192.168.0.1 , 23.45.67.89 '
e = Env()
assert e.cs_host(for_rpc=False) == ['192.168.0.1', '23.45.67.89']
def setup_session(data_dir, rpc_port):
conf = {'DB_DIRECTORY': data_dir,
'DAEMON_URL': 'http://*****:*****@localhost:{}/'.format(rpc_port)}
os.environ.update(conf)
controller = Controller(Env(LBCRegTest))
session = LBC.SESSIONCLS(controller, 'TCP')
return session
def test_ssl_SERVICES(service):
setup_base_env()
os.environ['SERVICES'] = service
with pytest.raises(Env.Error) as err:
Env()
assert 'SSL_CERTFILE' in str(err.value)
os.environ['SSL_CERTFILE'] = 'certfile'
with pytest.raises(Env.Error) as err:
Env()
assert 'SSL_KEYFILE' in str(err.value)
os.environ['SSL_KEYFILE'] = 'keyfile'
Env()
setup_base_env()
os.environ['SERVICES'] = service
os.environ['SSL_KEYFILE'] = 'keyfile'
with pytest.raises(Env.Error) as err:
Env()
assert 'SSL_CERTFILE' in str(err.value)
def block_processor(tmpdir_factory):
environ.clear()
environ['DB_DIRECTORY'] = tmpdir_factory.mktemp('db',
numbered=True).strpath
environ['DAEMON_URL'] = ''
env = Env(LBC)
bp = LBC.BLOCK_PROCESSOR(env, None, None)
yield bp
for attr in dir(bp): # hack to close dbs on tear down
obj = getattr(bp, attr)
if isinstance(obj, Storage):
obj.close()
async def start(self):
self.data_path = tempfile.mkdtemp()
conf = {
'DB_DIRECTORY': self.data_path,
'DAEMON_URL': 'http://*****:*****@localhost:50001/',
'REORG_LIMIT': '100',
'TCP_PORT': '1984'
}
os.environ.update(conf)
self.controller = Controller(Env(self.coin_class))
self.controller.start_time = time.time()
await self.controller.start_servers()
await self.controller.tcp_server_started.wait()
def run_test(db_dir):
environ.clear()
environ['DB_DIRECTORY'] = db_dir
environ['DAEMON_URL'] = ''
environ['COIN'] = 'BitcoinCash'
env = Env()
history = DB(env).history
# Test abstract compaction
check_hashX_compaction(history)
# Now test in with random data
histories = create_histories(history)
check_written(history, histories)
compact_history(history)
check_written(history, histories)
def test_RPC_HOST():
assert_default('RPC_HOST', 'rpc_host', 'localhost')
os.environ['RPC_HOST'] = ''
e = Env()
# Blank reverts to localhost
assert e.cs_host(for_rpc=True) == 'localhost'
os.environ['RPC_HOST'] = '127.0.0.1, ::1'
e = Env()
assert e.cs_host(for_rpc=True) == ['127.0.0.1', '::1']
async def run_test(db_dir):
environ.clear()
environ['DB_DIRECTORY'] = db_dir
environ['DAEMON_URL'] = ''
environ['COIN'] = 'Bitcoin'
db = DB(Env())
await db.open_for_serving()
history = db.history
# Test abstract compaction
check_hashX_compaction(history)
# Now test in with random data
histories = create_histories(history)
check_written(history, histories)
compact_history(history)
check_written(history, histories)
def test_tor_identity():
tor_host = 'something.onion'
os.environ.pop('REPORT_HOST', None)
os.environ.pop('REPORT_HOST_TOR', None)
e = Env()
assert len(e.identities) == 0
os.environ['REPORT_HOST_TOR'] = 'foo'
os.environ['REPORT_SSL_PORT_TOR'] = '123'
os.environ['TCP_PORT'] = '456'
with pytest.raises(Env.Error):
Env()
os.environ['REPORT_HOST_TOR'] = tor_host
e = Env()
assert len(e.identities) == 1
ident = e.identities[0]
assert ident.host == tor_host
assert ident.tcp_port == 456
assert ident.ssl_port == 123
assert ident.nick_suffix == '_tor'
os.environ['REPORT_TCP_PORT_TOR'] = os.environ['REPORT_SSL_PORT_TOR']
with pytest.raises(Env.Error):
Env()
os.environ['REPORT_HOST'] = 'foo.com'
os.environ['TCP_PORT'] = '456'
os.environ['SSL_PORT'] = '789'
os.environ['REPORT_TCP_PORT'] = '654'
os.environ['REPORT_SSL_PORT'] = '987'
os.environ['SSL_CERTFILE'] = 'certfile'
os.environ['SSL_KEYFILE'] = 'keyfile'
os.environ.pop('REPORT_TCP_PORT_TOR', None)
os.environ.pop('REPORT_SSL_PORT_TOR', None)
e = Env()
assert len(e.identities) == 2
ident = e.identities[1]
assert ident.host == tor_host
assert ident.tcp_port == 654
assert ident.ssl_port == 987
os.environ['REPORT_TCP_PORT_TOR'] = '234'
os.environ['REPORT_SSL_PORT_TOR'] = '432'
e = Env()
assert len(e.identities) == 2
ident = e.identities[1]
assert ident.host == tor_host
assert ident.tcp_port == 234
assert ident.ssl_port == 432
def test_REPORT_SERVICES_rpc():
setup_base_env()
os.environ['REPORT_SERVICES'] = 'rpc://foo.bar:1234'
with pytest.raises(ServiceError) as err:
Env()
assert 'bad protocol' in str(err.value)
def test_DAEMON_URL():
assert_required('DAEMON_URL')
setup_base_env()
e = Env()
assert e.daemon_url == BASE_DAEMON_URL
def test_DB_DIRECTORY():
assert_required('DB_DIRECTORY')
setup_base_env()
e = Env()
assert e.db_dir == BASE_DB_DIR
def test_minimal():
setup_base_env()
Env()
def test_coin_class_provided():
e = Env(lib_coins.BitcoinCash)
assert e.coin == lib_coins.BitcoinCash
def test_clearnet_identity():
os.environ['REPORT_TCP_PORT'] = '456'
e = Env()
assert len(e.identities) == 0
os.environ['REPORT_HOST'] = '8.8.8.8'
e = Env()
assert len(e.identities) == 1
assert e.identities[0].host == '8.8.8.8'
os.environ['REPORT_HOST'] = 'localhost'
with pytest.raises(Env.Error):
Env()
os.environ['REPORT_HOST'] = ''
with pytest.raises(Env.Error):
Env()
os.environ['REPORT_HOST'] = '127.0.0.1'
with pytest.raises(Env.Error):
Env()
os.environ['REPORT_HOST'] = '0.0.0.0'
with pytest.raises(Env.Error):
Env()
os.environ['REPORT_HOST'] = '224.0.0.2'
with pytest.raises(Env.Error):
Env()
os.environ['REPORT_HOST'] = '$HOST'
with pytest.raises(Env.Error):
Env()
# Accept private IP, unless PEER_ANNOUNCE
os.environ['PEER_ANNOUNCE'] = ''
os.environ['REPORT_HOST'] = '192.168.0.1'
os.environ['SSL_CERTFILE'] = 'certfile'
os.environ['SSL_KEYFILE'] = 'keyfile'
Env()
os.environ['PEER_ANNOUNCE'] = 'OK'
with pytest.raises(Env.Error) as err:
Env()
os.environ.pop('PEER_ANNOUNCE', None)
assert 'not a valid REPORT_HOST' in str(err)
os.environ['REPORT_HOST'] = '1.2.3.4'
os.environ['REPORT_SSL_PORT'] = os.environ['REPORT_TCP_PORT']
with pytest.raises(Env.Error) as err:
Env()
assert 'both resolve' in str(err)
os.environ['REPORT_SSL_PORT'] = '457'
os.environ['REPORT_HOST'] = 'foo.com'
e = Env()
assert len(e.identities) == 1
ident = e.identities[0]
assert ident.host == 'foo.com'
assert ident.tcp_port == 456
assert ident.ssl_port == 457
assert ident.nick_suffix == ''
def assert_required(env_var):
setup_base_env()
os.environ.pop(env_var, None)
with pytest.raises(Env.Error):
Env()
def test_MAX_SESSIONS():
too_big = 10000000
os.environ['MAX_SESSIONS'] = str(too_big)
e = Env()
assert e.max_sessions < too_big
def test_COIN_NET():
'''Test COIN and NET defaults and redirection.'''
setup_base_env()
e = Env()
assert e.coin == lib_coins.BitcoinSV
os.environ['NET'] = 'testnet'
e = Env()
assert e.coin == lib_coins.BitcoinSVTestnet
os.environ['NET'] = ' testnet '
e = Env()
assert e.coin == lib_coins.BitcoinSVTestnet
os.environ.pop('NET')
os.environ['COIN'] = ' Litecoin '
e = Env()
assert e.coin == lib_coins.Litecoin
os.environ['NET'] = 'testnet'
e = Env()
assert e.coin == lib_coins.LitecoinTestnet
os.environ.pop('NET')
os.environ['COIN'] = ' BitcoinGold '
e = Env()
assert e.coin == lib_coins.BitcoinGold
os.environ['NET'] = 'testnet'
e = Env()
assert e.coin == lib_coins.BitcoinGoldTestnet
os.environ['NET'] = 'regtest'
e = Env()
assert e.coin == lib_coins.BitcoinGoldRegtest
os.environ.pop('NET')
os.environ['COIN'] = ' Decred '
e = Env()
assert e.coin == lib_coins.Decred
os.environ['NET'] = 'testnet'
e = Env()
assert e.coin == lib_coins.DecredTestnet
os.environ.pop('NET')
os.environ['COIN'] = ' BitcoinGreen '
e = Env()
assert e.coin == lib_coins.Bitg
os.environ['NET'] = 'mainnet'
e = Env()
os.environ.pop('NET')
os.environ['COIN'] = ' Pivx '
os.environ['NET'] = 'mainnet'
e = Env()
assert e.coin == lib_coins.Pivx
os.environ['NET'] = 'testnet'
e = Env()
assert e.coin == lib_coins.PivxTestnet
os.environ.pop('NET')
os.environ['NET'] = 'mainnet'
os.environ['COIN'] = ' TokenPay '
e = Env()
assert e.coin == lib_coins.TokenPay
def test_REPORT_SERVICES_localhost():
setup_base_env()
os.environ['REPORT_SERVICES'] = 'tcp://localhost:1234'
with pytest.raises(ServiceError) as err:
Env()
assert 'bad host' in str(err.value)
def test_onion_SERVICES():
setup_base_env()
os.environ['SERVICES'] = 'tcp://foo.bar.onion:1234'
with pytest.raises(ServiceError) as err:
Env()
assert 'bad host' in str(err.value)
