def test_dialog_flow_get_parameter(app):
# ensure a deserialized payload is considered valid
with app.test_request_context(json={
"queryResult": {
"parameters": {
"bpm": 1,
"min": 2,
"max": 3
}
}
}):
dialog = DialogFlowRequest(request)
assert dialog.get_parameter("bpm") == 1
assert dialog.get_parameter("min") == 2
assert dialog.get_parameter("max") == 3
def test_dialog_flow_validation(app):
# ensure a deserialized payload is considered valid
with app.test_request_context(json={
"queryResult": {
"parameters": {
"bpm": 42,
"min": 42,
"max": 42
}
}
}):
assert DialogFlowRequest(request).validate() is True
def test_dialog_flow_init(app):
# ensure a DialogFlow class can parse Flask Request instance
with app.test_request_context(json={"key": "value"}):
dialog = DialogFlowRequest(request)
assert dialog._data == {"key": "value"}
def test_dialog_flow_validation_empty(app):
# ensure an empty payload raises a BadRequest exception
with app.test_request_context():
with pytest.raises(BadRequest):
DialogFlowRequest(request).validate()
def test_dialog_flow_validation_error(app):
# ensure a deserialized payload without all the values raises an Exception
with app.test_request_context(json={"queryResult": {"parameters": {}}}):
with pytest.raises(BadRequest):
DialogFlowRequest(request).validate()
def test_dialog_flow_init_malformed_json(app):
# ensure a DialogFlow class can parse Flask Request instance
with app.test_request_context(data="not_json_data",
content_type="application/json"):
dialog = DialogFlowRequest(request)
assert dialog._data is None
def test_dialog_flow_init_malformed(app):
# ensure a DialogFlow class can parse Flask Request instance
with app.test_request_context(data="some_value"):
dialog = DialogFlowRequest(request)
assert dialog._data is None
def entrypoint(request):
"""Cloud Function entrypoint.
Args:
request: Flask Request object
Returns:
The response text or any set of values that can be turned into a
Response object using
`make_response <http://flask.pocoo.org/docs/1.0/api/#flask.Flask.make_response>`.
"""
# Allow only POST methods
if request.method != "POST":
return abort(405)
try:
# Validate Environment Configuration
hev.conf.validate()
hev.auth.is_authorized(request, hev.conf.bearer_token)
# Validate Request Object
dialog = DialogFlowRequest(request)
dialog.validate()
except ConfigException as e:
logging.critical("Unable to configure Cloud Function: %s", str(e))
response = json.dumps({"message": "Configuration error"})
return (response, 500)
except NotAuthorized as e:
# TODO: update the response so that it can be parsed from a
# Google Action
logging.critical(str(e))
response = json.dumps({"message": "Not Authorized"})
return (response, 401)
except BadRequest as e:
response = json.dumps({"message": str(e)})
logging.critical(response)
return (response, 400)
# Prepare the API
api = DatadogAPI(hev.conf.dd_api_key, hev.conf.function_name,
hev.conf.dry_run)
# Get values from DialogFlow request
bpm = dialog.get_parameter("bpm")
value_min = dialog.get_parameter("min")
value_max = dialog.get_parameter("max")
# Send HEV parameters to Datadog
op_1 = api.send_bpm(bpm)
op_2 = api.send_pressure(value_min, kind=KIND_DIASTOLIC)
op_3 = api.send_pressure(value_max, kind=KIND_SYSTOLIC)
# Check all calls were a success
if op_1 and op_2 and op_3:
logging.info("Cloud Function executed correctly.")
response = json.dumps({"message": "Success"})
status = 201
else:
logging.error("Cloud Function executed with errors.")
response = json.dumps({"message": "Failed"})
status = 503
return (response, status)