mirror of
https://gitlab.com/apparmor/apparmor.git
synced 2025-03-04 16:35:02 +01:00
ea9f9aeff2
The tests for convert_regexp() were hidden in common_test.py, where they were never executed. This patch moves them to the new file test-aare.py and also converts the regex_tests.ini to a tests[] array to have the test data inside the test file. (All tests from regex_tests.ini are in test-aare.py, and two tests with prepended and appended path segments were added.) Also add some tests that check the raw behaviour of convert_regexp() - the tests "by example" are probably more useful and for sure more readable ;-) but I want to have some examples of the converted regexes available. Acked-by <timeout>
122 lines
4.1 KiB
Python
Executable file
122 lines
4.1 KiB
Python
Executable file
# ----------------------------------------------------------------------
|
|
# Copyright (C) 2013 Kshitij Gupta <kgupta8592@gmail.com>
|
|
# Copyright (C) 2015 Christian Boltz <apparmor@cboltz.de>
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of version 2 of the GNU General Public
|
|
# License as published by the Free Software Foundation.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# ----------------------------------------------------------------------
|
|
import unittest
|
|
import inspect
|
|
import os
|
|
import shutil
|
|
import sys
|
|
import tempfile
|
|
|
|
|
|
#def test_readkey(self):
|
|
# print("Please press the Y button on the keyboard.")
|
|
# self.assertEqual(apparmor.common.readkey().lower(), 'y', 'Error reading key from shell!')
|
|
|
|
|
|
class AATest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.maxDiff = None
|
|
self.AASetup()
|
|
|
|
def AASetup(self):
|
|
'''override this function if a test needs additional setup steps (instead of overriding setUp())'''
|
|
pass
|
|
|
|
def tearDown(self):
|
|
if self.tmpdir and os.path.exists(self.tmpdir):
|
|
shutil.rmtree(self.tmpdir)
|
|
|
|
self.AATeardown()
|
|
|
|
def AATeardown(self):
|
|
'''override this function if a test needs additional teardown steps (instead of overriding tearDown())'''
|
|
pass
|
|
|
|
def createTmpdir(self):
|
|
self.tmpdir = tempfile.mkdtemp(prefix='aa-test-')
|
|
|
|
def writeTmpfile(self, file, contents):
|
|
if not self.tmpdir:
|
|
self.createTmpdir()
|
|
return write_file(self.tmpdir, file, contents)
|
|
|
|
tests = []
|
|
tmpdir = None
|
|
|
|
class AAParseTest(unittest.TestCase):
|
|
parse_function = None
|
|
|
|
def _test_parse_rule(self, rule):
|
|
self.assertIsNot(self.parse_function, 'Test class did not set a parse_function')
|
|
parsed = self.parse_function(rule)
|
|
self.assertEqual(rule, parsed.serialize(),
|
|
'parse object %s returned "%s", expected "%s"' \
|
|
%(self.parse_function.__doc__, parsed.serialize(), rule))
|
|
|
|
def setup_all_loops(module_name):
|
|
'''call setup_tests_loop() for each class in module_name'''
|
|
for name, obj in inspect.getmembers(sys.modules[module_name]):
|
|
if inspect.isclass(obj):
|
|
if issubclass(obj, unittest.TestCase):
|
|
setup_tests_loop(obj)
|
|
|
|
def setup_tests_loop(test_class):
|
|
'''Create tests in test_class using test_class.tests and self._run_test()
|
|
|
|
test_class.tests should be tuples of (test_data, expected_results)
|
|
test_data and expected_results can be of any type as long as test_class._run_test()
|
|
know how to handle them.
|
|
|
|
A typical definition for _run_test() is:
|
|
def test_class._run_test(self, test_data, expected)
|
|
'''
|
|
|
|
for (i, (test_data, expected)) in enumerate(test_class.tests):
|
|
def stub_test(self, test_data=test_data, expected=expected):
|
|
self._run_test(test_data, expected)
|
|
|
|
stub_test.__doc__ = "test '%s'" % (test_data)
|
|
setattr(test_class, 'test_%d' % (i), stub_test)
|
|
|
|
|
|
def setup_regex_tests(test_class):
|
|
'''Create tests in test_class using test_class.tests and AAParseTest._test_parse_rule()
|
|
|
|
test_class.tests should be tuples of (line, description)
|
|
'''
|
|
for (i, (line, desc)) in enumerate(test_class.tests):
|
|
def stub_test(self, line=line):
|
|
self._test_parse_rule(line)
|
|
|
|
stub_test.__doc__ = "test '%s': %s" % (line, desc)
|
|
setattr(test_class, 'test_%d' % (i), stub_test)
|
|
|
|
def write_file(directory, file, contents):
|
|
'''construct path, write contents to it, and return the constructed path'''
|
|
path = os.path.join(directory, file)
|
|
with open(path, 'w+') as f:
|
|
f.write(contents)
|
|
return path
|
|
|
|
def read_file(path):
|
|
'''read and return file contents'''
|
|
with open(path, 'r') as f:
|
|
return f.read()
|
|
|
|
|
|
|
|
if __name__ == "__main__":
|
|
#import sys;sys.argv = ['', 'Test.test_RegexParser']
|
|
unittest.main()
|