fix tests for new query_handler_factory format

This commit is contained in:
Job Evers-Meltzer 2016-12-30 09:07:24 -06:00
parent 7d5dd1ff51
commit 22ff87b885

View file

@ -149,12 +149,12 @@ class LbryUploader(object):
def start_server(self): def start_server(self):
session = self.session session = self.session
query_handler_factories = { query_handler_factories = {
BlobAvailabilityHandlerFactory(session.blob_manager): True, 1: BlobAvailabilityHandlerFactory(session.blob_manager),
BlobRequestHandlerFactory( 2: BlobRequestHandlerFactory(
session.blob_manager, session.wallet, session.blob_manager, session.wallet,
session.payment_rate_manager, session.payment_rate_manager,
analytics.Track()): True, analytics.Track()),
session.wallet.get_wallet_info_query_handler_factory(): True, 3: session.wallet.get_wallet_info_query_handler_factory(),
} }
server_factory = ServerProtocolFactory(session.rate_limiter, server_factory = ServerProtocolFactory(session.rate_limiter,
query_handler_factories, query_handler_factories,
@ -266,12 +266,12 @@ def start_lbry_reuploader(sd_hash, kill_event, dead_event,
server_port = None server_port = None
query_handler_factories = { query_handler_factories = {
BlobAvailabilityHandlerFactory(session.blob_manager): True, 1: BlobAvailabilityHandlerFactory(session.blob_manager),
BlobRequestHandlerFactory( 2: BlobRequestHandlerFactory(
session.blob_manager, session.wallet, session.blob_manager, session.wallet,
session.payment_rate_manager, session.payment_rate_manager,
analytics.Track()): True, analytics.Track()),
session.wallet.get_wallet_info_query_handler_factory(): True, 3: session.wallet.get_wallet_info_query_handler_factory(),
} }
server_factory = ServerProtocolFactory(session.rate_limiter, server_factory = ServerProtocolFactory(session.rate_limiter,
@ -342,12 +342,12 @@ def start_live_server(sd_hash_queue, kill_event, dead_event):
def start_listening(): def start_listening():
logging.debug("Starting the server protocol") logging.debug("Starting the server protocol")
query_handler_factories = { query_handler_factories = {
CryptBlobInfoQueryHandlerFactory(stream_info_manager, session.wallet, 1: CryptBlobInfoQueryHandlerFactory(stream_info_manager, session.wallet,
session.payment_rate_manager): True, session.payment_rate_manager),
BlobRequestHandlerFactory(session.blob_manager, session.wallet, 2: BlobRequestHandlerFactory(session.blob_manager, session.wallet,
session.payment_rate_manager, session.payment_rate_manager,
analytics.Track()): True, analytics.Track()),
session.wallet.get_wallet_info_query_handler_factory(): True, 3: session.wallet.get_wallet_info_query_handler_factory()
} }
server_factory = ServerProtocolFactory(session.rate_limiter, server_factory = ServerProtocolFactory(session.rate_limiter,
@ -483,11 +483,11 @@ def start_blob_uploader(blob_hash_queue, kill_event, dead_event, slow, is_genero
server_port = None server_port = None
query_handler_factories = { query_handler_factories = {
BlobAvailabilityHandlerFactory(session.blob_manager): True, 1: BlobAvailabilityHandlerFactory(session.blob_manager),
BlobRequestHandlerFactory(session.blob_manager, session.wallet, 2: BlobRequestHandlerFactory(session.blob_manager, session.wallet,
session.payment_rate_manager, session.payment_rate_manager,
analytics.Track()): True, analytics.Track()),
session.wallet.get_wallet_info_query_handler_factory(): True, 3: session.wallet.get_wallet_info_query_handler_factory(),
} }
server_factory = ServerProtocolFactory(session.rate_limiter, server_factory = ServerProtocolFactory(session.rate_limiter,