test_content_handling

Complexity

Total Complexity

15

Size/Duplication

Total Lines	102
Duplicated Lines	0 %

Importance

Changes

0

11 Functions

Rating	Name	Duplication	Size	Complexity
A	ds_handler()	0	3	1
A	test_get_profiles()	0	8	1
A	test_get_data_streams()	0	2	1
A	test_get_checklists()	0	5	1
A	test_get_data_streams_checklists()	0	5	1
A	test_identify_files()	0	12	1
A	test_get_checklists_invalid()	0	4	2
A	test_init_not_scap_content()	0	4	2
A	test_init_xccdf_content()	0	4	2
A	test_get_profiles_with_default()	0	8	1
A	test_init_invalid_file_path()	0	4	2

import os
import glob

import pytest

from org_fedora_oscap import content_handling as ch


TESTING_FILES_PATH = os.path.join(
    os.path.dirname(__file__), os.path.pardir, "testing_files")
DS_FILEPATH = os.path.join(
    TESTING_FILES_PATH, "testing_ds.xml")

DS_IDS = "scap_org.open-scap_datastream_tst"
CHK_FIRST_ID = "scap_org.open-scap_cref_first-xccdf.xml"
CHK_SECOND_ID = "scap_org.open-scap_cref_second-xccdf.xml"

PROFILE1_ID = "xccdf_com.example_profile_my_profile"
PROFILE2_ID = "xccdf_com.example_profile_my_profile2"
PROFILE3_ID = "xccdf_com.example_profile_my_profile3"


@pytest.fixture()
def ds_handler():
    return ch.DataStreamHandler(DS_FILEPATH)


def test_init_invalid_file_path():
    with pytest.raises(ch.DataStreamHandlingError) as excinfo:
        ch.DataStreamHandler("testing_ds.xmlll")
    assert "Invalid file path" in str(excinfo.value)


def test_init_not_scap_content():
    with pytest.raises(ch.DataStreamHandlingError) as excinfo:
        ch.DataStreamHandler(os.path.join(TESTING_FILES_PATH, "testing_ks.cfg"))
    assert "not a valid SCAP content file" in str(excinfo.value)


def test_init_xccdf_content():
    with pytest.raises(ch.DataStreamHandlingError) as excinfo:
        ch.DataStreamHandler(os.path.join(TESTING_FILES_PATH, "xccdf.xml"))
    assert "not a data stream collection" in str(excinfo.value)


def test_get_data_streams(ds_handler):
    assert DS_IDS in ds_handler.get_data_streams()


def test_get_data_streams_checklists(ds_handler):
    expected_ids = {DS_IDS: [CHK_FIRST_ID, CHK_SECOND_ID]}

    ds_ids = ds_handler.get_data_streams_checklists()
    assert expected_ids == ds_ids


def test_get_checklists(ds_handler):
    expected_checklists = [CHK_FIRST_ID, CHK_SECOND_ID]

    chk_ids = ds_handler.get_checklists(DS_IDS)
    assert expected_checklists == chk_ids


def test_get_checklists_invalid(ds_handler):
    with pytest.raises(ch.DataStreamHandlingError) as excinfo:
        ds_handler.get_checklists("invalid.id")
        assert "Invalid data stream id given" in str(excinfo.value)


def test_get_profiles(ds_handler):
    profile_ids = ds_handler.get_profiles(DS_IDS, CHK_FIRST_ID)

    # When Benchmark doesn't contain Rules selected by default
    # the default Profile should not be present
    assert 2 == len(profile_ids)
    assert PROFILE1_ID == profile_ids[0].id
    assert PROFILE2_ID == profile_ids[1].id


def test_get_profiles_with_default(ds_handler):
    profile_ids = ds_handler.get_profiles(DS_IDS, CHK_SECOND_ID)

    # When Benchmark contains Rules selected by default
    # the default Profile should be present
    assert 2 == len(profile_ids)
    assert "default" == profile_ids[0].id
    assert PROFILE3_ID == profile_ids[1].id


def test_identify_files():
    filenames = glob.glob(TESTING_FILES_PATH + "/*")
    identified = ch.identify_files(filenames)
    assert identified[DS_FILEPATH] == ch.CONTENT_TYPES["DATASTREAM"]
    assert identified[
        os.path.join(TESTING_FILES_PATH, "scap-mycheck-oval.xml")] == ch.CONTENT_TYPES["OVAL"]
    assert identified[
        os.path.join(TESTING_FILES_PATH, "tailoring.xml")] == ch.CONTENT_TYPES["TAILORING"]
    assert identified[
        os.path.join(TESTING_FILES_PATH, "testing_xccdf.xml")] == ch.CONTENT_TYPES["XCCDF_CHECKLIST"]
    assert identified[
        os.path.join(TESTING_FILES_PATH, "cpe-dict.xml")] == ch.CONTENT_TYPES["CPE_DICT"]