def setUp(self) -> None:
        pub_key = "a04f30a45aae413d0ca0f219b4dcb7049857bc3f91a6351288cce603a2c9646294a02b987bf6586b370b2c22d74662355677007a14238bb037aedf41c2d08866"
        opts = gateway_helpers.get_gateway_opts(8000, pub_key=pub_key)

        node_ssl_service = MockNodeSSLService(EthGatewayNode.NODE_TYPE,
                                              MagicMock())
        self.node = EthGatewayNode(opts, node_ssl_service)
        self.relay_1 = MockNode(
            helpers.get_common_opts(8000, region="us-east-1"))
        self.relay_connection_1 = helpers.create_connection(
            EthRelayConnection,
            self.node,
            node_opts=self.relay_1.opts,
            file_no=1,
            ip="1.2.3.4")

        self.relay_2 = MockNode(
            helpers.get_common_opts(8001, region="eu-west-1"))
        self.relay_connection_2 = helpers.create_connection(
            EthRelayConnection,
            self.node,
            node_opts=self.relay_2.opts,
            file_no=1,
            ip="1.2.3.5")
        self.node.broadcast = MagicMock()
        self.node.has_active_blockchain_peer = MagicMock(return_value=True)
Exemple #2
0
def create_connection(port: int, node: AbstractGatewayNode,
                      **kwargs) -> GatewayConnection:
    connection = helpers.create_connection(GatewayConnection,
                                           node=node,
                                           port=port,
                                           **kwargs)
    return connection
    def setUp(self) -> None:
        opts = gateway_helpers.get_gateway_opts(8000)

        self.node = MockGatewayNode(opts)
        self.node.message_converter = EthNormalMessageConverter()
        self.connection = helpers.create_connection(EthRelayConnection,
                                                    self.node)
Exemple #4
0
 def setUp(self) -> None:
     self.socket_instance = MagicMock(spec=socket.socket)
     self.socket_instance.fileno = MagicMock(return_value=1)
     self.node = AbstractNode(helpers.get_common_opts(1234))
     self.sut = SocketConnection(self.socket_instance, self.node, False)
     self.connection = helpers.create_connection(MockConnection,
                                                 self.node,
                                                 from_me=True)
     self.connection.socket_connection = self.sut
 def setUp(self):
     self.node = MockNode(helpers.get_common_opts(8888))
     # noinspection PyTypeChecker
     connection = helpers.create_connection(MockConnection,
                                            self.node,
                                            port=9999)
     self.node.connection_pool = ConnectionPool()
     self.node.connection_pool.add(1, connection.peer_ip,
                                   connection.peer_port, connection)
     node_info_statistics.set_node(self.node)
Exemple #6
0
    def setUp(self) -> None:
        pub_key = "a04f30a45aae413d0ca0f219b4dcb7049857bc3f91a6351288cce603a2c9646294a02b987bf6586b370b2c22d74662355677007a14238bb037aedf41c2d08866"
        opts = gateway_helpers.get_gateway_opts(8000, pub_key=pub_key)

        node_ssl_service = MockNodeSSLService(EthGatewayNode.NODE_TYPE,
                                              MagicMock())
        self.node = EthGatewayNode(opts, node_ssl_service)
        self.connection = helpers.create_connection(EthRelayConnection,
                                                    self.node)
        self.node.broadcast = MagicMock()
        self.node.has_active_blockchain_peer = MagicMock(return_value=True)
Exemple #7
0
    def setUp(self):
        sdn_http_service.submit_peer_connection_error_event = MagicMock()

        self.node = TestNode(helpers.get_common_opts(4321), None)
        self.fileno = 1
        self.ip = "123.123.123.123"
        self.port = 8000
        self.connection = helpers.create_connection(MockConnection, self.node, file_no=self.fileno, ip=self.ip,
                                                    port=self.port, add_to_pool=False, from_me=True)
        self.connection.dispose = MagicMock(side_effect=self.connection.dispose)
        self.socket_connection = self.connection.socket_connection
