def mock_provisioning_pipeline(mocker, input_security_client, mock_mqtt_transport): provisioning_pipeline = ProvisioningPipeline(input_security_client) provisioning_pipeline.on_connected = mocker.MagicMock() provisioning_pipeline.on_disconnected = mocker.MagicMock() provisioning_pipeline.on_message_received = mocker.MagicMock() helpers.add_mock_method_waiter(provisioning_pipeline, "on_connected") helpers.add_mock_method_waiter(provisioning_pipeline, "on_disconnected") helpers.add_mock_method_waiter(mock_mqtt_transport, "publish") yield provisioning_pipeline provisioning_pipeline.disconnect()
def mock_provisioning_pipeline(params_security_clients): input_security_client = params_security_clients["client_class"]( **params_security_clients["init_kwargs"]) with patch( "azure.iot.device.provisioning.pipeline.provisioning_pipeline.pipeline_stages_mqtt.MQTTTransport" ): provisioning_pipeline = ProvisioningPipeline(input_security_client) provisioning_pipeline.on_connected = MagicMock() provisioning_pipeline.on_disconnected = MagicMock() provisioning_pipeline.on_message_received = MagicMock() helpers.add_mock_method_waiter(provisioning_pipeline, "on_connected") helpers.add_mock_method_waiter(provisioning_pipeline, "on_disconnected") helpers.add_mock_method_waiter(provisioning_pipeline._pipeline.transport, "publish") yield provisioning_pipeline provisioning_pipeline.disconnect()
def op(self, op_cls, callback): op = make_mock_op_or_event(op_cls) op.callback = callback op.action = "pend" add_mock_method_waiter(op, "callback") return op
def op(self, request, mocker): op = make_mock_op_or_event(request.param) op.callback = mocker.MagicMock() add_mock_method_waiter(op, "callback") return op
def op(self, op_cls, mocker): op = make_mock_op_or_event(op_cls) op.callback = mocker.MagicMock() add_mock_method_waiter(op, "callback") return op