oscap.py

from pathlib import PurePath

from fauxfactory import gen_string
from nailgun import entities
import pytest

from robottelo.config import robottelo_tmp_dir, settings
from robottelo.constants import OSCAP_PROFILE, OSCAP_TAILORING_FILE, DataFile


@pytest.fixture(scope="session")
def tailoring_file_path(session_target_sat):
    """Return Tailoring file path."""
    local = DataFile.OSCAP_TAILORING_FILE
    session_target_sat.put(
        local_path=local,
        remote_path=f'/tmp/{OSCAP_TAILORING_FILE}',
    )
    return {'local': local, 'satellite': f'/tmp/{OSCAP_TAILORING_FILE}'}


@pytest.fixture(scope="session")
def oscap_content_path(session_target_sat):
    """Download scap content from satellite and return local path of it."""
    local_file = robottelo_tmp_dir.joinpath(PurePath(settings.oscap.content_path).name)
    session_target_sat.get(remote_path=settings.oscap.content_path, local_path=str(local_file))
    return local_file


@pytest.fixture(scope="module")
def scap_content(import_ansible_roles, module_target_sat):
    title = f"rhel-content-{gen_string('alpha')}"
    scap_info = module_target_sat.cli_factory.scapcontent(
        {'title': title, 'scap-file': f'{settings.oscap.content_path}'}
    )
    scap_id = scap_info['id']
    scap_info = entities.ScapContents(id=scap_id).read()

    scap_profile_id = [
        profile['id']
        for profile in scap_info.scap_content_profiles
        if OSCAP_PROFILE[settings.oscap.profile] in profile['title']
    ][0]
    return {
        "title": title,
        "scap_id": scap_id,
        "scap_profile_id": scap_profile_id,
    }


@pytest.fixture(scope="module")
def tailoring_file(module_org, module_location, tailoring_file_path):
    """Create Tailoring file."""
    tailoring_file_name = f"tailoring-file-{gen_string('alpha')}"
    tf_info = entities.TailoringFile(
        name=f"{tailoring_file_name}",
        scap_file=f"{tailoring_file_path['local']}",
        organization=[module_org],
        location=[module_location],
    ).create()
    return {
        "name": tailoring_file_name,
        "tailoring_file_id": tf_info.id,
        "tailoring_file_profile_id": tf_info.tailoring_file_profiles[0]['id'],
    }