Function test_node_bias_persistence

tests/test_askrene.py:444–499

Test node bias persistence.

(node_factory)

def test_node_bias_persistence(node_factory):
    """Test node bias persistence."""
    # remove xpay, since it creates a layer!
    l1, l2 = node_factory.line_graph(
        2, wait_for_announce=True, opts={"disable-plugin": "cln-xpay"}
    )

    expect = {
        "layer": "mylayer",
        "persistent": True,
        "disabled_nodes": [],
        "created_channels": [],
        "channel_updates": [],
        "constraints": [],
        "biases": [],
        "node_biases": [],
    }
    l1.rpc.askrene_create_layer(layer="mylayer", persistent=True)
    r = l1.rpc.askrene_bias_node(
        layer="mylayer", node=l2.info["id"], direction="out", bias=14, relative=False
    )
    expect["node_biases"] = [
        {
            "node": l2.info["id"],
            "in_bias": 0,
            "out_bias": 14,
            "timestamp": r["node_biases"][0]["timestamp"],
        }
    ]
    assert l1.rpc.askrene_listlayers("mylayer") == {"layers": [expect]}
    # restarting the node we see the same data again
    l2.restart()
    assert l1.rpc.askrene_listlayers("mylayer") == {"layers": [expect]}

    r = l1.rpc.askrene_bias_node(
        layer="mylayer",
        node=l2.info["id"],
        direction="in",
        bias=11,
        relative=False,
        description="Some description",
    )
    expect["node_biases"] = [
        {
            "node": l2.info["id"],
            "in_bias": 11,
            "out_bias": 14,
            "timestamp": r["node_biases"][0]["timestamp"],
            "description": "Some description",
        }
    ]
    assert l1.rpc.askrene_listlayers("mylayer") == {"layers": [expect]}

    # restarting the node we see the same data again
    l2.restart()
    assert l1.rpc.askrene_listlayers("mylayer") == {"layers": [expect]}