import os
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