Support second level NSSI in terminate app

[optf/osdf.git] / test / functest / simulators / oof_dependencies_simulators.py

diff --git a/test/functest/simulators/oof_dependencies_simulators.py b/test/functest/simulators/oof_dependencies_simulators.py
index b0b7b32..d7b1f04 100644 (file)
--- a/test/functest/simulators/oof_dependencies_simulators.py
+++ b/test/functest/simulators/oof_dependencies_simulators.py
@@ -23,7 +23,7 @@ import glob
 import json
 import os
 
-from flask import Flask, jsonify
+from flask import Flask, jsonify, request
 
 from osdf.utils.interfaces import json_from_file
 
@@ -118,6 +118,40 @@ def get_nbr_list(cell_id, ts):
         return jsonify(data)
     return jsonify(data), 503
 
+@app.route("/simulated/aai/v23/nodes/service-instances/service-instance/<service_id>", methods=["GET"])
+def get_aai_instances(service_id):
+    data, status = get_payload_for_simulated_component('aai', 'nsi_instance.json')
+    if not status:
+        return jsonify(data)
+    return jsonify(data), 503
+
+@app.route("/simulated/aai/v23/dsl", methods=["PUT"])
+def dsl_query():
+
+    nssi_query = {"dsl": "service-instance*('service-instance-id','9629e36c-a3d9-4aed-8368-f72b8be1cd34')('workload-context', 'CN') > "
+                         "service-instance*('service-role','nsi')"}
+
+    nssi_with_nsi_query = {"dsl": "service-instance*('service-instance-id','9629e36c-a3d9-4aed-8368-f72b8be1cd34')('workload-context', 'CN') > "
+                                  "service-instance*('service-role','nsi')('service-instance-id',"
+                                  "'660ca85c-1a0f-4521-a559-65f23e794699')"}
+
+    queries = {
+        "nssi_query": nssi_query,
+        "nssi_with_nsi": nssi_with_nsi_query
+    }
+
+    count = {
+        "nssi_query": 1,
+        "nssi_with_nsi": 2
+    }
+
+    request_body = request.get_json()
+    service_count = 0
+    for query_type, query in queries.items():
+        if request_body == query:
+            service_count = count[query_type]
+    return {'results': [{'service-instance': service_count}]}
+
 
 if __name__ == "__main__":
     app.run(debug=True, host='0.0.0.0')