test/functest/simulators/oof_dependencies_simulators.py

   1 # -------------------------------------------------------------------------
   2 #   Copyright (c) 2018 AT&T Intellectual Property
   3 #
   4 #   Licensed under the Apache License, Version 2.0 (the "License");
   5 #   you may not use this file except in compliance with the License.
   6 #   You may obtain a copy of the License at
   7 #
   8 #       http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 #   Unless required by applicable law or agreed to in writing, software
  11 #   distributed under the License is distributed on an "AS IS" BASIS,
  12 #   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 #   See the License for the specific language governing permissions and
  14 #   limitations under the License.
  15 #
  16 # -------------------------------------------------------------------------
  17 #
  18
  19 """
  20 Simulators for dependencies of OSDF (e.g. HAS-API, Policy, SO-callback, etc.)
  21 """
  22 import glob
  23 import json
  24 import os
  25
  26 from flask import Flask, jsonify, request
  27
  28 from osdf.utils.interfaces import json_from_file
  29
  30 app = Flask(__name__)
  31
  32
  33 @app.route("/simulated/ERROR/<component>", methods=["GET", "POST"])
  34 @app.route("/simulated/unhealthy/<component>", methods=["GET", "POST"])
  35 def error_for_component(component):
  36     """Send an HTTP error for component"""
  37     return jsonify({"error": "{} error".format(component)}), 503
  38
  39
  40 @app.route("/simulated/healthy/<component>", methods=["GET", "POST"])
  41 def healthy_status_for_component(component):
  42     """Send a health-OK response for component"""
  43     return jsonify({"success": "Passed Health Check for Component {}".format(component)})
  44
  45
  46 @app.route("/simulated/success/<component>", methods=["GET", "POST"])
  47 def successful_call_for_component(component):
  48     """Send a message about successful call to component"""
  49     return jsonify({"success": "made a call to Component: {}".format(component)})
  50
  51
  52 @app.route("/simulated/oof/has-api/<path:mainpath>", methods=["GET", "POST"])
  53 def has_api_calls(mainpath):
  54     data, status = get_payload_for_simulated_component('has-api', mainpath)
  55     if not status:
  56         return jsonify(data)
  57     return jsonify(data), 503
  58
  59
  60 def get_payload_for_simulated_component(component, mainpath):
  61     """
  62     Get the payload for the given path for the given component
  63     :param component: Component we are using (e.g. HAS-API, Policy, SO-callback, etc.)
  64     :param mainpath: path within the URL (e.g. /main/X1/y1/)
  65     :return: Content if file exists, or else 503 error
  66     """
  67     file_name = "{}/response-payloads/{}".format(component, mainpath)
  68     data = json_from_file(file_name)
  69     if not data:
  70         return {"Error": "Unable to read File {}".format(file_name)}, 503
  71     return data, None
  72
  73
  74 @app.route("/simulated/policy/<sub_component>/getConfig", methods=["POST"])
  75 def get_policies(sub_component):
  76     """
  77     Get all policies for this folder
  78     :param sub_component: The folder we are interested in (e.g. "pdp-has-vcpe-good", "pdp-has-vcpe-bad")
  79     :return: A list of policies as a json object (each element is one policy)
  80     """
  81     main_dir = "policy/response-payloads/" + sub_component
  82     files = glob.glob("{}/*.json".format(main_dir))
  83     list_json = []
  84     for x in files:
  85         list_json.append({
  86             "policyConfigMessage": "Config Retrieved! ",
  87             "policyConfigStatus": "CONFIG_RETRIEVED",
  88             "type": "JSON",
  89             "config": json.dumps(json_from_file(x)),
  90             "policyName": os.path.basename(x),
  91             "policyType": "MicroService",
  92             "policyVersion": "1"
  93         })
  94     return jsonify(list_json)
  95
  96
  97 @app.route("/simulated/policy/pdpx/decision/v1", methods=["POST"])
  98 def get_pdx_policies():
  99     """
 100     get the pdpx policy
 101     """
 102     return jsonify(json_from_file("policy/response-payloads/policy_response.json"))
 103
 104
 105 @app.route("/simulated/configdb/getCellList/<network_id>/<ts>", methods=["GET"])
 106 def get_cell_list(network_id, ts):
 107     data, status = get_payload_for_simulated_component('configdb',
 108                                                        'getCellList-' + network_id + '.json')
 109     if not status:
 110         return jsonify(data)
 111     return jsonify(data), 503
 112
 113
 114 @app.route("/simulated/configdb/getNbrList/<cell_id>/<ts>", methods=["GET"])
 115 def get_nbr_list(cell_id, ts):
 116     data, status = get_payload_for_simulated_component('configdb', 'getNbrList-' + cell_id + '.json')
 117     if not status:
 118         return jsonify(data)
 119     return jsonify(data), 503
 120
 121 @app.route("/simulated/aai/v23/nodes/service-instances/service-instance/<service_id>", methods=["GET"])
 122 def get_aai_instances(service_id):
 123     data, status = get_payload_for_simulated_component('aai', 'nsi_instance.json')
 124     if not status:
 125         return jsonify(data)
 126     return jsonify(data), 503
 127
 128 @app.route("/simulated/aai/v23/dsl", methods=["PUT"])
 129 def dsl_query():
 130
 131     nssi_query = {"dsl": "service-instance*('service-instance-id','9629e36c-a3d9-4aed-8368-f72b8be1cd34')('workload-context', 'CN') > "
 132                          "service-instance*('service-role','nsi')"}
 133
 134     nssi_with_nsi_query = {"dsl": "service-instance*('service-instance-id','9629e36c-a3d9-4aed-8368-f72b8be1cd34')('workload-context', 'CN') > "
 135                                   "service-instance*('service-role','nsi')('service-instance-id',"
 136                                   "'660ca85c-1a0f-4521-a559-65f23e794699')"}
 137
 138     queries = {
 139         "nssi_query": nssi_query,
 140         "nssi_with_nsi": nssi_with_nsi_query
 141     }
 142
 143     count = {
 144         "nssi_query": 1,
 145         "nssi_with_nsi": 2
 146     }
 147
 148     request_body = request.get_json()
 149     service_count = 0
 150     for query_type, query in queries.items():
 151         if request_body == query:
 152             service_count = count[query_type]
 153     return {'results': [{'service-instance': service_count}]}
 154
 155
 156 if __name__ == "__main__":
 157     app.run(debug=True, host='0.0.0.0')