Merge branch 'feature/add-coriant-tnms-config' into develop

inventory_provider/config.py

@@ -58,6 +58,18 @@ CONFIG_SCHEMA = {
                 "required": ["name", "address"],
                 "additionalProperties": False
             }
+        },
+        "coriant-tnms": {
+            "type": "array",
+            "items": {
+                "type": "object",
+                "properties": {
+                    "name": {"type": "string"},
+                    "address": {"type": "string"}
+                },
+                "required": ["name", "address"],
+                "additionalProperties": False
+            }
         }
     },
     "required": [
@@ -67,7 +79,8 @@ CONFIG_SCHEMA = {
         "routers_community.conf",
         "ssh",
         "redis",
-        "infinera-dna"],
+        "infinera-dna",
+        "coriant-tnms"],
     "additionalProperties": False
 }
 

inventory_provider/routes/classifier.py

@@ -36,6 +36,14 @@ def infinera_addresses():
     return jsonify([dna['address'] for dna in infinera_config])
 
 
+@routes.route("/coriant-tnms-addresses", methods=['GET', 'POST'])
+@require_accepts_json
+def coriant_addresses():
+    coriant_config = current_app.config[
+        "INVENTORY_PROVIDER_CONFIG"]["coriant-tnms"]
+    return jsonify([tnms['address'] for tnms in coriant_config])
+
+
 @routes.route("/juniper-server-addresses", methods=['GET', 'POST'])
 @require_accepts_json
 def juniper_addresses():

test/conftest.py

@@ -47,6 +47,11 @@ def data_config_filename(tmp_dir_name):
             {"name": "name1", "address": "123.456.789.0"},
             {"name": "name2", "address": "012.345.678.9"},
             {"name": "name3", "address": "111.222.333.000"}
+        ],
+        "coriant-tnms": [
+            {"name": "name1", "address": "123.456.789.0"},
+            {"name": "name2", "address": "012.345.678.9"},
+            {"name": "name3", "address": "111.222.333.000"}
         ]
     }
 

test/test_classifier_routes.py

@@ -23,6 +23,22 @@ def test_infinera_addresses(client):
         response_schema)
 
 
+def test_coriant_addresses(client):
+    response_schema = {
+        "$schema": "http://json-schema.org/draft-07/schema#",
+        "type": "array",
+        "items": {"type": "string"}
+    }
+
+    rv = client.post(
+        "/classifier/coriant-tnms-addresses",
+        headers=DEFAULT_REQUEST_HEADERS)
+    assert rv.status_code == 200
+    jsonschema.validate(
+        json.loads(rv.data.decode("utf-8")),
+        response_schema)
+
+
 def test_juniper_addresses(mocker, client):
 
     test_data = [