def test_check_exists():
"""Tests for check_exists"""
sim_pros_mock = mock.Mock()
callsign = types.Callsign("FAKE")
# Test error handling
sim_pros_mock.aircraft.exists.return_value = "Error"
with api.FLASK_APP.test_request_context():
resp = utils.check_exists(sim_pros_mock, callsign)
assert isinstance(resp, Response)
assert resp.status_code == HTTPStatus.INTERNAL_SERVER_ERROR
assert resp.data.decode(
) == "Could not check if the aircraft exists: Error"
# Test missing callsign
sim_pros_mock.aircraft.exists.return_value = False
with api.FLASK_APP.test_request_context():
resp = utils.check_exists(sim_pros_mock, callsign)
assert isinstance(resp, Response)
assert resp.status_code == HTTPStatus.BAD_REQUEST
assert resp.data.decode() == 'Aircraft "FAKE" does not exist'
callsign = types.Callsign("TEST")
# Test valid callsign
sim_pros_mock.aircraft.exists.return_value = True
with api.FLASK_APP.test_request_context():
resp = utils.check_exists(sim_pros_mock, callsign)
assert not resp
def get():
"""
Logic for GET events. If the request contains an identifier to an existing
aircraft, then information about its route (FMS flightplan) is returned
"""
req_args = utils.parse_args(_PARSER)
callsign = req_args[utils.CALLSIGN_LABEL]
resp = utils.check_exists(utils.sim_proxy(), callsign)
if resp:
return resp
route_info = utils.sim_proxy().aircraft.route(callsign)
if not isinstance(route_info, tuple):
if route_info == "Aircraft has no route":
return responses.bad_request_resp(route_info)
return responses.internal_err_resp(route_info)
data = {
utils.CALLSIGN_LABEL: str(callsign),
"route_name": route_info[0],
"next_waypoint": route_info[1],
"route_waypoints": route_info[2],
}
return responses.ok_resp(data)
def post():
"""
Logic for POST events. If the request contains valid aircraft information, then
a request is sent to the simulator to create it
"""
req_args = utils.parse_args(_PARSER)
callsign = req_args[utils.CALLSIGN_LABEL]
resp = utils.check_exists(utils.sim_proxy(), callsign, negate=True)
if resp:
return resp
position_or_resp = utils.try_parse_lat_lon(req_args)
if not isinstance(position_or_resp, LatLon):
return position_or_resp
if not req_args["type"]:
return responses.bad_request_resp(
"Aircraft type must be specified")
err = utils.sim_proxy().aircraft.create(
callsign,
req_args["type"],
position_or_resp,
req_args["hdg"],
req_args["alt"],
req_args["gspd"],
)
return responses.checked_resp(err, HTTPStatus.CREATED)
def post():
"""
Logic for POST events. If the request contains an existing aircraft ID, then a
request is sent to alter its ground speed
"""
req_args = utils.parse_args(_PARSER)
callsign = req_args[utils.CALLSIGN_LABEL]
resp = utils.check_exists(utils.sim_proxy(), callsign)
if resp:
return resp
err = utils.sim_proxy().aircraft.set_ground_speed(callsign, req_args["gspd"])
return checked_resp(err)
def post():
"""
Logic for POST events. If the request contains an existing aircraft ID, then a
request is sent to alter its heading
"""
req_args = utils.parse_args(_PARSER)
callsign = req_args[utils.CALLSIGN_LABEL]
resp = utils.check_exists(utils.sim_proxy(), callsign)
if resp:
return resp
heading = req_args["hdg"]
err = utils.sim_proxy().aircraft.set_heading(callsign, heading)
return responses.checked_resp(err)
def get():
"""Logic for GET events. Returns properties for the specified aircraft"""
req_args = utils.parse_args(_PARSER)
callsign = req_args[utils.CALLSIGN_LABEL]
sim_props = utils.sim_proxy().simulation.properties
if not isinstance(sim_props, SimProperties):
return responses.internal_err_resp(sim_props)
if callsign:
resp = utils.check_exists(utils.sim_proxy(), callsign)
if resp:
return resp
props = utils.sim_proxy().aircraft.properties(callsign)
if not isinstance(props, AircraftProperties):
return internal_err_resp(props)
data = utils.convert_aircraft_props(props)
data.update({"scenario_time": sim_props.scenario_time})
return responses.ok_resp(data)
# else: get_all_properties
props = utils.sim_proxy().aircraft.all_properties
if isinstance(props, str):
return responses.internal_err_resp(
f"Couldn't get the aircraft properties: {props}")
if not props:
return responses.bad_request_resp("No aircraft in the simulation")
data = {}
for prop in props.values():
data.update(utils.convert_aircraft_props(prop))
data["scenario_time"] = sim_props.scenario_time
return responses.ok_resp(data)
def post():
"""
Requests that the specified aircraft proceeds immediately to the specified
waypoint
"""
req_args = utils.parse_args(_PARSER)
waypoint_name = req_args["waypoint"]
if not waypoint_name:
return responses.bad_request_resp(
"Waypoint name must be specified")
callsign = req_args[utils.CALLSIGN_LABEL]
resp = utils.check_exists(utils.sim_proxy(), callsign)
if resp:
return resp
err = utils.sim_proxy().aircraft.direct_to_waypoint(
callsign, waypoint_name)
return responses.checked_resp(err)