Fix bgp_app_peer test
[integration/test.git] / tools / fastbgp / bgp_app_peer.py
index 2af032abd6715150a8ecb49da6ac9d7f3778f475..16f658f834a47fdd496cbfb5f5b2974704ff4859 100755 (executable)
@@ -6,17 +6,19 @@
 # terms of the Eclipse Public License v1.0 which accompanies this distribution,
 # and is available at http://www.eclipse.org/legal/epl-v10.html
 
-__author__ = "Radovan Sajben"
-__copyright__ = "Copyright(c) 2015, Cisco Systems, Inc."
-__license__ = "Eclipse Public License v1.0"
-__email__ = "rsajben@cisco.com"
-
 import requests
 import ipaddr
 import argparse
 import logging
 import time
 import xml.dom.minidom as md
+import os.path
+
+
+__author__ = "Radovan Sajben"
+__copyright__ = "Copyright(c) 2015, Cisco Systems, Inc."
+__license__ = "Eclipse Public License v1.0"
+__email__ = "rsajben@cisco.com"
 
 
 def _build_url(odl_ip, port, uri):
@@ -37,7 +39,7 @@ def _build_url(odl_ip, port, uri):
     return url
 
 
-def _stream_data(xml_template, prefix_base, prefix_len, count, element="ipv4-routes"):
+def _stream_data(xml_template, prefix_base, prefix_len, count, route_key=False, element="ipv4-routes"):
     """Stream list of routes based on xml template. Memory non-consumable
     data generation (on the fly).
 
@@ -50,19 +52,27 @@ def _stream_data(xml_template, prefix_base, prefix_len, count, element="ipv4-rou
 
         :count: number of routes to be generated
 
+        :route_key: bool deciding route-key tag existence
+
         :element: element to be returned
 
     Returns:
         :yield xml_data: requested data by elements as xml data
     """
     global total_build_data_time_counter
-
-    routes = md.parse(xml_template)
+    if os.path.isfile(xml_template + "." + stream):
+        routes = md.parse(xml_template + "." + stream)
+    elif os.path.isfile(xml_template):
+        routes = md.parse(xml_template)
+    else:
+        logger.error("Template '{}' does not exist.".format(xml_template))
 
     routes_node = routes.getElementsByTagName("ipv4-routes")[0]
     route_node = routes.getElementsByTagName("ipv4-route")[0]
     routes_node.removeChild(route_node)
     route_prefix = route_node.getElementsByTagName("prefix")[0]
+    if route_key:
+        key = route_node.getElementsByTagName("route-key")[0]
     prefix_gap = 2 ** (32 - prefix_len)
     prefix_index_list = range(count)
     if element == routes_node.tagName:
@@ -81,6 +91,8 @@ def _stream_data(xml_template, prefix_base, prefix_len, count, element="ipv4-rou
         prefix = prefix_base + prefix_index * prefix_gap
         prefix_str = str(prefix) + "/" + str(prefix_len)
         route_prefix.childNodes[0].nodeValue = prefix_str
+        if route_key:
+            key.childNodes[0].nodeValue = prefix_str
         xml_data = route_node.toxml()
         if prefix_index == 0:
             xml_data = xml_head + xml_data
@@ -184,13 +196,13 @@ def get_prefixes(odl_ip, port, uri, auth, prefix_base=None, prefix_len=None,
             if "prefix" in item:
                 prefixes += item + ","
                 prefix_count += 1
-        prefixes = prefixes[:len(prefixes)-1]
+        prefixes = prefixes[:len(prefixes) - 1]
         logger.debug("prefix_list=%s", prefixes)
         logger.info("prefix_count=%s", prefix_count)
 
 
 def post_prefixes(odl_ip, port, uri, auth, prefix_base=None, prefix_len=None,
-                  count=0, xml_template=None):
+                  count=0, route_key=False, xml_template=None):
     """Send a http POST request for creating a prefix list.
 
     Args:
@@ -208,6 +220,8 @@ def post_prefixes(odl_ip, port, uri, auth, prefix_base=None, prefix_len=None,
 
         :param count: number of prefixes to be processed
 
+        :route_key: bool deciding route-key tag existence
+
         :param xml_template: xml template for building the xml data (not used)
 
     Returns:
@@ -215,12 +229,12 @@ def post_prefixes(odl_ip, port, uri, auth, prefix_base=None, prefix_len=None,
     """
     logger.info("Post %s prefix(es) in a single request (starting from %s/%s) into %s:%s/restconf/%s",
                 count, prefix_base, prefix_len, odl_ip, port, uri)
-    xml_stream = _stream_data(xml_template, prefix_base, prefix_len, count)
+    xml_stream = _stream_data(xml_template, prefix_base, prefix_len, count, route_key)
     send_request("POST", odl_ip, port, uri, auth, xml_data=xml_stream, expect_status_code=204)
 
 
 def put_prefixes(odl_ip, port, uri, auth, prefix_base, prefix_len, count,
-                 xml_template=None):
+                 route_key, xml_template=None):
     """Send a http PUT request for updating the prefix list.
 
     Args:
@@ -246,12 +260,12 @@ def put_prefixes(odl_ip, port, uri, auth, prefix_base, prefix_len, count,
     uri_add_prefix = uri + _uri_suffix_ipv4_routes
     logger.info("Put %s prefix(es) in a single request (starting from %s/%s) into %s:%s/restconf/%s",
                 count, prefix_base, prefix_len, odl_ip, port, uri_add_prefix)
-    xml_stream = _stream_data(xml_template, prefix_base, prefix_len, count)
+    xml_stream = _stream_data(xml_template, prefix_base, prefix_len, count, route_key)
     send_request("PUT", odl_ip, port, uri_add_prefix, auth, xml_data=xml_stream)
 
 
 def add_prefixes(odl_ip, port, uri, auth, prefix_base, prefix_len, count,
-                 xml_template=None):
+                 route_key, xml_template=None):
     """Send a consequent http POST request for adding prefixes.
 
     Args:
