1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
# -------------------------------------------------------------------------
# Copyright (c) 2018 AT&T Intellectual Property
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
# -------------------------------------------------------------------------
#
"""
Simulators for dependencies of OSDF (e.g. HAS-API, Policy, SO-callback, etc.)
"""
import glob
from osdf.utils.interfaces import json_from_file
from flask import Flask, jsonify, request
app = Flask(__name__)
@app.route("/simulated/ERROR/<component>", methods=["GET", "POST"])
@app.route("/simulated/unhealthy/<component>", methods=["GET", "POST"])
def error_for_component(component):
"""Send an HTTP error for component"""
return jsonify({"error": "{} error".format(component)}), 503
@app.route("/simulated/healthy/<component>", methods=["GET", "POST"])
def healthy_status_for_component(component):
"""Send a health-OK response for component"""
return jsonify({"success": "Passed Health Check for Component {}".format(component)})
@app.route("/simulated/success/<component>", methods=["GET", "POST"])
def successful_call_for_component(component):
"""Send a message about successful call to component"""
return jsonify({"success": "made a call to Component: {}".format(component)})
@app.route("/simulated/oof/has-api/<path:mainpath>", methods=["GET", "POST"])
def has_api_calls(mainpath):
data, status = get_payload_for_simulated_component('has-api', mainpath)
if not status:
return jsonify(data)
return jsonify(data), 503
def get_payload_for_simulated_component(component, mainpath):
"""
Get the payload for the given path for the given component
:param component: Component we are using (e.g. HAS-API, Policy, SO-callback, etc.)
:param mainpath: path within the URL (e.g. /main/X1/y1/)
:return: Content if file exists, or else 503 error
"""
file_name = "{}/response-payloads/{}".format(component, mainpath)
data = json_from_file(file_name)
if not data:
return {"Error": "Unable to read File {}".format(file_name)}, 503
return data, None
@app.route("/simulated/policy/<sub_component>/getConfig", methods=["POST"])
def get_policies(sub_component):
"""
Get all policies for this folder
:param sub_component: The folder we are interested in (e.g. "pdp-has-vcpe-good", "pdp-has-vcpe-bad")
:return: A list of policies as a json object (each element is one policy)
"""
main_dir = "policy/response-payloads/" + sub_component
files = glob.glob("{}/*.json".format(main_dir))
return jsonify([json_from_file(x) for x in files])
if __name__ == "__main__":
app.run(debug=True)
|
