def on_sas_token_updated(self):
logger.info(
"{}: New sas token received. Passing down UpdateSasTokenOperation."
.format(self.name))
@pipeline_thread.runs_on_pipeline_thread
def on_token_update_complete(op):
if op.error:
logger.error(
"{}({}): token update operation failed. Error={}".format(
self.name, op.name, op.error))
unhandled_exceptions.exception_caught_in_background_thread(
op.error)
else:
logger.debug(
"{}({}): token update operation is complete".format(
self.name, op.name))
operation_flow.pass_op_to_next_stage(
stage=self,
op=pipeline_ops_base.UpdateSasTokenOperation(
sas_token=self.auth_provider.get_current_sas_token(),
callback=on_token_update_complete,
),
)
def on_sas_token_updated(self):
logger.info(
"{}: New sas token received. Passing down UpdateSasTokenOperation.".format(self.name)
)
@pipeline_thread.runs_on_pipeline_thread
def on_token_update_complete(op, error):
if error:
logger.error(
"{}({}): token update operation failed. Error={}".format(
self.name, op.name, error
)
)
handle_exceptions.handle_background_exception(error)
else:
logger.debug(
"{}({}): token update operation is complete".format(self.name, op.name)
)
self.send_op_down(
op=pipeline_ops_base.UpdateSasTokenOperation(
sas_token=self.auth_provider.get_current_sas_token(),
callback=on_token_update_complete,
)
)
def test_mqtt_publish(self, mocker, stage, create_transport):
cb = mocker.MagicMock()
op_update_sas_token = pipeline_ops_base.UpdateSasTokenOperation(
sas_token=fake_sas_token, callback=cb)
stage.run_op(op_update_sas_token)
assert_callback_succeeded(op_update_sas_token)
assert stage.sas_token == fake_sas_token
def test_mqtt_publish(self, mocker, stage, create_transport):
cb = mocker.MagicMock()
op_update_sas_token = pipeline_ops_base.UpdateSasTokenOperation(
sas_token=fake_sas_token, callback=cb)
mocker.spy(op_update_sas_token, "complete")
stage.run_op(op_update_sas_token)
assert op_update_sas_token.complete.call_count == 1
assert op_update_sas_token.complete.call_args == mocker.call()
assert stage.sas_token == fake_sas_token
def op(self, mocker):
return pipeline_ops_base.UpdateSasTokenOperation(
sas_token="new_fake_sas_token", callback=mocker.MagicMock())