@@ -282,7 +296,8 @@ def add_prefixes(odl_ip, port, uri, auth, prefix_base, prefix_len, count,
         prefix = prefix_base + prefix_index * prefix_gap
         logger.info("Adding prefix %s/%s to %s:%s/restconf/%s",
                     prefix, prefix_len, odl_ip, port, uri)
-        xml_stream = _stream_data(xml_template, prefix, prefix_len, 1, "ipv4-route")
+        xml_stream = _stream_data(xml_template, prefix, prefix_len, 1, route_key,
+                                  element="ipv4-route")
         send_request("POST", odl_ip, port, uri_add_prefix, auth,
                      xml_data=xml_stream, expect_status_code=204)
 
@@ -313,14 +328,15 @@ def delete_prefixes(odl_ip, port, uri, auth, prefix_base, prefix_len, count,
     """
     logger.info("Delete %s prefix(es) (starting from %s/%s) from %s:%s/restconf/%s",
                 count, prefix_base, prefix_len, odl_ip, port, uri)
+    partkey = "/0"
     uri_del_prefix = uri + _uri_suffix_ipv4_routes + _uri_suffix_ipv4_route
     prefix_gap = 2 ** (32 - prefix_len)
     for prefix_index in range(count):
         prefix = prefix_base + prefix_index * prefix_gap
-        logger.info("Deleting prefix %s/%s from %s:%s/restconf/%s",
-                    prefix, prefix_len, odl_ip, port, uri)
+        logger.info("Deleting prefix %s/%s/%s from %s:%s/restconf/%s",
+                    prefix, prefix_len, partkey, odl_ip, port, uri)
         send_request("DELETE", odl_ip, port,
-                     uri_del_prefix + str(prefix) + "%2F" + str(prefix_len), auth)
+                     uri_del_prefix + str(prefix) + "%2F" + str(prefix_len) + partkey, auth)
 
 
 def delete_all_prefixes(odl_ip, port, uri, auth, prefix_base=None,
@@ -392,6 +408,7 @@ if __name__ == "__main__":
                         const=logging.DEBUG, default=logging.INFO,
                         help="Set log level to debug (default is info)")
     parser.add_argument("--logfile", default="bgp_app_peer.log", help="Log file name")
+    parser.add_argument("--stream", default="", help="ODL Stream - oxygen, fluorine ...")
 
     args = parser.parse_args()
 
@@ -413,10 +430,14 @@ if __name__ == "__main__":
     prefix_base = args.prefix
     prefix_len = args.prefixlen
     count = args.count
-    auth = (args.user, args.password)
     uri = args.uri
+    stream = args.stream
     xml_template = args.xml
 
+    # From Fluorine onward route-key argument is mandatory for identification.
+    route_key_stream = ["oxygen"]
+    route_key = True if args.stream not in route_key_stream else False
+
     test_start_time = time.time()
     total_build_data_time_counter = 0
     total_response_time_counter = 0
@@ -424,13 +445,13 @@ if __name__ == "__main__":
 
     if command == "post":
         post_prefixes(odl_ip, port, uri, auth, prefix_base, prefix_len, count,
-                      xml_template)
+                      route_key, xml_template)
     if command == "put":
         put_prefixes(odl_ip, port, uri, auth, prefix_base, prefix_len, count,
-                     xml_template)
+                     route_key, xml_template)
     if command == "add":
         add_prefixes(odl_ip, port, uri, auth, prefix_base, prefix_len, count,
-                     xml_template)
+                     route_key, xml_template)
     elif command == "delete":
         delete_prefixes(odl_ip, port, uri, auth, prefix_base, prefix_len, count)
     elif command == "delete-all":