def op(self, stage, mocker):
        op = pipeline_ops_provisioning.PollStatusOperation(
            fake_operation_id, " ", callback=mocker.MagicMock())
        yield op

        # Clean up any timers set on it
        if op.polling_timer:
            op.polling_timer.cancel()
        if op.retry_after_timer:
            op.retry_after_timer.cancel()
        if op.provisioning_timeout_timer:
            op.provisioning_timeout_timer.cancel()
 def send_query_op(self, mocker):
     op = pipeline_ops_provisioning.PollStatusOperation(
         fake_operation_id, " ", callback=mocker.MagicMock())
     return op