Function test_gossip_throttle

tests/test_gossip.py:2166–2256

Make some gossip, test it gets throttled

(node_factory, bitcoind, chainparams)

def test_gossip_throttle(node_factory, bitcoind, chainparams):
    """Make some gossip, test it gets throttled"""
    l1, l2, l3, l4 = node_factory.line_graph(4, wait_for_announce=True,
                                             opts=[{}, {}, {},
                                                   {'broken_log': 'Throttling incoming peer',
                                                    'dev-throttle-gossip': None}])

    # We expect: self-advertizement (3 messages for l1 and l4) plus
    # 4 node announcements, 3 channel announcements and 6 channel updates.
    expected = 4 + 4 + 3 + 6

    # l1 is unlimited
    start_fast = time.time()
    out1 = subprocess.run(['devtools/gossipwith',
                           '--all-gossip',
                           '--hex',
                           '--network={}'.format(TEST_NETWORK),
                           '--filter=256,257,258',
                           '--max-messages={}'.format(expected),
                           '{}@localhost:{}'.format(l1.info['id'], l1.port)],
                          check=True,
                          timeout=TIMEOUT, stdout=subprocess.PIPE).stdout.split()
    time_fast = time.time() - start_fast
    assert time_fast < 2

    # l4 is throttled
    start_slow = time.time()
    out2 = subprocess.run(['devtools/gossipwith',
                           '--all-gossip',
                           '--hex',
                           '--network={}'.format(TEST_NETWORK),
                           '--filter=256,257,258',
                           '--max-messages={}'.format(expected),
                           '{}@localhost:{}'.format(l4.info['id'], l4.port)],
                          check=True,
                          timeout=TIMEOUT, stdout=subprocess.PIPE).stdout.split()
    time_slow = time.time() - start_slow
    assert time_slow > 3

    # Contents should be identical (once uniquified, since each
    # doubles-up on its own gossip)
    assert set(out1) == set(out2)

    encoded = subprocess.run(['devtools/mkencoded', '--scids', '00',
                              first_scid(l1, l2),
                              first_scid(l2, l3),
                              first_scid(l3, l4)],
                             check=True,
                             timeout=TIMEOUT,
                             stdout=subprocess.PIPE).stdout.strip().decode()

    query = subprocess.run(['devtools/mkquery',
                            'query_short_channel_ids',
                            chainparams['chain_hash'],
                            encoded,
                            # We want channel announce, updates and node ann.
                            '00', '1F1F1F'],
                           check=True,