test_static_rt_bfd.py

from unittest.mock import patch
#from unittest.mock import MagicMock, patch

from staticroutebfd.main import *
from swsscommon import swsscommon

@patch('swsscommon.swsscommon.DBConnector.__init__')
@patch('swsscommon.swsscommon.ProducerStateTable.__init__')
@patch('swsscommon.swsscommon.Table.__init__')
def constructor(mock_db, mock_producer, mock_tbl):
    mock_db.return_value = None
    mock_producer.return_value = None
    mock_tbl.return_value = None

    srt_bfd = StaticRouteBfd()
    return srt_bfd

def set_del_test(dut, hdlr, op, args, e_bfd_dict, e_srt_dict):
    set_del_test.bfd_dict = {}
    set_del_test.srt_dict = {}

    def bfd_app_set(key, data):
        set_del_test.bfd_dict["set_"+key] = data.copy()
    def bfd_app_del(key):
        set_del_test.bfd_dict["del_"+key] = {}
    def srt_app_set(key, data):
        set_del_test.srt_dict["set_"+key] = data.copy()
    def srt_app_del(key):
        set_del_test.srt_dict["del_"+key] = {}

    def compare_dict(r, e):
        if len(r) == 0 and len(e) == 0:
            return True
        if len(r) != len(e):
            return False
        for k in e:
            if k not in r:
                return False
            if type(e[k]) is str:
                r_sort = "".join(sorted([x.strip() for x in r[k].split(',')]))
                e_sort = "".join(sorted([x.strip() for x in e[k].split(',')]))
                if r_sort != e_sort:
                    return False
            if type(e[k]) is dict:
                ret = compare_dict(r[k], e[k])
                if not ret:
                    return False
        return True

    dut.set_bfd_session_into_appl_db = bfd_app_set
    dut.del_bfd_session_from_appl_db = bfd_app_del
    dut.set_static_route_into_appl_db = srt_app_set
    dut.del_static_route_from_appl_db = srt_app_del

    if op == "SET":
        if hdlr == "bfd":
            dut.bfd_state_set_handler(*args)
        if hdlr == "srt":
            dut.static_route_set_handler(*args)
        if hdlr == "intf":
            dut.interface_set_handler(*args)
    elif op == "DEL":
        if hdlr == "bfd":
            dut.bfd_state_del_handler(*args)
        if hdlr == "srt":
            dut.static_route_del_handler(*args)
        if hdlr == "intf":
            dut.interface_del_handler(*args)
    else:
        assert False, "Wrong operation"

    assert compare_dict(set_del_test.bfd_dict, e_bfd_dict)
    assert compare_dict(set_del_test.srt_dict, e_srt_dict)

def intf_setup(dut):
    set_del_test(dut, "intf",
        "SET",
        ("if1|192.168.1.1/24", {}
        ),
        {},
        {}
    )
    set_del_test(dut, "intf",
        "SET",
        ("if2|192.168.2.1/24", {}
        ),
        {},
        {}
    )
    set_del_test(dut, "intf",
        "SET",
        ("if3|192.168.3.1/24", {}
        ),
        {},
        {}
    )

