class MockRaidenService:
def __init__(self, message_handler=None, state_transition=None):
self.chain = MockChain()
self.private_key, self.address = factories.make_privatekey_address()
self.chain.node_address = self.address
self.message_handler = message_handler
if state_transition is None:
state_transition = node.state_transition
serializer = JSONSerializer
state_manager = StateManager(state_transition, None)
storage = SQLiteStorage(':memory:', serializer)
self.wal = WriteAheadLog(state_manager, storage)
state_change = ActionInitChain(
random.Random(),
0,
self.chain.node_address,
self.chain.network_id,
)
self.wal.log_and_dispatch(state_change)
def on_message(self, message):
if self.message_handler:
self.message_handler.on_message(self, message)
def handle_state_change(self, state_change):
pass
def sign(self, message):
message.sign(self.private_key)
class MockRaidenService:
def __init__(self,
message_handler=None,
state_transition=None,
private_key=None,
config=None):
if private_key is None:
self.privkey, self.address = factories.make_privkey_address()
else:
self.privkey = private_key
self.address = privatekey_to_address(private_key)
self.rpc_client = MockJSONRPCClient(self.address)
self.proxy_manager = MockProxyManager(node_address=self.address)
self.signer = LocalSigner(self.privkey)
self.message_handler = message_handler
self.routing_mode = RoutingMode.PRIVATE
self.config = config
self.user_deposit = Mock()
self.default_registry = Mock()
self.default_registry.address = factories.make_address()
self.default_one_to_n_address = factories.make_address()
self.default_msc_address = factories.make_address()
self.targets_to_identifiers_to_statuses = defaultdict(dict)
self.route_to_feedback_token = {}
if state_transition is None:
state_transition = node.state_transition
serializer = JSONSerializer
state_manager = StateManager(state_transition, None)
storage = SerializedSQLiteStorage(":memory:", serializer)
self.wal = WriteAheadLog(state_manager, storage)
state_change = ActionInitChain(
pseudo_random_generator=random.Random(),
block_number=0,
block_hash=factories.make_block_hash(),
our_address=self.rpc_client.address,
chain_id=self.rpc_client.chain_id,
)
self.wal.log_and_dispatch([state_change])
def on_message(self, message):
if self.message_handler:
self.message_handler.on_message(self, message)
def handle_and_track_state_changes(self, state_changes):
pass
def handle_state_changes(self, state_changes):
pass
def sign(self, message):
message.sign(self.signer)
class MockRaidenService:
def __init__(self,
message_handler=None,
state_transition=None,
private_key=None):
if private_key is None:
self.private_key, self.address = factories.make_privkey_address()
else:
self.private_key = private_key
self.address = privatekey_to_address(private_key)
self.chain = MockChain(network_id=17, node_address=self.address)
self.signer = LocalSigner(self.private_key)
self.message_handler = message_handler
self.user_deposit = Mock()
if state_transition is None:
state_transition = node.state_transition
serializer = JSONSerializer
state_manager = StateManager(state_transition, None)
storage = SerializedSQLiteStorage(":memory:", serializer)
self.wal = WriteAheadLog(state_manager, storage)
state_change = ActionInitChain(
pseudo_random_generator=random.Random(),
block_number=0,
block_hash=factories.make_block_hash(),
our_address=self.chain.node_address,
chain_id=self.chain.network_id,
)
self.wal.log_and_dispatch(state_change)
def on_message(self, message):
if self.message_handler:
self.message_handler.on_message(self, message)
def handle_and_track_state_change(self, state_change):
pass
def handle_state_change(self, state_change):
pass
def sign(self, message):
message.sign(self.signer)