Exemple #8
0
 def setUp(self):
     self.node = TestNode(helpers.get_common_opts(4321))
     self.fileno = 1
     self.ip = "123.123.123.123"
     self.port = 8000
     self.connection = helpers.create_connection(MockConnection,
                                                 self.node,
                                                 fileno=self.fileno,
                                                 ip=self.ip,
                                                 port=self.port,
                                                 add_to_pool=False)
     self.connection.close = MagicMock(side_effect=self.connection.close)
     self.socket_connection = self.connection.socket_connection
Exemple #9
0
    def setUp(self) -> None:
        pub_key = "a04f30a45aae413d0ca0f219b4dcb7049857bc3f91a6351288cce603a2c9646294a02b987bf6586b370b2c22d74662355677007a14238bb037aedf41c2d08866"
        node_ssl_service = MockNodeSSLService(EthGatewayNode.NODE_TYPE, MagicMock())
        opts = gateway_helpers.get_gateway_opts(
            8000,
            include_default_eth_args=True,
            blockchain_protocol=BlockchainProtocol.ETHEREUM.name,
            blockchain_network_num=5,
            pub_key=pub_key,
        )
        self.node = EthGatewayNode(opts, node_ssl_service)
        self.enqueued_messages = []
        self.broadcast_to_node_messages = []
        self.broadcast_messages = []
        self.node.opts.has_fully_updated_tx_service = True
        self.node.init_live_feeds()

        self.connection = helpers.create_connection(
            EthNodeConnection, self.node, opts, port=opts.blockchain_port
        )

        gateway_helpers.add_blockchain_peer(self.node, self.connection)
        self.block_queuing_service = self.node.block_queuing_service_manager.get_designated_block_queuing_service()

        self.connection.on_connection_established()
        self.node.node_conn = self.connection
        gateway_bdn_performance_stats_service.set_node(self.node)

        def mocked_enqueue_msg(msg, prepend=False):
            self.enqueued_messages.append(msg)

        self.connection.enqueue_msg = mocked_enqueue_msg

        def mocked_broadcast(msg, broadcasting_conn=None, prepend_to_queue=False, connection_types=None):
            if connection_types is None:
                connection_types = [ConnectionType.RELAY_ALL]
            if len(connection_types) == 1 and connection_types[0] == ConnectionType.BLOCKCHAIN_NODE:
                self.broadcast_to_node_messages.append(msg)
            else:
                self.broadcast_messages.append(msg)

        self.node.broadcast = mocked_broadcast

        self.cleanup_service = self.node.block_cleanup_service
        self.node.block_processing_service.queue_block_for_processing = MagicMock()

        self.sut = self.connection.connection_protocol
        self.sut._waiting_checkpoint_headers_request = False
        self.node.opts.ws = True
        self.node.publish_block = MagicMock()
Exemple #10
0
    def setUp(self) -> None:
        pub_key = "a04f30a45aae413d0ca0f219b4dcb7049857bc3f91a6351288cce603a2c9646294a02b987bf6586b370b2c22d74662355677007a14238bb037aedf41c2d08866"
        opts = gateway_helpers.get_gateway_opts(
            8000,
            include_default_eth_args=True,
            pub_key=pub_key,
            track_detailed_sent_messages=True)
        if opts.use_extensions:
            helpers.set_extensions_parallelism()
        node_ssl_service = MockNodeSSLService(EthGatewayNode.NODE_TYPE,
                                              MagicMock())
        self.node = EthGatewayNode(opts, node_ssl_service)
        self.connection_fileno = 1
        self.connection = helpers.create_connection(
            EthNodeConnection, node=self.node, file_no=self.connection_fileno)

        cipher1, cipher2 = AbstractRLPxCipherTest().setup_ciphers()
        self.connection.connection_protocol.rlpx_cipher = cipher1
Exemple #11
0
 def setUp(self):
     self.connection = create_connection(self.TestAbstractConnection)
Exemple #12
0
 def setUp(self):
     self.connection = helpers.create_connection(InternalNodeConnection)
     self.connection.on_connection_established()
     self.alarm_queue = AlarmQueue()
     self.connection.node.alarm_queue = self.alarm_queue
Exemple #13
0
 def setUp(self):
     self.connection = helpers.create_connection(InternalNodeConnection)
     self.connection.state = self.connection.state | ConnectionState.ESTABLISHED
     self.alarm_queue = AlarmQueue()
     self.connection.node.alarm_queue = self.alarm_queue