def mock_matrix(
monkeypatch,
retry_interval,
retries_before_backoff,
local_matrix_server,
private_rooms,
):
from matrix_client.user import User
monkeypatch.setattr(User, 'get_display_name', lambda _: 'random_display_name')
def mock_get_user(klass, user: Union[User, str]) -> User:
return User(None, USERID1)
def mock_get_room_ids_for_address(
klass,
address: Address,
filter_private: bool=None,
) -> List[str]:
return ['!roomID:server']
def mock_set_room_id_for_address(self, address: Address, room_id: Optional[str]):
pass
def mock_receive_message(klass, message):
# We are just unit testing the matrix transport receive so do nothing
assert message
def mock_receive_delivered(klass, delivered):
# We are just unit testing the matrix transport receive so do nothing
assert delivered
config = dict(
retry_interval=retry_interval,
retries_before_backoff=retries_before_backoff,
server=local_matrix_server,
server_name='matrix.local.raiden',
available_servers=[],
discovery_room='discovery',
private_rooms=private_rooms,
)
transport = MatrixTransport(config)
transport.raiden = MockRaidenService(ADDR)
transport._stop_event.clear()
transport._address_to_userids[HOP1] = USERID1
monkeypatch.setattr(MatrixTransport, '_get_user', mock_get_user)
monkeypatch.setattr(
MatrixTransport,
'_get_room_ids_for_address',
mock_get_room_ids_for_address,
)
monkeypatch.setattr(MatrixTransport, '_set_room_id_for_address', mock_set_room_id_for_address)
monkeypatch.setattr(MatrixTransport, '_receive_message', mock_receive_message)
return transport
def mock_matrix(
monkeypatch,
retry_interval,
retries_before_backoff,
):
from matrix_client.user import User
monkeypatch.setattr(User, 'get_display_name',
lambda _: 'random_display_name')
def mock_get_user(klass, user: Union[User, str]) -> User:
return User(None, USERID1)
def mock_get_room_id_for_address(klass, address: Address) -> Optional[str]:
return '42'
def mock_set_room_id_for_address(self, address: Address,
room_id: Optional[str]):
pass
def mock_receive_message(klass, message):
# We are just unit testing the matrix transport receive so do nothing
assert message
def mock_receive_delivered(klass, delivered):
# We are just unit testing the matrix transport receive so do nothing
assert delivered
config = dict(
retry_interval=retry_interval,
retries_before_backoff=retries_before_backoff,
server='auto',
available_servers=[
'https://transport01.raiden.network',
'https://transport02.raiden.network',
'https://transport03.raiden.network',
],
discovery_room={
'alias_fragment': 'discovery',
'server': 'transport01.raiden.network',
},
)
transport = MatrixTransport(config)
transport.raiden = MockRaidenService(ADDR)
transport._stop_event.clear()
transport._address_to_userids[HOP1] = USERID1
monkeypatch.setattr(MatrixTransport, '_get_user', mock_get_user)
monkeypatch.setattr(MatrixTransport, '_get_room_id_for_address',
mock_get_room_id_for_address)
monkeypatch.setattr(MatrixTransport, '_set_room_id_for_address',
mock_set_room_id_for_address)
monkeypatch.setattr(MatrixTransport, '_receive_message',
mock_receive_message)
return transport