def test_set_del():
    dut = constructor()
    intf_setup(dut)

    #test #1
    set_del_test(dut, "srt",
        "SET",
        ("2.2.2.0/24", {
            "bfd": "true",
            "nexthop": "192.168.1.2 , 192.168.2.2, 192.168.3.2",
            "ifname": "if1, if2, if3",
        }),
        { 
            "set_default:default:192.168.1.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.1.1'},
            "set_default:default:192.168.2.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.2.1'},
            "set_default:default:192.168.3.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.3.1'}
        },
        {}
    )

    set_del_test(dut, "bfd",
        "SET",
        ("192.168.1.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.1.2', 'ifname': 'if1', 'nexthop-vrf': 'default', 'expiry': 'false'}}
    )
    set_del_test(dut, "bfd",
        "SET",
        ("192.168.2.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2 ', 'ifname': 'if2,if1', 'nexthop-vrf': 'default,default', 'expiry': 'false'}}
    )
    set_del_test(dut, "bfd",
        "SET",
        ("192.168.3.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2,192.168.3.2 ', 'ifname': 'if2,if1,if3', 'nexthop-vrf': 'default,default,default', 'expiry': 'false'}}
    )

    #test #2
    set_del_test(dut, "srt",
        "SET",
        ("2.2.2.0/24", {
            "bfd": "true",
            "nexthop": "192.168.1.2 , 192.168.2.2",
            "ifname": "if1, if2",
        }),
        { 
            "del_default:default:192.168.3.2" : {}
        },
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2 ', 'ifname': 'if2,if1', 'nexthop-vrf': 'default,default', 'expiry': 'false'}}
    )

    #test #3
    set_del_test(dut, "srt",
        "DEL",
        ("2.2.2.0/24", {
            "bfd": "true",
            "nexthop": "192.168.1.2 , 192.168.2.2",
            "ifname": "if1, if2",
        }),
        { 
            "del_default:default:192.168.1.2" : {},
            "del_default:default:192.168.2.2" : {}
        },
        {'del_default:2.2.2.0/24': {}}
    )

    # test add a non-bfd static route
    set_del_test(dut, "srt",
        "SET",
        ("3.3.3.0/24", {
            "nexthop": "192.168.1.2 , 192.168.2.2",
            "ifname": "if1, if2",
        }),
        {},
        {}
    )

    # test delete a non-bfd static route
    set_del_test(dut, "srt",
        "DEL",
        ("3.3.3.0/24", {}),
        {},
        {}
    )

def test_set_del_vrf():
    dut = constructor()
    intf_setup(dut)

    set_del_test(dut, "srt",
        "SET",
        ("vrfred|2.2.2.0/24", {
            "bfd": "true",
            "nexthop": "192.168.1.2 , 192.168.2.2, 192.168.3.2",
            "ifname": "if1, if2, if3",
            "nexthop-vrf": "testvrf1, , default",
        }),
        { 
            "set_testvrf1:default:192.168.1.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.1.1'},
            "set_vrfred:default:192.168.2.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.2.1'},
            "set_default:default:192.168.3.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.3.1'}
        },
        {}
    )

    set_del_test(dut, "bfd",
        "SET",
        ("testvrf1|default|192.168.1.2", {
            "state": "Up"
        }),
        {},
        {'set_vrfred:2.2.2.0/24': {'nexthop': '192.168.1.2', 'ifname': 'if1', 'nexthop-vrf': 'testvrf1', 'expiry': 'false'}}
    )
    set_del_test(dut, "bfd",
        "SET",
        ("vrfred|default|192.168.2.2", {
            "state": "Up"
        }),
        {},
        {'set_vrfred:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2 ', 'ifname': 'if2,if1', 'nexthop-vrf': 'testvrf1,vrfred', 'expiry': 'false'}}
    )
    set_del_test(dut, "bfd",
        "SET",
        ("default|default|192.168.3.2", {
            "state": "Up"
        }),
        {},
        {'set_vrfred:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2,192.168.3.2 ', 'ifname': 'if2,if1,if3', 'nexthop-vrf': 'testvrf1,vrfred,default', 'expiry': 'false'}}
    )

    set_del_test(dut, "srt",
        "SET",
        ("vrfred|2.2.2.0/24", {
            "bfd": "true",
            "nexthop": "192.168.1.2 , 192.168.2.2",
            "ifname": "if1, if2",
            "nexthop-vrf": "testvrf1,",
        }),
        { 
            "del_default:default:192.168.3.2" : {}
        },
        {'set_vrfred:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2 ', 'ifname': 'if2,if1', 'nexthop-vrf': 'vrfred, testvrf1', 'expiry': 'false'}}
    )

    set_del_test(dut, "srt",
        "DEL",
        ("vrfred|2.2.2.0/24", { }),
        { 
            "del_testvrf1:default:192.168.1.2" : {},
            "del_vrfred:default:192.168.2.2" : {}
        },
        {'del_vrfred:2.2.2.0/24': {}}
    )

