tests: add new def for redistribution in ospf

Signed-off-by: ckishimo <carles.kishimoto@gmail.com>

diff --git a/tests/topotests/ospf_basic_functionality/test_ospf_ecmp.py b/tests/topotests/ospf_basic_functionality/test_ospf_ecmp.py
index 5ef6b9b0bed0d2d9d1e0bad0ca1ec8c906d6d807..7569629a9cde2002507a90a5f49ec21d3ff72f22 100644 (file)
--- a/tests/topotests/ospf_basic_functionality/test_ospf_ecmp.py
+++ b/tests/topotests/ospf_basic_functionality/test_ospf_ecmp.py
@@ -216,6 +216,19 @@ def red_connected(dut, config=True):
     assert result is True, "Testcase: Failed \n Error: {}".format(result)
 
 
+def redistribute(dut, route_type, **kwargs):
+    """Local def for redstribution of routes inside ospf."""
+    global topo
+    tgen = get_topogen()
+
+    ospf_red = {dut: {"ospf": {"redistribute": [{"redist_type": route_type}]}}}
+    for k, v in kwargs.items():
+        ospf_red[dut]["ospf"]["redistribute"][0][k] = v
+
+    result = create_router_ospf(tgen, topo, ospf_red)
+    assert result is True, "Testcase : Failed \n Error: {}".format(result)
+
+
 # ##################################
 # Test cases start here.
 # ##################################

diff --git a/tests/topotests/ospf_basic_functionality/test_ospf_ecmp_lan.py b/tests/topotests/ospf_basic_functionality/test_ospf_ecmp_lan.py
index 967bc448790d3402df35d26a0ec885acd21cf538..f56a8849980a1c0366b3ca44410bf47e652cd11c 100644 (file)
--- a/tests/topotests/ospf_basic_functionality/test_ospf_ecmp_lan.py
+++ b/tests/topotests/ospf_basic_functionality/test_ospf_ecmp_lan.py
@@ -223,6 +223,19 @@ def red_connected(dut, config=True):
     assert result is True, "Testcase: Failed \n Error: {}".format(result)
 
 
+def redistribute(dut, route_type, **kwargs):
+    """Local def for redstribution of routes inside ospf."""
+    global topo
+    tgen = get_topogen()
+
+    ospf_red = {dut: {"ospf": {"redistribute": [{"redist_type": route_type}]}}}
+    for k, v in kwargs.items():
+        ospf_red[dut]["ospf"]["redistribute"][0][k] = v
+
+    result = create_router_ospf(tgen, topo, ospf_red)
+    assert result is True, "Testcase : Failed \n Error: {}".format(result)
+
+
 # ##################################
 # Test cases start here.
 # ##################################

diff --git a/tests/topotests/ospf_basic_functionality/test_ospf_lan.py b/tests/topotests/ospf_basic_functionality/test_ospf_lan.py
index 0f1115f81576aa379945752fe19aa8cd1685f3f8..d611c0539f7602356482fd586e4cf04a45049783 100644 (file)
--- a/tests/topotests/ospf_basic_functionality/test_ospf_lan.py
+++ b/tests/topotests/ospf_basic_functionality/test_ospf_lan.py
@@ -219,6 +219,19 @@ def red_connected(dut, config=True):
     assert result is True, "Testcase: Failed \n Error: {}".format(result)
 
 
+def redistribute(dut, route_type, **kwargs):
+    """Local def for redstribution of routes inside ospf."""
+    global topo
+    tgen = get_topogen()
+
+    ospf_red = {dut: {"ospf": {"redistribute": [{"redist_type": route_type}]}}}
+    for k, v in kwargs.items():
+        ospf_red[dut]["ospf"]["redistribute"][0][k] = v
+
+    result = create_router_ospf(tgen, topo, ospf_red)
+    assert result is True, "Testcase : Failed \n Error: {}".format(result)
+
+
 # ##################################
 # Test cases start here.
 # ##################################

diff --git a/tests/topotests/ospf_basic_functionality/test_ospf_nssa.py b/tests/topotests/ospf_basic_functionality/test_ospf_nssa.py
index 82a34d046cdd72f384f80ee8209426c62cd23492..44eaf814378a84a291fe1b5420923feec0662759 100644 (file)
--- a/tests/topotests/ospf_basic_functionality/test_ospf_nssa.py
+++ b/tests/topotests/ospf_basic_functionality/test_ospf_nssa.py
@@ -213,6 +213,19 @@ def red_connected(dut, config=True):
     assert result is True, "Testcase: Failed \n Error: {}".format(result)
 
 
+def redistribute(dut, route_type, **kwargs):
+    """Local def for redstribution of routes inside ospf."""
+    global topo
+    tgen = get_topogen()
+
+    ospf_red = {dut: {"ospf": {"redistribute": [{"redist_type": route_type}]}}}
+    for k, v in kwargs.items():
+        ospf_red[dut]["ospf"]["redistribute"][0][k] = v
+
+    result = create_router_ospf(tgen, topo, ospf_red)
+    assert result is True, "Testcase : Failed \n Error: {}".format(result)
+
+
 # ##################################
 # Test cases start here.
 # ##################################

diff --git a/tests/topotests/ospf_basic_functionality/test_ospf_routemaps.py b/tests/topotests/ospf_basic_functionality/test_ospf_routemaps.py
index 88667a6ac8e322a3a482d519a65dc6a23586a302..907f4383d6d2383410afc23d9e2a8f46dae64abc 100644 (file)
--- a/tests/topotests/ospf_basic_functionality/test_ospf_routemaps.py
+++ b/tests/topotests/ospf_basic_functionality/test_ospf_routemaps.py
@@ -193,6 +193,19 @@ def teardown_module(mod):
     logger.info("=" * 40)
 
 
+def redistribute(dut, route_type, **kwargs):
+    """Local def for redstribution of routes inside ospf."""
+    global topo
+    tgen = get_topogen()
+
+    ospf_red = {dut: {"ospf": {"redistribute": [{"redist_type": route_type}]}}}
+    for k, v in kwargs.items():
+        ospf_red[dut]["ospf"]["redistribute"][0][k] = v
+
+    result = create_router_ospf(tgen, topo, ospf_red)
+    assert result is True, "Testcase : Failed \n Error: {}".format(result)
+
+
 # ##################################
 # Test cases start here.
 # ##################################

diff --git a/tests/topotests/ospf_basic_functionality/test_ospf_rte_calc.py b/tests/topotests/ospf_basic_functionality/test_ospf_rte_calc.py
index 434d7f8ef5cd7a971687a0ef69253b84e2f7a777..ecd7798340eaa3a6fa7ed138472d649652966d3f 100644 (file)
--- a/tests/topotests/ospf_basic_functionality/test_ospf_rte_calc.py
+++ b/tests/topotests/ospf_basic_functionality/test_ospf_rte_calc.py
@@ -219,6 +219,19 @@ def red_connected(dut, config=True):
     assert result is True, "Testcase: Failed \n Error: {}".format(result)
 
 
+def redistribute(dut, route_type, **kwargs):
+    """Local def for redstribution of routes inside ospf."""
+    global topo
+    tgen = get_topogen()
+
+    ospf_red = {dut: {"ospf": {"redistribute": [{"redist_type": route_type}]}}}
+    for k, v in kwargs.items():
+        ospf_red[dut]["ospf"]["redistribute"][0][k] = v
+
+    result = create_router_ospf(tgen, topo, ospf_red)
+    assert result is True, "Testcase : Failed \n Error: {}".format(result)
+
+
 # ##################################
 # Test cases start here.
 # ##################################