def test_check_bin(self):
self.test_ini['binary_req'] = 'cpf'
self.loaded_ini[self.filename] = []
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dirname, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
self.assertTrue(self._return_check('check_rpm_to "" "cpf"'))
def test_missing_tag_check_script(self):
"""Basic test for whole program"""
self.test_ini.pop('check_script', None)
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingTagsIniFileError,
lambda: list(self.xml_utils.prepare_sections()))
def test_check_rpm(self):
self.test_ini['requires'] = 'test_rpm'
self.loaded_ini[self.filename] = []
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dirname, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
self.assertTrue(self._return_check('check_rpm_to "test_rpm" ""'))
def test_xml_not_migrate_not_upgrade(self):
test_ini = {
'content_title': 'Testing only migrate title',
'content_description': ' some content description',
'author': 'test <*****@*****.**>',
'config_file': '/etc/named.conf',
'check_script': self.check_script,
'solution': self.test_solution,
'applies_to': 'test',
'requires': 'bash',
'binary_req': 'sed'
}
ini = {}
old_settings = settings.UPGRADE_PATH
migrate, upgrade = self._create_temporary_dir()
ini[self.filename] = []
ini[self.filename].append(test_ini)
xml_utils = XmlUtils(self.dirname, ini)
xml_utils.prepare_sections()
migrate_file = FileHelper.get_file_content(migrate, 'rb', method=True)
tag = [
x.strip() for x in migrate_file
if 'xccdf_preupg_rule_test_check_script' in x.strip()
]
self.assertIsNotNone(tag)
upgrade_file = FileHelper.get_file_content(upgrade, 'rb', method=True)
tag = [
x.strip() for x in upgrade_file
if 'xccdf_preupg_rule_test_check_script' in x.strip()
]
self.assertIsNotNone(tag)
self._delete_temporary_dir(migrate, upgrade)
settings.UPGRADE_PATH = old_settings
def test_file_solution_not_exists(self):
"""Test of missing 'solution' file - SystemExit should be raised"""
self.test_ini['solution'] = "this_should_be_unexpected_file.txt"
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingTagsIniFileError,
lambda: list(self.xml_utils.prepare_sections()))
def test_missing_tag_solution_script(self):
"""Test of missing tag 'solution' - SystemExit should be raised"""
self.test_ini.pop('solution', None)
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingTagsIniFileError,
lambda: list(self.xml_utils.prepare_sections()))
def test_file_check_script_not_exists(self):
"""Test of missing 'check_script' file"""
self.test_ini['check_script'] = "this_should_be_unexpected_file.txt"
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingFileInContentError,
lambda: list(self.xml_utils.prepare_sections()))
def test_xml_upgrade_not_migrate(self):
test_ini = {'content_title': 'Testing only migrate title',
'content_description': ' some content description',
'author': 'test <*****@*****.**>',
'config_file': '/etc/named.conf',
'applies_to': 'test',
'requires': 'bash',
'binary_req': 'sed',
'mode': 'upgrade'}
ini = {}
old_settings = settings.UPGRADE_PATH
migrate, upgrade = self._create_temporary_dir()
ini[self.filename] = test_ini
xml_utils = XmlUtils(self.root_dir_name, self.dirname, ini)
xml_utils.prepare_sections()
upgrade_file = FileHelper.get_file_content(upgrade, 'rb', method=True)
tag = [x.strip() for x in upgrade_file if 'xccdf_preupg_rule_test_check_script' in x.strip()]
self.assertIsNotNone(tag)
try:
migrate_file = FileHelper.get_file_content(migrate, 'rb', method=True)
except IOError:
migrate_file = None
self.assertIsNone(migrate_file)
self._delete_temporary_dir(migrate, upgrade)
settings.UPGRADE_PATH = old_settings
def test_applies_to(self):
self.test_ini['applies_to'] = 'test_rpm'
self.loaded_ini[self.filename] = self.test_ini
self.xml_utils = XmlUtils(self.root_dir_name, self.dirname,
self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
self.assertTrue(self._return_check('check_applies_to "test_rpm"'))
def test_check_rpm_bin(self):
self.test_ini['binary_req'] = 'cpf'
self.test_ini['requires'] = 'test_rpm'
self.loaded_ini[self.filename] = self.test_ini
self.xml_utils = XmlUtils(self.root_dir_name, self.dirname,
self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
self.assertTrue(self._return_check('check_rpm_to "test_rpm" "cpf"'))
def test_xml_solution_type_text(self):
self.xml_utils = XmlUtils(self.root_dir_name, self.dirname,
self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
fix_text = [
x for x in self.rule
if "<fixtext>_test_SOLUTION_MSG</fixtext>" in x
]
self.assertTrue(fix_text)
def test_xml_solution_type_html(self):
self.loaded_ini[self.filename][0]['solution_type'] = "html"
self.xml_utils = XmlUtils(self.dirname, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
fix_text = [
x for x in self.rule
if "<fixtext>_test_SOLUTION_MSG_HTML</fixtext>" in x
]
self.assertTrue(fix_text)
def test_check_script_is_directory(self):
"""
Directory as input instead of regular file is incorrect input
Tests issue #29
"""
self.test_ini['check_script'] = '.'
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingTagsIniFileError,
lambda: list(self.xml_utils.prepare_sections()))
def test_secret_check_script(self):
"""Check occurrence of secret file for check script"""
self.test_ini['check_script'] = '.minicheck'
text = """#!/usr/bin/sh\necho 'ahojky'\n"""
FileHelper.write_to_file(
os.path.join(self.dir_name, self.check_script), "wb", text)
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingFileInContentError,
lambda: list(self.xml_utils.prepare_sections()))
def write_xml(self):
"""The function is used for storing a group.xml file"""
self.find_all_ini()
self.write_list_rules()
xml_utils = XmlUtils(self.dirname, self.loaded)
self.rule = xml_utils.prepare_sections()
file_name = os.path.join(self.dirname, "group.xml")
try:
FileHelper.write_to_file(file_name, "wb", ["%s" % item for item in self.rule])
except IOError as ior:
print ('Problem with write data to the file ', file_name, ior.message)
def write_xml(self):
"""The function is used for storing a group.xml file"""
self.find_all_ini()
self.write_list_rules()
xml_utils = XmlUtils(self.module_set_dir, self.dirname, self.loaded)
self.rule = xml_utils.prepare_sections()
file_name = os.path.join(self.dirname, "group.xml")
try:
FileHelper.write_to_file(file_name, "wb",
["%s" % item for item in self.rule])
except IOError as ior:
raise IOError('Problem with writing to file %s.\nDetails: %s' %
(file_name, ior.message))
def test_group_ini(self):
"""Basic test creation group.xml file"""
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
group_tag = [
x for x in self.rule if
'<Group id="xccdf_preupg_group_test_group" selected="true">' in x
]
self.assertTrue(group_tag)
title_tag = [
x for x in self.rule if '<title>Testing content title</title>' in x
]
self.assertTrue(title_tag)
def test_check_rpm(self):
self.test_ini['requires'] = 'test_rpm'
self.loaded_ini[self.filename] = self.test_ini
self.xml_utils = XmlUtils(self.root_dir_name, self.dirname,
self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
self.assertTrue(self._return_check('check_rpm_to "test_rpm" ""'))
class TestGroupXML(base.TestCase):
"""Basic test for creating group.xml file"""
dir_name = None
filename = None
rule = []
loaded_ini = {}
xml_utils = None
def setUp(self):
self.root_dir_name = "tests/FOOBAR6_7"
self.dir_name = os.path.join(self.root_dir_name, "test_group")
os.makedirs(self.dir_name)
self.filename = os.path.join(self.dir_name, 'group.ini')
test_ini = {'group_title': 'Testing content title'}
self.assertTrue(test_ini)
self.loaded_ini[self.filename] = test_ini
def tearDown(self):
shutil.rmtree(self.dir_name)
def test_group_ini(self):
"""Basic test creation group.xml file"""
self.xml_utils = XmlUtils(self.root_dir_name, self.dir_name,
self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
group_tag = [x for x in self.rule if '<Group id="xccdf_preupg_group_test_group" selected="true">' in x]
self.assertTrue(group_tag)
title_tag = [x for x in self.rule if '<title>Testing content title</title>' in x]
self.assertTrue(title_tag)
def test_applies_to(self):
self.test_ini['applies_to'] = 'test_rpm'
self.loaded_ini[self.filename] = []
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dirname, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
self.assertTrue(self._return_check('check_applies_to "test_rpm"'))
def test_check_bin(self):
self.test_ini['binary_req'] = 'cpf'
self.loaded_ini[self.filename] = self.test_ini
self.xml_utils = XmlUtils(self.root_dir_name, self.dirname,
self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
self.assertTrue(self._return_check('check_rpm_to "" "cpf"'))
class TestGroupXML(base.TestCase):
"""Basic test for creating group.xml file"""
dir_name = None
filename = None
rule = []
loaded_ini = {}
xml_utils = None
def setUp(self):
self.dir_name = "tests/FOOBAR6_7-results/test_group"
os.makedirs(self.dir_name)
self.filename = os.path.join(self.dir_name, 'group.ini')
test_ini = {'group_title': 'Testing content title'}
self.assertTrue(test_ini)
self.loaded_ini[self.filename] = []
self.loaded_ini[self.filename].append(test_ini)
def tearDown(self):
shutil.rmtree(self.dir_name)
def test_group_ini(self):
"""Basic test creation group.xml file"""
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
group_tag = [
x for x in self.rule if
'<Group id="xccdf_preupg_group_test_group" selected="true">' in x
]
self.assertTrue(group_tag)
title_tag = [
x for x in self.rule if '<title>Testing content title</title>' in x
]
self.assertTrue(title_tag)
def test_secret_check_script(self):
"""Check occurrence of secret file for check script"""
self.test_ini['check_script'] = '.minicheck'
text = """#!/usr/bin/sh\necho 'ahojky'\n"""
FileHelper.write_to_file(os.path.join(self.dir_name, self.check_script), "wb", text)
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingFileInContentError, lambda: list(self.xml_utils.prepare_sections()))
def test_group_ini(self):
"""Basic test creation group.xml file"""
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
group_tag = [x for x in self.rule if '<Group id="xccdf_preupg_group_test_group" selected="true">' in x]
self.assertTrue(group_tag)
title_tag = [x for x in self.rule if '<title>Testing content title</title>' in x]
self.assertTrue(title_tag)
def test_check_rpm_bin(self):
self.test_ini['binary_req'] = 'cpf'
self.test_ini['requires'] = 'test_rpm'
self.loaded_ini[self.filename] = []
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dirname, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
self.assertTrue(self._return_check('check_rpm_to "test_rpm" "cpf"'))
def test_check_script_is_directory(self):
"""
Directory as input instead of regular file is incorrect input
Tests issue #29
"""
self.test_ini['check_script'] = '.'
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingTagsIniFileError, lambda: list(self.xml_utils.prepare_sections()))
def setUp(self):
self.dirname = os.path.join("tests",
"FOOBAR6_7" + settings.results_postfix,
"test")
if os.path.exists(self.dirname):
shutil.rmtree(self.dirname)
os.makedirs(self.dirname)
self.filename = os.path.join(self.dirname, 'test.ini')
self.rule = []
self.test_solution = "test_solution.txt"
self.check_script = "check_script.sh"
self.loaded_ini = {}
test_ini = {
'content_title': 'Testing content title',
'content_description': ' some content description',
'author': 'test <*****@*****.**>',
'config_file': '/etc/named.conf',
'check_script': self.check_script,
'solution': self.test_solution,
'applies_to': 'test',
'requires': 'bash',
'binary_req': 'sed'
}
self.loaded_ini[self.filename] = []
self.loaded_ini[self.filename].append(test_ini)
self.check_sh = """#!/bin/bash
#END GENERATED SECTION
#This is testing check script
"""
check_name = os.path.join(self.dirname, self.check_script)
FileHelper.write_to_file(check_name, "wb", self.check_sh)
os.chmod(check_name, stat.S_IEXEC | stat.S_IRWXG | stat.S_IRWXU)
self.solution_text = """
A solution text for test suite"
"""
test_solution_name = os.path.join(self.dirname, self.test_solution)
FileHelper.write_to_file(test_solution_name, "wb", self.solution_text)
os.chmod(check_name, stat.S_IEXEC | stat.S_IRWXG | stat.S_IRWXU)
self.xml_utils = XmlUtils(self.dirname, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
def setUp(self):
self.dirname = os.path.join("tests",
"FOOBAR6_7" + settings.results_postfix,
"test")
if os.path.exists(self.dirname):
shutil.rmtree(self.dirname)
os.makedirs(self.dirname)
self.filename = os.path.join(self.dirname, 'test.ini')
self.rule = []
self.test_solution = "test_solution.txt"
self.check_script = "check_script.sh"
self.loaded_ini = {}
test_ini = {'content_title': 'Testing content title',
'content_description': ' some content description',
'author': 'test <*****@*****.**>',
'config_file': '/etc/named.conf',
'check_script': self.check_script,
'solution': self.test_solution,
'applies_to': 'test',
'requires': 'bash',
'binary_req': 'sed'}
self.loaded_ini[self.filename] = []
self.loaded_ini[self.filename].append(test_ini)
self.check_sh = """#!/bin/bash
#END GENERATED SECTION
#This is testing check script
"""
check_name = os.path.join(self.dirname, self.check_script)
FileHelper.write_to_file(check_name, "wb", self.check_sh)
os.chmod(check_name, stat.S_IEXEC | stat.S_IRWXG | stat.S_IRWXU)
self.solution_text = """
A solution text for test suite"
"""
test_solution_name = os.path.join(self.dirname, self.test_solution)
FileHelper.write_to_file(test_solution_name, "wb", self.solution_text)
os.chmod(check_name, stat.S_IEXEC | stat.S_IRWXG | stat.S_IRWXU)
self.xml_utils = XmlUtils(self.dirname, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
class TestIncorrectINI(base.TestCase):
"""
Tests right processing of INI files including incorrect input which
could make for crash with traceback.
"""
dir_name = None
filename = None
rule = None
test_solution = None
check_script = None
loaded_ini = {}
test_ini = None
xml_utils = None
def setUp(self):
self.dir_name = "tests/FOOBAR6_7/incorrect_ini"
os.makedirs(self.dir_name)
self.filename = os.path.join(self.dir_name, 'test.ini')
self.rule = []
self.test_solution = "test_solution.sh"
self.check_script = "check_script.sh"
self.loaded_ini[self.filename] = []
self.test_ini = {
'content_title': 'Testing content title',
'content_description': 'Some content description',
'author': 'test <*****@*****.**>',
'config_file': '/etc/named.conf',
'check_script': self.check_script,
'solution': self.test_solution,
'applies_to': 'test'
}
solution_text = """
A solution text for test suite"
"""
check_sh = """#!/bin/bash
#END GENERATED SECTION
#This is testing check script
"""
FileHelper.write_to_file(
os.path.join(self.dir_name, self.test_solution), "wb",
solution_text)
FileHelper.write_to_file(
os.path.join(self.dir_name, self.check_script), "wb", check_sh)
def tearDown(self):
shutil.rmtree(self.dir_name)
def test_missing_tag_check_script(self):
"""Basic test for whole program"""
self.test_ini.pop('check_script', None)
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingTagsIniFileError,
lambda: list(self.xml_utils.prepare_sections()))
def test_missing_tag_solution_script(self):
"""Test of missing tag 'solution' - SystemExit should be raised"""
self.test_ini.pop('solution', None)
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingTagsIniFileError,
lambda: list(self.xml_utils.prepare_sections()))
def test_file_solution_not_exists(self):
"""Test of missing 'solution' file - SystemExit should be raised"""
self.test_ini['solution'] = "this_should_be_unexpected_file.txt"
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingTagsIniFileError,
lambda: list(self.xml_utils.prepare_sections()))
def test_file_check_script_not_exists(self):
"""Test of missing 'check_script' file"""
self.test_ini['check_script'] = "this_should_be_unexpected_file.txt"
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingFileInContentError,
lambda: list(self.xml_utils.prepare_sections()))
def test_check_script_is_directory(self):
"""
Directory as input instead of regular file is incorrect input
Tests issue #29
"""
self.test_ini['check_script'] = '.'
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingTagsIniFileError,
lambda: list(self.xml_utils.prepare_sections()))
def test_incorrect_tag(self):
"""
Check occurrence of incorrect tag
Tests issue #30
"""
text_ini = '[preupgrade]\n'
text_ini += '\n'.join(
[key + " = " + self.test_ini[key] for key in self.test_ini])
text_ini += '\n[]\neliskk\n'
FileHelper.write_to_file(self.filename, "wb", text_ini)
oscap = OscapGroupXml(self.dir_name)
self.assertRaises(SystemExit, oscap.find_all_ini)
def test_secret_check_script(self):
"""Check occurrence of secret file for check script"""
self.test_ini['check_script'] = '.minicheck'
text = """#!/usr/bin/sh\necho 'ahojky'\n"""
FileHelper.write_to_file(
os.path.join(self.dir_name, self.check_script), "wb", text)
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingFileInContentError,
lambda: list(self.xml_utils.prepare_sections()))
class TestScriptGenerator(base.TestCase):
"""Main testing of right generating of XML files for OSCAP."""
dirname = None
filename = None
test_solution = None
loaded_ini = None
check_script = None
check_sh = None
solution_text = None
rule = None
xml_utils = None
test_ini = []
def setUp(self):
self.root_dir_name = "tests/FOOBAR6_7" + settings.results_postfix
self.dirname = os.path.join(self.root_dir_name, "test")
if os.path.exists(self.dirname):
shutil.rmtree(self.dirname)
os.makedirs(self.dirname)
self.filename = os.path.join(self.dirname, 'test.ini')
self.rule = []
self.loaded_ini = {}
self.test_ini = {'content_title': 'Testing content title',
'content_description': ' some content description',
'author': 'test <*****@*****.**>',
'config_file': '/etc/named.conf'
}
self.check_sh = """#!/bin/bash
#END GENERATED SECTION
#This is testing check script
"""
check_name = os.path.join(self.dirname, settings.check_script)
FileHelper.write_to_file(check_name, "wb", self.check_sh)
os.chmod(check_name, stat.S_IEXEC | stat.S_IRWXG | stat.S_IRWXU)
self.solution_text = """
A solution text for test suite"
"""
test_solution_name = os.path.join(self.dirname, settings.solution_txt)
FileHelper.write_to_file(test_solution_name, "wb", self.solution_text)
os.chmod(check_name, stat.S_IEXEC | stat.S_IRWXG | stat.S_IRWXU)
def _return_check(self, text):
content = FileHelper.get_file_content(os.path.join(
self.dirname, settings.check_script), "rb", method=True)
found = [x for x in content if x.startswith(text)]
return found
def test_applies_to(self):
self.test_ini['applies_to'] = 'test_rpm'
self.loaded_ini[self.filename] = self.test_ini
self.xml_utils = XmlUtils(self.root_dir_name, self.dirname,
self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
self.assertTrue(self._return_check('check_applies_to "test_rpm"'))
def test_check_bin(self):
self.test_ini['binary_req'] = 'cpf'
self.loaded_ini[self.filename] = self.test_ini
self.xml_utils = XmlUtils(self.root_dir_name, self.dirname,
self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
self.assertTrue(self._return_check('check_rpm_to "" "cpf"'))
def test_check_rpm(self):
self.test_ini['requires'] = 'test_rpm'
self.loaded_ini[self.filename] = self.test_ini
self.xml_utils = XmlUtils(self.root_dir_name, self.dirname,
self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
self.assertTrue(self._return_check('check_rpm_to "test_rpm" ""'))
def test_check_rpm_bin(self):
self.test_ini['binary_req'] = 'cpf'
self.test_ini['requires'] = 'test_rpm'
self.loaded_ini[self.filename] = self.test_ini
self.xml_utils = XmlUtils(self.root_dir_name, self.dirname,
self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
self.assertTrue(self._return_check('check_rpm_to "test_rpm" "cpf"'))
def test_applies_to_bin(self):
self.test_ini['applies_to'] = 'test_rpm'
self.test_ini['binary_req'] = 'cpf'
self.loaded_ini[self.filename] = self.test_ini
self.xml_utils = XmlUtils(self.root_dir_name, self.dirname,
self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
self.assertTrue(self._return_check('check_applies_to "test_rpm"'))
self.assertTrue(self._return_check('check_rpm_to "" "cpf"'))
def tearDown(self):
shutil.rmtree(self.dirname)
def test_xml_solution_type_html(self):
self.loaded_ini[self.filename][0]['solution_type'] = "html"
self.xml_utils = XmlUtils(self.dirname, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
fix_text = [x for x in self.rule if "<fixtext>_test_SOLUTION_MSG_HTML</fixtext>" in x]
self.assertTrue(fix_text)
class TestIncorrectINI(base.TestCase):
"""
Tests right processing of INI files including incorrect input which
could make for crash with traceback.
"""
dir_name = None
filename = None
rule = None
test_solution = None
check_script = None
loaded_ini = {}
test_ini = None
xml_utils = None
def setUp(self):
self.dir_name = "tests/FOOBAR6_7/incorrect_ini"
os.makedirs(self.dir_name)
self.filename = os.path.join(self.dir_name, 'test.ini')
self.rule = []
self.test_solution = "test_solution.sh"
self.check_script = "check_script.sh"
self.loaded_ini[self.filename] = []
self.test_ini = {'content_title': 'Testing content title',
'content_description': 'Some content description',
'author': 'test <*****@*****.**>',
'config_file': '/etc/named.conf',
'check_script': self.check_script,
'solution': self.test_solution,
'applies_to': 'test'
}
solution_text = """
A solution text for test suite"
"""
check_sh = """#!/bin/bash
#END GENERATED SECTION
#This is testing check script
"""
FileHelper.write_to_file(os.path.join(self.dir_name, self.test_solution), "wb", solution_text)
FileHelper.write_to_file(os.path.join(self.dir_name, self.check_script), "wb", check_sh)
def tearDown(self):
shutil.rmtree(self.dir_name)
def test_missing_tag_check_script(self):
"""Basic test for whole program"""
self.test_ini.pop('check_script', None)
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingTagsIniFileError, lambda: list(self.xml_utils.prepare_sections()))
def test_missing_tag_solution_script(self):
"""Test of missing tag 'solution' - SystemExit should be raised"""
self.test_ini.pop('solution', None)
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingTagsIniFileError, lambda: list(self.xml_utils.prepare_sections()))
def test_file_solution_not_exists(self):
"""Test of missing 'solution' file - SystemExit should be raised"""
self.test_ini['solution'] = "this_should_be_unexpected_file.txt"
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingTagsIniFileError, lambda: list(self.xml_utils.prepare_sections()))
def test_file_check_script_not_exists(self):
"""Test of missing 'check_script' file"""
self.test_ini['check_script'] = "this_should_be_unexpected_file.txt"
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingFileInContentError, lambda: list(self.xml_utils.prepare_sections()))
def test_check_script_is_directory(self):
"""
Directory as input instead of regular file is incorrect input
Tests issue #29
"""
self.test_ini['check_script'] = '.'
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingTagsIniFileError, lambda: list(self.xml_utils.prepare_sections()))
def test_incorrect_tag(self):
"""
Check occurrence of incorrect tag
Tests issue #30
"""
text_ini = '[preupgrade]\n'
text_ini += '\n'.join([key + " = " + self.test_ini[key] for key in self.test_ini])
text_ini += '\n[]\neliskk\n'
FileHelper.write_to_file(self.filename, "wb", text_ini)
oscap = OscapGroupXml(self.dir_name)
self.assertRaises(SystemExit, oscap.find_all_ini)
def test_secret_check_script(self):
"""Check occurrence of secret file for check script"""
self.test_ini['check_script'] = '.minicheck'
text = """#!/usr/bin/sh\necho 'ahojky'\n"""
FileHelper.write_to_file(os.path.join(self.dir_name, self.check_script), "wb", text)
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingFileInContentError, lambda: list(self.xml_utils.prepare_sections()))
def test_missing_tag_check_script(self):
"""Basic test for whole program"""
self.test_ini.pop('check_script', None)
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingTagsIniFileError, lambda: list(self.xml_utils.prepare_sections()))
def test_missing_tag_solution_script(self):
"""Test of missing tag 'solution' - SystemExit should be raised"""
self.test_ini.pop('solution', None)
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingTagsIniFileError, lambda: list(self.xml_utils.prepare_sections()))
def test_file_solution_not_exists(self):
"""Test of missing 'solution' file - SystemExit should be raised"""
self.test_ini['solution'] = "this_should_be_unexpected_file.txt"
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingTagsIniFileError, lambda: list(self.xml_utils.prepare_sections()))
class TestXML(base.TestCase):
"""Main testing of right generating of XML files for OSCAP."""
dirname = None
filename = None
test_solution = None
loaded_ini = None
check_script = None
check_sh = None
solution_text = None
rule = None
xml_utils = None
def setUp(self):
self.dirname = os.path.join("tests",
"FOOBAR6_7" + settings.results_postfix,
"test")
if os.path.exists(self.dirname):
shutil.rmtree(self.dirname)
os.makedirs(self.dirname)
self.filename = os.path.join(self.dirname, 'test.ini')
self.rule = []
self.test_solution = "test_solution.txt"
self.check_script = "check_script.sh"
self.loaded_ini = {}
test_ini = {
'content_title': 'Testing content title',
'content_description': ' some content description',
'author': 'test <*****@*****.**>',
'config_file': '/etc/named.conf',
'check_script': self.check_script,
'solution': self.test_solution,
'applies_to': 'test',
'requires': 'bash',
'binary_req': 'sed'
}
self.loaded_ini[self.filename] = []
self.loaded_ini[self.filename].append(test_ini)
self.check_sh = """#!/bin/bash
#END GENERATED SECTION
#This is testing check script
"""
check_name = os.path.join(self.dirname, self.check_script)
FileHelper.write_to_file(check_name, "wb", self.check_sh)
os.chmod(check_name, stat.S_IEXEC | stat.S_IRWXG | stat.S_IRWXU)
self.solution_text = """
A solution text for test suite"
"""
test_solution_name = os.path.join(self.dirname, self.test_solution)
FileHelper.write_to_file(test_solution_name, "wb", self.solution_text)
os.chmod(check_name, stat.S_IEXEC | stat.S_IRWXG | stat.S_IRWXU)
self.xml_utils = XmlUtils(self.dirname, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
def tearDown(self):
shutil.rmtree(self.dirname)
if os.path.exists(os.path.join(os.getcwd(), 'migrate')):
os.unlink(os.path.join(os.getcwd(), 'migrate'))
if os.path.exists(os.path.join(os.getcwd(), 'upgrade')):
os.unlink(os.path.join(os.getcwd(), 'upgrade'))
def test_group_xml(self):
"""Basic test for whole program"""
self.assertTrue(self.loaded_ini[self.filename])
self.assertTrue(self.rule)
def test_xml_rule_id(self):
rule_id = [
x for x in self.rule if
'<Rule id="xccdf_preupg_rule_test_check_script" selected="true">'
in x
]
self.assertTrue(rule_id)
def test_xml_profile_id(self):
profile = [
x for x in self.rule
if '<Profile id="xccdf_preupg_profile_default">' in x
]
self.assertTrue(profile)
def test_xml_rule_title(self):
rule_title = [
x for x in self.rule if "<title>Testing content title</title>" in x
]
self.assertTrue(rule_title)
def test_xml_config_file(self):
conf_file = [
x for x in self.rule if "<xhtml:li>/etc/named.conf</xhtml:li>" in x
]
self.assertTrue(conf_file)
def test_xml_fix_text(self):
fix_text = [
x for x in self.rule
if "<fixtext>_test_SOLUTION_MSG_TEXT</fixtext>" in x
]
self.assertTrue(fix_text)
def test_xml_solution_type_text(self):
self.loaded_ini[self.filename][0]['solution_type'] = "text"
self.xml_utils = XmlUtils(self.dirname, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
fix_text = [
x for x in self.rule
if "<fixtext>_test_SOLUTION_MSG_TEXT</fixtext>" in x
]
self.assertTrue(fix_text)
def test_xml_solution_type_html(self):
self.loaded_ini[self.filename][0]['solution_type'] = "html"
self.xml_utils = XmlUtils(self.dirname, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
fix_text = [
x for x in self.rule
if "<fixtext>_test_SOLUTION_MSG_HTML</fixtext>" in x
]
self.assertTrue(fix_text)
def test_check_script_author(self):
settings.autocomplete = True
self.rule = self.xml_utils.prepare_sections()
lines = FileHelper.get_file_content(os.path.join(
self.dirname, self.check_script),
"rb",
method=True)
author = [x for x in lines if "test <*****@*****.**>" in x]
self.assertTrue(author)
def test_xml_check_export_tmp_preupgrade(self):
self.rule = self.xml_utils.prepare_sections()
check_export = [
x for x in self.rule if 'xccdf_preupg_value_tmp_preupgrade' in x
]
self.assertTrue(check_export)
def test_xml_current_directory(self):
self.rule = self.xml_utils.prepare_sections()
cur_directory = [
x for x in self.rule if
'<check-export export-name="CURRENT_DIRECTORY" value-id="xccdf_preupg_value_test_check_script_state_current_directory" />'
in x
]
self.assertTrue(cur_directory)
def _create_temporary_dir(self):
settings.UPGRADE_PATH = tempfile.mkdtemp()
if os.path.exists(settings.UPGRADE_PATH):
shutil.rmtree(settings.UPGRADE_PATH)
os.makedirs(settings.UPGRADE_PATH)
migrate = os.path.join(settings.UPGRADE_PATH, 'migrate')
upgrade = os.path.join(settings.UPGRADE_PATH, 'upgrade')
return migrate, upgrade
def _delete_temporary_dir(self, migrate, upgrade):
if os.path.exists(migrate):
os.unlink(migrate)
if os.path.exists(upgrade):
os.unlink(upgrade)
if os.path.exists(settings.UPGRADE_PATH):
shutil.rmtree(settings.UPGRADE_PATH)
def test_xml_migrate_not_upgrade(self):
test_ini = {
'content_title': 'Testing only migrate title',
'content_description': ' some content description',
'author': 'test <*****@*****.**>',
'config_file': '/etc/named.conf',
'check_script': self.check_script,
'solution': self.test_solution,
'applies_to': 'test',
'requires': 'bash',
'binary_req': 'sed',
'mode': 'migrate'
}
ini = {}
old_settings = settings.UPGRADE_PATH
migrate, upgrade = self._create_temporary_dir()
ini[self.filename] = []
ini[self.filename].append(test_ini)
xml_utils = XmlUtils(self.dirname, ini)
xml_utils.prepare_sections()
migrate_file = FileHelper.get_file_content(migrate, 'rb', method=True)
tag = [
x.strip() for x in migrate_file
if 'xccdf_preupg_rule_test_check_script' in x.strip()
]
self.assertIsNotNone(tag)
try:
upgrade_file = FileHelper.get_file_content(upgrade,
'rb',
method=True)
except IOError:
upgrade_file = None
self.assertIsNone(upgrade_file)
self._delete_temporary_dir(migrate, upgrade)
settings.UPGRADE_PATH = old_settings
def test_xml_upgrade_not_migrate(self):
test_ini = {
'content_title': 'Testing only migrate title',
'content_description': ' some content description',
'author': 'test <*****@*****.**>',
'config_file': '/etc/named.conf',
'check_script': self.check_script,
'solution': self.test_solution,
'applies_to': 'test',
'requires': 'bash',
'binary_req': 'sed',
'mode': 'upgrade'
}
ini = {}
old_settings = settings.UPGRADE_PATH
migrate, upgrade = self._create_temporary_dir()
ini[self.filename] = []
ini[self.filename].append(test_ini)
xml_utils = XmlUtils(self.dirname, ini)
xml_utils.prepare_sections()
upgrade_file = FileHelper.get_file_content(upgrade, 'rb', method=True)
tag = [
x.strip() for x in upgrade_file
if 'xccdf_preupg_rule_test_check_script' in x.strip()
]
self.assertIsNotNone(tag)
try:
migrate_file = FileHelper.get_file_content(migrate,
'rb',
method=True)
except IOError:
migrate_file = None
self.assertIsNone(migrate_file)
self._delete_temporary_dir(migrate, upgrade)
settings.UPGRADE_PATH = old_settings
def test_xml_migrate_and_upgrade(self):
test_ini = {
'content_title': 'Testing only migrate title',
'content_description': ' some content description',
'author': 'test <*****@*****.**>',
'config_file': '/etc/named.conf',
'check_script': self.check_script,
'solution': self.test_solution,
'applies_to': 'test',
'requires': 'bash',
'binary_req': 'sed',
'mode': 'migrate, upgrade'
}
ini = {}
old_settings = settings.UPGRADE_PATH
migrate, upgrade = self._create_temporary_dir()
ini[self.filename] = []
ini[self.filename].append(test_ini)
xml_utils = XmlUtils(self.dirname, ini)
xml_utils.prepare_sections()
migrate_file = FileHelper.get_file_content(migrate, 'rb', method=True)
tag = [
x.strip() for x in migrate_file
if 'xccdf_preupg_rule_test_check_script' in x.strip()
]
self.assertIsNotNone(tag)
upgrade_file = FileHelper.get_file_content(upgrade, 'rb', method=True)
tag = [
x.strip() for x in upgrade_file
if 'xccdf_preupg_rule_test_check_script' in x.strip()
]
self.assertIsNotNone(tag)
self._delete_temporary_dir(migrate, upgrade)
settings.UPGRADE_PATH = old_settings
def test_xml_not_migrate_not_upgrade(self):
test_ini = {
'content_title': 'Testing only migrate title',
'content_description': ' some content description',
'author': 'test <*****@*****.**>',
'config_file': '/etc/named.conf',
'check_script': self.check_script,
'solution': self.test_solution,
'applies_to': 'test',
'requires': 'bash',
'binary_req': 'sed'
}
ini = {}
old_settings = settings.UPGRADE_PATH
migrate, upgrade = self._create_temporary_dir()
ini[self.filename] = []
ini[self.filename].append(test_ini)
xml_utils = XmlUtils(self.dirname, ini)
xml_utils.prepare_sections()
migrate_file = FileHelper.get_file_content(migrate, 'rb', method=True)
tag = [
x.strip() for x in migrate_file
if 'xccdf_preupg_rule_test_check_script' in x.strip()
]
self.assertIsNotNone(tag)
upgrade_file = FileHelper.get_file_content(upgrade, 'rb', method=True)
tag = [
x.strip() for x in upgrade_file
if 'xccdf_preupg_rule_test_check_script' in x.strip()
]
self.assertIsNotNone(tag)
self._delete_temporary_dir(migrate, upgrade)
settings.UPGRADE_PATH = old_settings
def test_xml_check_script_reference(self):
self.rule = self.xml_utils.prepare_sections()
check_script_reference = [
x for x in self.rule
if '<check-content-ref href="check_script.sh" />' in x
]
self.assertTrue(check_script_reference)
def test_values_id(self):
self.rule = self.xml_utils.prepare_sections()
value_current_dir = [
x for x in self.rule if
'<Value id="xccdf_preupg_value_test_check_script_state_current_directory"'
in x
]
self.assertTrue(value_current_dir)
value_current_dir_set = [
x for x in self.rule if '<value>SCENARIO/test</value>' in x
]
self.assertTrue(value_current_dir_set)
def test_check_script_applies_to(self):
self.rule = self.xml_utils.prepare_sections()
lines = FileHelper.get_file_content(os.path.join(
self.dirname, self.check_script),
"rb",
method=True)
applies = [x for x in lines if 'check_applies_to "test"' in x]
self.assertTrue(applies)
def test_check_script_common(self):
self.rule = self.xml_utils.prepare_sections()
lines = FileHelper.get_file_content(os.path.join(
self.dirname, self.check_script),
"rb",
method=True)
common = [x for x in lines if '. /usr/share/preupgrade/common.sh' in x]
self.assertTrue(common)
def test_check_script_requires(self):
self.rule = self.xml_utils.prepare_sections()
lines = FileHelper.get_file_content(os.path.join(
self.dirname, self.check_script),
"rb",
method=True)
check_rpm_to = [x for x in lines if 'check_rpm_to "bash" "sed"' in x]
self.assertTrue(check_rpm_to)
def test_xml_solution_type_text(self):
self.xml_utils = XmlUtils(self.root_dir_name, self.dirname,
self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
fix_text = [x for x in self.rule if "<fixtext>_test_SOLUTION_MSG</fixtext>" in x]
self.assertTrue(fix_text)
class TestScriptGenerator(base.TestCase):
"""Main testing of right generating of XML files for OSCAP."""
dirname = None
filename = None
test_solution = None
loaded_ini = None
check_script = None
check_sh = None
solution_text = None
rule = None
xml_utils = None
test_ini = []
def setUp(self):
self.dirname = os.path.join("tests",
"FOOBAR6_7" + settings.results_postfix,
"test")
if os.path.exists(self.dirname):
shutil.rmtree(self.dirname)
os.makedirs(self.dirname)
self.filename = os.path.join(self.dirname, 'test.ini')
self.rule = []
self.test_solution = "test_solution.txt"
self.check_script = "check_script.sh"
self.loaded_ini = {}
self.test_ini = {
'content_title': 'Testing content title',
'content_description': ' some content description',
'author': 'test <*****@*****.**>',
'config_file': '/etc/named.conf',
'check_script': self.check_script,
'solution': self.test_solution
}
self.check_sh = """#!/bin/bash
#END GENERATED SECTION
#This is testing check script
"""
check_name = os.path.join(self.dirname, self.check_script)
FileHelper.write_to_file(check_name, "wb", self.check_sh)
os.chmod(check_name, stat.S_IEXEC | stat.S_IRWXG | stat.S_IRWXU)
self.solution_text = """
A solution text for test suite"
"""
test_solution_name = os.path.join(self.dirname, self.test_solution)
FileHelper.write_to_file(test_solution_name, "wb", self.solution_text)
os.chmod(check_name, stat.S_IEXEC | stat.S_IRWXG | stat.S_IRWXU)
def _return_check(self, text):
content = FileHelper.get_file_content(os.path.join(
self.dirname, self.check_script),
"rb",
method=True)
found = [x for x in content if x.startswith(text)]
return found
def test_applies_to(self):
self.test_ini['applies_to'] = 'test_rpm'
self.loaded_ini[self.filename] = []
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dirname, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
self.assertTrue(self._return_check('check_applies_to "test_rpm"'))
def test_check_bin(self):
self.test_ini['binary_req'] = 'cpf'
self.loaded_ini[self.filename] = []
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dirname, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
self.assertTrue(self._return_check('check_rpm_to "" "cpf"'))
def test_check_rpm(self):
self.test_ini['requires'] = 'test_rpm'
self.loaded_ini[self.filename] = []
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dirname, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
self.assertTrue(self._return_check('check_rpm_to "test_rpm" ""'))
def test_check_rpm_bin(self):
self.test_ini['binary_req'] = 'cpf'
self.test_ini['requires'] = 'test_rpm'
self.loaded_ini[self.filename] = []
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dirname, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
self.assertTrue(self._return_check('check_rpm_to "test_rpm" "cpf"'))
def test_applies_to_bin(self):
self.test_ini['applies_to'] = 'test_rpm'
self.test_ini['binary_req'] = 'cpf'
self.loaded_ini[self.filename] = []
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dirname, self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
self.assertTrue(self._return_check('check_applies_to "test_rpm"'))
self.assertTrue(self._return_check('check_rpm_to "" "cpf"'))
def tearDown(self):
shutil.rmtree(self.dirname)
def test_file_check_script_not_exists(self):
"""Test of missing 'check_script' file"""
self.test_ini['check_script'] = "this_should_be_unexpected_file.txt"
self.loaded_ini[self.filename].append(self.test_ini)
self.xml_utils = XmlUtils(self.dir_name, self.loaded_ini)
self.assertRaises(MissingFileInContentError, lambda: list(self.xml_utils.prepare_sections()))
class TestXML(base.TestCase):
"""Main testing of right generating of XML files for OSCAP."""
dirname = None
filename = None
test_solution = None
loaded_ini = None
check_script = None
check_sh = None
solution_text = None
rule = None
xml_utils = None
def setUp(self):
self.root_dir_name = "tests/FOOBAR6_7" + settings.results_postfix
self.dirname = os.path.join(self.root_dir_name, "test")
if os.path.exists(self.dirname):
shutil.rmtree(self.dirname)
os.makedirs(self.dirname)
self.filename = os.path.join(self.dirname, 'test.ini')
self.rule = []
self.loaded_ini = {}
test_ini = {'content_title': 'Testing content title',
'content_description': ' some content description',
'author': 'test <*****@*****.**>',
'config_file': '/etc/named.conf',
'applies_to': 'test',
'requires': 'bash',
'binary_req': 'sed'}
self.loaded_ini[self.filename] = test_ini
self.check_sh = """#!/bin/bash
#END GENERATED SECTION
#This is testing check script
"""
check_name = os.path.join(self.dirname, settings.check_script)
FileHelper.write_to_file(check_name, "wb", self.check_sh)
os.chmod(check_name, stat.S_IEXEC | stat.S_IRWXG | stat.S_IRWXU)
self.solution_text = """
A solution text for test suite"
"""
test_solution_name = os.path.join(self.dirname, settings.solution_txt)
FileHelper.write_to_file(test_solution_name, "wb", self.solution_text)
os.chmod(check_name, stat.S_IEXEC | stat.S_IRWXG | stat.S_IRWXU)
self.xml_utils = XmlUtils(self.root_dir_name, self.dirname,
self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
def tearDown(self):
shutil.rmtree(self.dirname)
if os.path.exists(os.path.join(os.getcwd(), 'migrate')):
os.unlink(os.path.join(os.getcwd(), 'migrate'))
if os.path.exists(os.path.join(os.getcwd(), 'upgrade')):
os.unlink(os.path.join(os.getcwd(), 'upgrade'))
def test_group_xml(self):
"""Basic test for whole program"""
self.assertTrue(self.loaded_ini[self.filename])
self.assertTrue(self.rule)
def test_xml_rule_id(self):
rule_id = [x for x in self.rule if '<Rule id="xccdf_preupg_rule_test_check" selected="true">' in x]
self.assertTrue(rule_id)
def test_xml_profile_id(self):
profile = [x for x in self.rule if '<Profile id="xccdf_preupg_profile_default">' in x]
self.assertTrue(profile)
def test_xml_rule_title(self):
rule_title = [x for x in self.rule if "<title>Testing content title</title>" in x]
self.assertTrue(rule_title)
def test_xml_config_file(self):
conf_file = [x for x in self.rule if "<xhtml:li>/etc/named.conf</xhtml:li>" in x]
self.assertTrue(conf_file)
def test_xml_fix_text(self):
fix_text = [x for x in self.rule if "<fixtext>_test_SOLUTION_MSG</fixtext>" in x]
self.assertTrue(fix_text)
def test_xml_solution_type_text(self):
self.xml_utils = XmlUtils(self.root_dir_name, self.dirname,
self.loaded_ini)
self.rule = self.xml_utils.prepare_sections()
fix_text = [x for x in self.rule if "<fixtext>_test_SOLUTION_MSG</fixtext>" in x]
self.assertTrue(fix_text)
def test_check_script_author(self):
settings.autocomplete = True
self.rule = self.xml_utils.prepare_sections()
lines = FileHelper.get_file_content(os.path.join(
self.dirname, settings.check_script), "rb", method=True)
author = [x for x in lines if "test <*****@*****.**>" in x]
self.assertTrue(author)
def test_xml_check_export_tmp_preupgrade(self):
self.rule = self.xml_utils.prepare_sections()
check_export = [x for x in self.rule if 'xccdf_preupg_value_tmp_preupgrade' in x]
self.assertTrue(check_export)
def test_xml_current_directory(self):
self.rule = self.xml_utils.prepare_sections()
cur_directory = [x for x in self.rule if '<check-export export-name="CURRENT_DIRECTORY" value-id="xccdf_preupg_value_test_check_state_current_directory" />' in x]
self.assertTrue(cur_directory)
def _create_temporary_dir(self):
settings.UPGRADE_PATH = tempfile.mkdtemp()
if os.path.exists(settings.UPGRADE_PATH):
shutil.rmtree(settings.UPGRADE_PATH)
os.makedirs(settings.UPGRADE_PATH)
migrate = os.path.join(settings.UPGRADE_PATH, 'migrate')
upgrade = os.path.join(settings.UPGRADE_PATH, 'upgrade')
return migrate, upgrade
def _delete_temporary_dir(self, migrate, upgrade):
if os.path.exists(migrate):
os.unlink(migrate)
if os.path.exists(upgrade):
os.unlink(upgrade)
if os.path.exists(settings.UPGRADE_PATH):
shutil.rmtree(settings.UPGRADE_PATH)
def test_xml_migrate_not_upgrade(self):
test_ini = {'content_title': 'Testing only migrate title',
'content_description': ' some content description',
'author': 'test <*****@*****.**>',
'config_file': '/etc/named.conf',
'applies_to': 'test',
'requires': 'bash',
'binary_req': 'sed',
'mode': 'migrate'}
ini = {}
old_settings = settings.UPGRADE_PATH
migrate, upgrade = self._create_temporary_dir()
ini[self.filename] = test_ini
xml_utils = XmlUtils(self.root_dir_name, self.dirname, ini)
xml_utils.prepare_sections()
migrate_file = FileHelper.get_file_content(migrate, 'rb', method=True)
tag = [x.strip() for x in migrate_file if 'xccdf_preupg_rule_test_check_script' in x.strip()]
self.assertIsNotNone(tag)
try:
upgrade_file = FileHelper.get_file_content(upgrade, 'rb', method=True)
except IOError:
upgrade_file = None
self.assertIsNone(upgrade_file)
self._delete_temporary_dir(migrate, upgrade)
settings.UPGRADE_PATH = old_settings
def test_xml_upgrade_not_migrate(self):
test_ini = {'content_title': 'Testing only migrate title',
'content_description': ' some content description',
'author': 'test <*****@*****.**>',
'config_file': '/etc/named.conf',
'applies_to': 'test',
'requires': 'bash',
'binary_req': 'sed',
'mode': 'upgrade'}
ini = {}
old_settings = settings.UPGRADE_PATH
migrate, upgrade = self._create_temporary_dir()
ini[self.filename] = test_ini
xml_utils = XmlUtils(self.root_dir_name, self.dirname, ini)
xml_utils.prepare_sections()
upgrade_file = FileHelper.get_file_content(upgrade, 'rb', method=True)
tag = [x.strip() for x in upgrade_file if 'xccdf_preupg_rule_test_check_script' in x.strip()]
self.assertIsNotNone(tag)
try:
migrate_file = FileHelper.get_file_content(migrate, 'rb', method=True)
except IOError:
migrate_file = None
self.assertIsNone(migrate_file)
self._delete_temporary_dir(migrate, upgrade)
settings.UPGRADE_PATH = old_settings
def test_xml_migrate_and_upgrade(self):
test_ini = {'content_title': 'Testing only migrate title',
'content_description': ' some content description',
'author': 'test <*****@*****.**>',
'config_file': '/etc/named.conf',
'applies_to': 'test',
'requires': 'bash',
'binary_req': 'sed',
'mode': 'migrate, upgrade'}
ini = {}
old_settings = settings.UPGRADE_PATH
migrate, upgrade = self._create_temporary_dir()
ini[self.filename] = test_ini
xml_utils = XmlUtils(self.root_dir_name, self.dirname, ini)
xml_utils.prepare_sections()
migrate_file = FileHelper.get_file_content(migrate, 'rb', method=True)
tag = [x.strip() for x in migrate_file if 'xccdf_preupg_rule_test_check_script' in x.strip()]
self.assertIsNotNone(tag)
upgrade_file = FileHelper.get_file_content(upgrade, 'rb', method=True)
tag = [x.strip() for x in upgrade_file if 'xccdf_preupg_rule_test_check_script' in x.strip()]
self.assertIsNotNone(tag)
self._delete_temporary_dir(migrate, upgrade)
settings.UPGRADE_PATH = old_settings
def test_xml_not_migrate_not_upgrade(self):
test_ini = {'content_title': 'Testing only migrate title',
'content_description': ' some content description',
'author': 'test <*****@*****.**>',
'config_file': '/etc/named.conf',
'applies_to': 'test',
'requires': 'bash',
'binary_req': 'sed'}
ini = {}
old_settings = settings.UPGRADE_PATH
migrate, upgrade = self._create_temporary_dir()
ini[self.filename] = test_ini
xml_utils = XmlUtils(self.root_dir_name, self.dirname, ini)
xml_utils.prepare_sections()
migrate_file = FileHelper.get_file_content(migrate, 'rb', method=True)
tag = [x.strip() for x in migrate_file if 'xccdf_preupg_rule_test_check_script' in x.strip()]
self.assertIsNotNone(tag)
upgrade_file = FileHelper.get_file_content(upgrade, 'rb', method=True)
tag = [x.strip() for x in upgrade_file if 'xccdf_preupg_rule_test_check_script' in x.strip()]
self.assertIsNotNone(tag)
self._delete_temporary_dir(migrate, upgrade)
settings.UPGRADE_PATH = old_settings
def test_xml_check_script_reference(self):
self.rule = self.xml_utils.prepare_sections()
check_script_reference = [x for x in self.rule if '<check-content-ref href="check" />' in x]
self.assertTrue(check_script_reference)
def test_values_id(self):
self.rule = self.xml_utils.prepare_sections()
value_current_dir = [x for x in self.rule if '<Value id="xccdf_preupg_value_test_check_state_current_directory"' in x]
self.assertTrue(value_current_dir)
value_current_dir_set = [x for x in self.rule if '<value>SCENARIO/test</value>' in x]
self.assertTrue(value_current_dir_set)
def test_check_script_applies_to(self):
self.rule = self.xml_utils.prepare_sections()
lines = FileHelper.get_file_content(os.path.join(
self.dirname, settings.check_script), "rb", method=True)
applies = [x for x in lines if 'check_applies_to "test"' in x]
self.assertTrue(applies)
def test_check_script_common(self):
self.rule = self.xml_utils.prepare_sections()
lines = FileHelper.get_file_content(os.path.join(
self.dirname, settings.check_script), "rb", method=True)
common = [x for x in lines if '. /usr/share/preupgrade/common.sh' in x]
self.assertTrue(common)
def test_check_script_requires(self):
self.rule = self.xml_utils.prepare_sections()
lines = FileHelper.get_file_content(os.path.join(
self.dirname, settings.check_script), "rb", method=True)
check_rpm_to = [x for x in lines if 'check_rpm_to "bash" "sed"' in x]
self.assertTrue(check_rpm_to)