def security_stage(mocker, arbitrary_exception, arbitrary_base_exception):
return make_mock_stage(
mocker=mocker,
stage_to_make=pipeline_stages_provisioning.UseSecurityClientStage,
exc_to_raise=arbitrary_exception,
base_exc_to_raise=arbitrary_base_exception,
)
def stage(self, mocker, arbitrary_exception, arbitrary_base_exception):
return make_mock_stage(
mocker=mocker,
stage_to_make=pipeline_stages_iothub.HandleTwinOperationsStage,
exc_to_raise=arbitrary_exception,
base_exc_to_raise=arbitrary_base_exception,
)
def stage(self, mocker, arbitrary_exception, arbitrary_base_exception):
return make_mock_stage(
mocker=mocker,
stage_to_make=cls,
exc_to_raise=arbitrary_exception,
base_exc_to_raise=arbitrary_base_exception,
)
def stage(self, mocker, arbitrary_exception, arbitrary_base_exception):
return make_mock_stage(
mocker=mocker,
stage_to_make=pipeline_stages_iothub.UseAuthProviderStage,
exc_to_raise=arbitrary_exception,
base_exc_to_raise=arbitrary_base_exception,
)
def stage(mocker, arbitrary_exception, arbitrary_base_exception):
return make_mock_stage(
mocker=mocker,
stage_to_make=MockPipelineStage,
exc_to_raise=arbitrary_exception,
base_exc_to_raise=arbitrary_base_exception,
)
def stage(self, mocker):
stage = make_mock_stage(mocker,
pipeline_stages_iothub.UseAuthProviderStage)
auth_provider = mocker.MagicMock()
auth_provider.get_current_sas_token = mocker.MagicMock(
return_value=fake_sas_token)
stage.auth_provider = auth_provider
return stage
def mock_stage(mocker, arbitrary_exception, arbitrary_base_exception):
return make_mock_stage(
mocker=mocker,
stage_to_make=pipeline_stages_provisioning_mqtt.
ProvisioningMQTTConverterStage,
exc_to_raise=arbitrary_exception,
base_exc_to_raise=arbitrary_base_exception,
)
def stage(self, mocker, arbitrary_exception, arbitrary_base_exception):
return make_mock_stage(
mocker=mocker,
stage_to_make=pipeline_stages_base.
CoordinateRequestAndResponseStage,
exc_to_raise=arbitrary_exception,
base_exc_to_raise=arbitrary_base_exception,
)
def stage(self, mocker, arbitrary_exception, arbitrary_base_exception):
stage = make_mock_stage(
mocker=mocker,
stage_to_make=pipeline_stages_base.SerializeConnectOpsStage,
exc_to_raise=arbitrary_exception,
base_exc_to_raise=arbitrary_base_exception,
)
stage.next.run_op = mocker.MagicMock()
return stage
def stage(self, mocker, arbitrary_exception, arbitrary_base_exception):
stage = make_mock_stage(
mocker=mocker,
stage_to_make=pipeline_stages_iothub.UseAuthProviderStage,
exc_to_raise=arbitrary_exception,
base_exc_to_raise=arbitrary_base_exception,
)
auth_provider = mocker.MagicMock()
auth_provider.get_current_sas_token = mocker.MagicMock(return_value=fake_sas_token)
stage.auth_provider = auth_provider
return stage
def stage(mocker):
return make_mock_stage(
mocker, pipeline_stages_iothub_mqtt.IoTHubMQTTConverterStage)
def mock_stage(mocker):
return make_mock_stage(
mocker,
pipeline_stages_provisioning_mqtt.ProvisioningMQTTConverterStage)
def stage(self, mocker):
return make_mock_stage(mocker=mocker, stage_to_make=cls)
def stage(self, mocker):
return make_mock_stage(
mocker, pipeline_stages_base.CoordinateRequestAndResponseStage)
def stage(self, mocker):
stage = make_mock_stage(
mocker=mocker,
stage_to_make=pipeline_stages_base.SerializeConnectOpsStage)
stage.next.run_op = mocker.MagicMock()
return stage
def security_stage(mocker):
return make_mock_stage(mocker,
pipeline_stages_provisioning.UseSecurityClientStage)
def stage(mocker):
return make_mock_stage(mocker, MockPipelineStage)
def stage(self, mocker):
return make_mock_stage(mocker, pipeline_stages_iothub.HandleTwinOperationsStage)
def stage(self, mocker):
return make_mock_stage(mocker, pipeline_stages_iothub.UseAuthProviderStage)