def test_bfd_del():
    dut = constructor()
    intf_setup(dut)

    set_del_test(dut, "srt",
        "SET",
        ("2.2.2.0/24", {
            "bfd": "true",
            "nexthop": "192.168.1.2 , 192.168.2.2, 192.168.3.2",
            "ifname": "if1, if2, if3",
        }),
        { 
            "set_default:default:192.168.1.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.1.1'},
            "set_default:default:192.168.2.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.2.1'},
            "set_default:default:192.168.3.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.3.1'}
        },
        {}
    )

    set_del_test(dut, "bfd",
        "SET",
        ("192.168.1.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.1.2', 'ifname': 'if1', 'nexthop-vrf': 'default', 'expiry': 'false'}}
    )
    set_del_test(dut, "bfd",
        "SET",
        ("192.168.2.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2 ', 'ifname': 'if2,if1', 'nexthop-vrf': 'default,default', 'expiry': 'false'}}
    )
    set_del_test(dut, "bfd",
        "SET",
        ("192.168.3.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2,192.168.3.2 ', 'ifname': 'if2,if1,if3', 'nexthop-vrf': 'default,default,default', 'expiry': 'false'}}
    )

    #test bfd state del
    set_del_test(dut, "bfd",
        "DEL",
        ({"192.168.2.2"}),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.1.2,192.168.3.2 ', 'ifname': 'if1,if3', 'nexthop-vrf': 'default,default', 'expiry': 'false'}}
    )

def test_set_2routes():
    dut = constructor()
    intf_setup(dut)

    #test #4
    set_del_test(dut, "srt",
        "SET",
        ("2.2.2.0/24", {
            "bfd": "true",
            "nexthop": "192.168.1.2 , 192.168.2.2, 192.168.3.2",
            "ifname": "if1, if2, if3",
        }),
        { 
            "set_default:default:192.168.1.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.1.1'},
            "set_default:default:192.168.2.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.2.1'},
            "set_default:default:192.168.3.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.3.1'}
        },
        {}
    )

    set_del_test(dut, "bfd",
        "SET",
        ("192.168.1.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.1.2', 'ifname': 'if1', 'nexthop-vrf': 'default', 'expiry': 'false'}}
    )
    set_del_test(dut, "bfd",
        "SET",
        ("192.168.2.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2 ', 'ifname': 'if2,if1', 'nexthop-vrf': 'default,default', 'expiry': 'false'}}
    )
    set_del_test(dut, "bfd",
        "SET",
        ("192.168.3.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2,192.168.3.2 ', 'ifname': 'if2,if1,if3', 'nexthop-vrf': 'default,default,default', 'expiry': 'false'}}
    )

    set_del_test(dut, "srt",
        "SET",
        ("3.3.3.0/24", {
            "bfd": "true",
            "nexthop": "192.168.2.2",
            "ifname": "if2",
        }),
        {},
        {'set_default:3.3.3.0/24': {'nexthop': '192.168.2.2', 'ifname': 'if2', 'nexthop-vrf': 'default', 'expiry': 'false'}}
    )

    #test #5
    set_del_test(dut, "bfd",
        "SET",
        ("192.168.2.2", {
            "state": "Down"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.3.2,192.168.1.2 ', 'ifname': 'if3,if1', 'nexthop-vrf': 'default,default', 'expiry': 'false'}, 'del_default:3.3.3.0/24': {}}
    )

    #test #6
    set_del_test(dut, "bfd",
        "SET",
        ("192.168.2.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2,192.168.3.2 ', 'ifname': 'if2,if1,if3', 'nexthop-vrf': 'default,default,default', 'expiry': 'false'},
         'set_default:3.3.3.0/24': {'nexthop': '192.168.2.2', 'ifname': 'if2', 'nexthop-vrf': 'default', 'expiry': 'false'}}
    )

def test_set_bfd_change_hold():
    dut = constructor()
    intf_setup(dut)

    #test #9 bfd: true -> false
    set_del_test(dut, "srt",
        "SET",
        ("2.2.2.0/24", {
            "bfd": "true",
            "nexthop": "192.168.1.2 , 192.168.2.2, 192.168.3.2",
            "ifname": "if1, if2, if3",
        }),
        { 
            "set_default:default:192.168.1.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.1.1'},
            "set_default:default:192.168.2.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.2.1'},
            "set_default:default:192.168.3.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.3.1'}
        },
        {}
    )

    set_del_test(dut, "bfd",
        "SET",
        ("192.168.1.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.1.2', 'ifname': 'if1', 'nexthop-vrf': 'default', 'expiry': 'false'}}
    )
    set_del_test(dut, "bfd",
        "SET",
        ("192.168.2.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2 ', 'ifname': 'if2,if1', 'nexthop-vrf': 'default,default', 'expiry': 'false'}}
    )
    set_del_test(dut, "bfd",
        "SET",
        ("192.168.3.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2,192.168.3.2 ', 'ifname': 'if2,if1,if3', 'nexthop-vrf': 'default,default,default', 'expiry': 'false'}}
    )

    set_del_test(dut, "srt",
        "SET",
        ("2.2.2.0/24", {
            "bfd": "false",
            "nexthop": "192.168.1.2 , 192.168.2.2, 192.168.3.2",
            "ifname": "if1, if2, if3",
        }),
        { 
            "del_default:default:192.168.1.2" : {},
            "del_default:default:192.168.2.2" : {},
            "del_default:default:192.168.3.2" : {}
        },
        {
         'del_default:2.2.2.0/24': {}
        }
    )
    return

    #test #10 'bfd': false --> true, write original rout first
    set_del_test(dut, "srt",
        "SET",
        ("2.2.2.0/24", {
            "bfd": "true",
            "nexthop": "192.168.1.2 , 192.168.2.2, 192.168.3.2",
            "ifname": "if1, if2, if3",
        }),
        { 
            "set_default:default:192.168.1.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.1.1'},
            "set_default:default:192.168.2.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.2.1'},
            "set_default:default:192.168.3.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.3.1'}
        },
        {'set_default:2.2.2.0/24': {'bfd':'false', 'nexthop': '192.168.2.2,192.168.1.2,192.168.3.2 ', 'ifname': 'if2,if1,if3', 'expiry': 'false'}}
    )

    set_del_test(dut, "bfd",
        "SET",
        ("192.168.1.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.1.2', 'ifname': 'if1', 'nexthop-vrf': 'default', 'expiry': 'false'}}
    )
    set_del_test(dut, "bfd",
        "SET",
        ("192.168.2.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2 ', 'ifname': 'if2,if1', 'nexthop-vrf': 'default,default', 'expiry': 'false'}}
    )
    set_del_test(dut, "bfd",
        "SET",
        ("192.168.3.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2,192.168.3.2 ', 'ifname': 'if2,if1,if3', 'nexthop-vrf': 'default,default,default', 'expiry': 'false'}}
    )


def test_set_bfd_change_no_hold():
    dut = constructor()
    intf_setup(dut)

    #setup runtime "bfd"="false" condition``
    set_del_test(dut, "srt",
        "SET",
        ("2.2.2.0/24", {
            "bfd": "true",
            "nexthop": "192.168.1.2 , 192.168.2.2, 192.168.3.2",
            "ifname": "if1, if2, if3",
        }),
        { 
            "set_default:default:192.168.1.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.1.1'},
            "set_default:default:192.168.2.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.2.1'},
            "set_default:default:192.168.3.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.3.1'}
        },
        {}
    )
    set_del_test(dut, "bfd",
        "SET",
        ("192.168.1.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.1.2', 'ifname': 'if1', 'nexthop-vrf': 'default', 'expiry': 'false'}}
    )
    set_del_test(dut, "bfd",
        "SET",
        ("192.168.2.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2 ', 'ifname': 'if2,if1', 'nexthop-vrf': 'default,default', 'expiry': 'false'}}
    )
    set_del_test(dut, "bfd",
        "SET",
        ("192.168.3.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2,192.168.3.2 ', 'ifname': 'if2,if1,if3', 'nexthop-vrf': 'default,default,default', 'expiry': 'false'}}
    )
    set_del_test(dut, "srt",
        "SET",
        ("3.3.3.0/24", {
            "bfd": "true",
            "nexthop": "192.168.2.2",
            "ifname": "if2",
        }),
        {},
        {'set_default:3.3.3.0/24': {'nexthop': '192.168.2.2', 'ifname': 'if2', 'nexthop-vrf': 'default', 'expiry': 'false'}}
    )

    set_del_test(dut, "srt",
        "SET",
        ("2.2.2.0/24", {
            "bfd": "false",
            "nexthop": "192.168.1.2 , 192.168.2.2, 192.168.3.2",
            "ifname": "if1, if2, if3",
        }),
        { 
            "del_default:default:192.168.1.2" : {},
            "del_default:default:192.168.3.2" : {}
        },
        {
         'del_default:2.2.2.0/24': {}
        }
    )

    #test #10 change 'bfd': false to true, because the bfd session "default:default:192.168.2.2" is up, so add that nexthop right after "bfd" change to "true" 
    set_del_test(dut, "srt",
        "SET",
        ("2.2.2.0/24", {
            "bfd": "true",
            "nexthop": "192.168.1.2 , 192.168.2.2, 192.168.3.2",
            "ifname": "if1, if2, if3",
        }),
        { 
            "set_default:default:192.168.1.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.1.1'},
            "set_default:default:192.168.3.2" : {'multihop': 'false', 'rx_interval': '50', 'tx_interval': '50', 'multiplier': '3', 'local_addr': '192.168.3.1'}
        },
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.2.2', 'ifname': 'if2', 'nexthop-vrf': 'default', 'expiry': 'false'}}
    )

    set_del_test(dut, "bfd",
        "SET",
        ("192.168.1.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2 ', 'ifname': 'if2,if1', 'nexthop-vrf': 'default,default', 'expiry': 'false'}}
    )
    set_del_test(dut, "bfd",
        "SET",
        ("192.168.2.2", {
            "state": "Up"
        }),
        {},
        {}
    )
    set_del_test(dut, "bfd",
        "SET",
        ("192.168.3.2", {
            "state": "Up"
        }),
        {},
        {'set_default:2.2.2.0/24': {'nexthop': '192.168.2.2,192.168.1.2,192.168.3.2 ', 'ifname': 'if2,if1,if3', 'nexthop-vrf': 'default,default,default', 'expiry': 'false'}}
    )