def test_compose(self): expected_contents = ["failed", "fixed", "needs_action", "needs_inspection", "not_applicable", "pass"] for content in expected_contents: compose_xml = ComposeXML() result_dir = os.path.join(self.result_dir, content) compose_xml.collect_group_xmls(self.dir_name, content=content) self.assertTrue(os.path.exists(os.path.join(result_dir, "group.xml"))) self.assertFalse(os.path.exists(os.path.join(result_dir, "all-xccdf.xml")))
def test_final_compose(self): expected_contents = ["failed", "fixed", "needs_action", "needs_inspection", "not_applicable", "pass"] for content in expected_contents: compose_xml = ComposeXML() dir_name = os.path.join(self.temp_dir, FOO_DIR, "dummy") compose_xml.collect_group_xmls(dir_name, content=content) xccdf_compose = XCCDFCompose(os.path.join(self.temp_dir, FOO_DIR)) xccdf_compose.generate_xml() all_xccdf = os.path.join(self.result_dir, settings.content_file) self.assertTrue(os.path.exists(all_xccdf)) dummy_lines = utils.get_file_content(all_xccdf, "rb")
def test_compose(self): expected_contents = [ 'failed', 'fixed', 'needs_action', 'needs_inspection', 'not_applicable', 'pass' ] for content in expected_contents: compose_xml = ComposeXML() result_dir = os.path.join(self.result_dir, content) compose_xml.collect_group_xmls(self.dir_name, content=content) self.assertTrue( os.path.exists(os.path.join(result_dir, 'group.xml'))) self.assertFalse( os.path.exists(os.path.join(result_dir, 'all-xccdf.xml')))
def test_final_compose(self): expected_contents = [ 'failed', 'fixed', 'needs_action', 'needs_inspection', 'not_applicable', 'pass' ] for content in expected_contents: compose_xml = ComposeXML() dir_name = os.path.join(self.temp_dir, FOO_DIR, 'dummy') compose_xml.collect_group_xmls(dir_name, content=content) xccdf_compose = XCCDFCompose(os.path.join(self.temp_dir, FOO_DIR)) xccdf_compose.generate_xml() all_xccdf = os.path.join(self.result_dir, settings.content_file) self.assertTrue(os.path.exists(all_xccdf)) dummy_lines = utils.get_file_content(all_xccdf, 'rb')
def setUp(self): dir_name = os.path.join(os.getcwd(), 'tests', 'FOOBAR6_7') self.result_dir = os.path.join(dir_name+variables.result_prefix) dir_name = os.path.join(dir_name, 'dummy') if os.path.exists(self.result_dir): shutil.rmtree(self.result_dir) shutil.copytree(dir_name, self.result_dir) template_file = ComposeXML.get_template_file() self.tree = None try: self.tree = ElementTree.parse(template_file).getroot() except IOError: assert False settings.autocomplete = False self.target_tree = ComposeXML.run_compose(self.tree, self.result_dir) self.assertTrue(self.target_tree)
def prepare_xml(path_name): settings.autocomplete = False template_file = ComposeXML.get_template_file() tree = None try: f = open(template_file, "r") tree = ElementTree.fromstring(f.read()) f.close() except IOError: assert False tree = ComposeXML.run_compose(tree, os.path.dirname(path_name), content=os.path.basename(path_name)) try: file = open(os.path.join(path_name, 'all-xccdf.xml'), "w") file.write(ElementTree.tostring(tree, encoding="utf-8").decode('utf-8')) file.close() except IOError as e: raise
def prepare_xml(path_name): settings.autocomplete = False template_file = ComposeXML.get_template_file() tree = None try: f = open(template_file, "r") tree = ElementTree.fromstring(f.read()) f.close() except IOError: assert False tree = ComposeXML.run_compose(tree, os.path.dirname(path_name), content=os.path.basename(path_name)) try: file = open(os.path.join(path_name, 'all-xccdf.xml'), "w") file.write( ElementTree.tostring(tree, encoding="utf-8").decode('utf-8')) file.close() except IOError as e: raise
def test_unicode_script_author(self): """Test processing of non-ascii characters for author section""" u_author = b'Petr Stod\xc5\xaflka'.decode(settings.defenc) script_file = os.path.join(self.result_dir, "unicode", "dummy_unicode.sh") settings.autocomplete = True self.target_tree = ComposeXML.run_compose(self.tree, self.result_dir) self.assertTrue(self.target_tree) try: lines = get_file_content(script_file, "rb", True) except IOError: assert False author = filter(lambda x: u_author in x, lines) self.assertTrue(author)