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