def _execute_op(self, op):
def map_twin_error(original_op, twin_op):
if twin_op.error:
original_op.error = twin_op.error
elif twin_op.status_code >= 300:
# TODO map error codes to correct exceptions
logger.error("Error {} received from twin operation".format(
twin_op.status_code))
logger.error("response body: {}".format(twin_op.response_body))
original_op.error = Exception(
"twin operation returned status {}".format(
twin_op.status_code))
if isinstance(op, pipeline_ops_iothub.GetTwinOperation):
def on_twin_response(twin_op):
logger.info("{}({}): Got response for GetTwinOperation".format(
self.name, op.name))
map_twin_error(original_op=op, twin_op=twin_op)
if not twin_op.error:
op.twin = json.loads(twin_op.response_body.decode("utf-8"))
operation_flow.complete_op(self, op)
operation_flow.pass_op_to_next_stage(
self,
pipeline_ops_base.SendIotRequestAndWaitForResponseOperation(
request_type=constant.TWIN,
method="GET",
resource_location="/",
request_body=" ",
callback=on_twin_response,
),
)
elif isinstance(
op, pipeline_ops_iothub.PatchTwinReportedPropertiesOperation):
def on_twin_response(twin_op):
logger.info(
"{}({}): Got response for PatchTwinReportedPropertiesOperation operation"
.format(self.name, op.name))
map_twin_error(original_op=op, twin_op=twin_op)
operation_flow.complete_op(self, op)
logger.info("{}({}): Sending reported properties patch: {}".format(
self.name, op.name, op.patch))
operation_flow.pass_op_to_next_stage(
self,
(pipeline_ops_base.SendIotRequestAndWaitForResponseOperation(
request_type=constant.TWIN,
method="PATCH",
resource_location="/properties/reported/",
request_body=json.dumps(op.patch),
callback=on_twin_response,
)),
)
else:
operation_flow.pass_op_to_next_stage(self, op)
def _execute_op(self, op):
def map_twin_error(error, twin_op):
if error:
return error
elif twin_op.status_code >= 300:
# TODO map error codes to correct exceptions
logger.error("Error {} received from twin operation".format(
twin_op.status_code))
logger.error("response body: {}".format(twin_op.response_body))
return exceptions.ServiceError(
"twin operation returned status {}".format(
twin_op.status_code))
if isinstance(op, pipeline_ops_iothub.GetTwinOperation):
def on_twin_response(twin_op, error):
logger.debug(
"{}({}): Got response for GetTwinOperation".format(
self.name, op.name))
error = map_twin_error(error=error, twin_op=twin_op)
if not error:
op.twin = json.loads(twin_op.response_body.decode("utf-8"))
self._complete_op(op, error=error)
self._send_op_down(
pipeline_ops_base.SendIotRequestAndWaitForResponseOperation(
request_type=constant.TWIN,
method="GET",
resource_location="/",
request_body=" ",
callback=on_twin_response,
))
elif isinstance(
op, pipeline_ops_iothub.PatchTwinReportedPropertiesOperation):
def on_twin_response(twin_op, error):
logger.debug(
"{}({}): Got response for PatchTwinReportedPropertiesOperation operation"
.format(self.name, op.name))
error = map_twin_error(error=error, twin_op=twin_op)
self._complete_op(op, error=error)
logger.debug(
"{}({}): Sending reported properties patch: {}".format(
self.name, op.name, op.patch))
self._send_op_down(
pipeline_ops_base.SendIotRequestAndWaitForResponseOperation(
request_type=constant.TWIN,
method="PATCH",
resource_location="/properties/reported/",
request_body=json.dumps(op.patch),
callback=on_twin_response,
))
else:
self._send_op_down(op)
def make_fake_request_and_response(mocker):
return pipeline_ops_base.SendIotRequestAndWaitForResponseOperation(
request_type=fake_request_type,
method=fake_method,
resource_location=fake_resource_location,
request_body=fake_request_body,
callback=mocker.MagicMock(),
)