Source code for stepler.cli_clients.fixtures.glance

"""
--------------------------
Glance CLI client fixtures
--------------------------
"""

# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
# implied.
# See the License for the specific language governing permissions and
# limitations under the License.

import pytest

from stepler.cli_clients import steps

__all__ = [
    'cli_glance_steps',
    'cli_download_image',
]


@pytest.fixture
def cli_glance_steps(remote_executor):
    """Function fixture to glance CLI steps.

    Args:
        remote_executor (callable): function to execute command on
            controller nodes

    Returns:
        CliGlanceSteps: instantiated glance CLI steps
    """
    return steps.CliGlanceSteps(remote_executor)


@pytest.fixture
def cli_download_image(nova_api_node,
                       os_faults_steps,
                       cli_glance_steps):
    """Callable function fixture to download image via CLI.

    Args:
        nova_api_node (obj): controller (node with nova-api service)
        os_faults_steps (OsFaultsSteps): instantiated os-faults steps
        cli_glance_steps (CliGlanceSteps): instantiated glance CLI steps

    Returns:
        function: function to download image via CLI.
    """
    def _download_image(*args, **kwgs):
        remote_path = cli_glance_steps.download_image(*args, **kwgs)
        return os_faults_steps.download_file(nova_api_node, remote_path)

    return _download_image