tests: Adding framework support for bgp_always_compare_med

Added/enhanced bgp.py to support config/
unconfig bgp always compare med related
configurations.

Signed-off-by: Kuldeep Kashyap <kashyapk@vmware.com>

diff --git a/tests/topotests/lib/bgp.py b/tests/topotests/lib/bgp.py
index 0bd9408c28f58a72fe3ae13ab8ac7faffd28d9a9..21d4567d6ba0642f09289dc75b4b8fec1c74d6f5 100644 (file)
--- a/tests/topotests/lib/bgp.py
+++ b/tests/topotests/lib/bgp.py
@@ -50,6 +50,7 @@ def create_router_bgp(tgen, topo=None, input_dict=None, build=False, load_config
             "bgp": {
                 "local_as": "200",
                 "router_id": "22.22.22.22",
+                "bgp_always_compare_med": True,
                 "graceful-restart": {
                     "graceful-restart": True,
                     "preserve-fw-state": True,
@@ -343,6 +344,13 @@ def __create_bgp_global(tgen, input_dict, router, build=False):
 
                 config_data.append(cmd)
 
+    if "bgp_always_compare_med" in bgp_data:
+        bgp_always_compare_med = bgp_data["bgp_always_compare_med"]
+        if bgp_always_compare_med == True:
+            config_data.append("bgp always-compare-med")
+        elif bgp_always_compare_med == False:
+            config_data.append("no bgp always-compare-med")
+
     logger.debug("Exiting lib API: {}".format(sys._getframe().f_code.co_name))
     return config_data