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 from flask import Flask, jsonify, request
  26
  27 from osdf.utils.interfaces import json_from_file
  28
  29 app = Flask(__name__)
  30
  31
  32 @app.route("/simulated/ERROR/<component>", methods=["GET", "POST"])
  33 @app.route("/simulated/unhealthy/<component>", methods=["GET", "POST"])
  34 def error_for_component(component):
  35     """Send an HTTP error for component"""
  36     return jsonify({"error": "{} error".format(component)}), 503
  37
  38
  39 @app.route("/simulated/healthy/<component>", methods=["GET", "POST"])
  40 def healthy_status_for_component(component):
  41     """Send a health-OK response for component"""
  42     return jsonify({"success": "Passed Health Check for Component {}".format(component)})
  43
  44
  45 @app.route("/simulated/success/<component>", methods=["GET", "POST"])
  46 def successful_call_for_component(component):
  47     """Send a message about successful call to component"""
  48     return jsonify({"success": "made a call to Component: {}".format(component)})
  49
  50
  51 @app.route("/simulated/oof/has-api/<path:mainpath>", methods=["GET", "POST"])
  52 def has_api_calls(mainpath):
  53     data, status = get_payload_for_simulated_component('has-api', mainpath)
  54     if not status:
  55         return jsonify(data)
  56     return jsonify(data), 503
  57
  58
  59 def get_payload_for_simulated_component(component, mainpath):
  60     """
  61     Get the payload for the given path for the given component
  62     :param component: Component we are using (e.g. HAS-API, Policy, SO-callback, etc.)
  63     :param mainpath: path within the URL (e.g. /main/X1/y1/)
  64     :return: Content if file exists, or else 503 error
  65     """
  66     file_name = "{}/response-payloads/{}".format(component, mainpath)
  67     data = json_from_file(file_name)
  68     if not data:
  69         return {"Error": "Unable to read File {}".format(file_name)}, 503
  70     return data, None
  71
  72
  73 @app.route("/simulated/policy/<sub_component>/getConfig", methods=["POST"])
  74 def get_policies(sub_component):
  75     """
  76     Get all policies for this folder
  77     :param sub_component: The folder we are interested in (e.g. "pdp-has-vcpe-good", "pdp-has-vcpe-bad")
  78     :return: A list of policies as a json object (each element is one policy)
  79     """
  80     main_dir = "policy/response-payloads/" + sub_component
  81     files = glob.glob("{}/*.json".format(main_dir))
  82     list_json = []
  83     for x in files:
  84         list_json.append({
  85             "policyConfigMessage": "Config Retrieved! ",
  86             "policyConfigStatus": "CONFIG_RETRIEVED",
  87             "type": "JSON",
  88             "config": json.dumps(json_from_file(x)),
  89             "policyName": os.path.basename(x),
  90             "policyType": "MicroService",
  91             "policyVersion": "1"
  92         })
  93     return jsonify(list_json)
  94
  95
  96 @app.route("/simulated/configdb/getCellList", methods=["GET"])
  97 def get_cell_list():
  98     data, status = get_payload_for_simulated_component('configdb',
  99                                                        'getCellList-' + request.args.get('networkId') + '.json')
 100     if not status:
 101         return jsonify(data)
 102     return jsonify(data), 503
 103
 104
 105 @app.route("/simulated/configdb/getNbrList", methods=["GET"])
 106 def get_nbr_list():
 107     data, status = get_payload_for_simulated_component('configdb', 'getNbrList-' + request.args.get('cellId') + '.json')
 108     if not status:
 109         return jsonify(data)
 110     return jsonify(data), 503
 111
 112
 113 if __name__ == "__main__":
 114     app.run(debug=True, host='0.0.0.0')