added service info to bgp peer info response

inventory_provider/routes/classifier.py

@@ -88,6 +88,30 @@ def find_interfaces(address):
             yield info
 
 
+def find_interfaces_and_services(address):
+    """
+
+    :param address: an ipaddress object
+    :return:
+    """
+    r = common.get_redis()
+    for interface in find_interfaces(address):
+
+        services = r.get(
+            'opsdb:interface_services:%s:%s' % (
+                interface['router'],
+                interface['interface name']))
+        if not services:
+            services = []
+        else:
+            services = json.loads(services.decode('utf=8'))
+
+        yield {
+            'interface': interface,
+            'services': services
+        }
+
+
 @routes.route("/peer-info/<address>", methods=['GET', 'POST'])
 @common.require_accepts_json
 def peer_info(address):
@@ -124,7 +148,7 @@ def peer_info(address):
             info = info.decode('utf-8')
             result['vpn-rr-peer-info'] = json.loads(info)
 
-        interfaces = list(find_interfaces(address_obj))
+        interfaces = list(find_interfaces_and_services(address_obj))
         if interfaces:
             result['interfaces'] = interfaces
 

test/test_classifier_routes.py

@@ -101,6 +101,19 @@ def test_peer_info(
                 "required": [
                     "name", "interface address", "interface name", "router"],
                 "additionalProperties": False
+            },
+            "service-info": {
+                "type": "object"
+            },
+            "interface-lookup-info": {
+                "type": "object",
+                "properties": {
+                    "interface": {"$ref": "#/definitions/interface-info"},
+                    "services": {
+                        "type": "array",
+                        "items": {"$ref": "#/definitions/service-info"}
+                    }
+                }
             }
         },
 
@@ -112,7 +125,7 @@ def test_peer_info(
             "vpn-rr-peer-info": {"$ref": "#/definitions/vpn-rr-peer"},
             "interfaces": {
                 "type": "array",
-                "items": {"$ref": "#/definitions/interface-info"}
+                "items": {"$ref": "#/definitions/interface-lookup-info"}
             }
         },
         "additionalProperties": False