mirror of
https://gitlab.com/apparmor/apparmor
synced 2025-08-22 10:07:12 +00:00
6648f2cbeb
In the past, this class and function were used by test-signal_parse.py - which was deleted in April 2016 (5f58d7f124139784b9ba70ce37cc26716bbc4e0f). Maybe there were also other users, but none of them survived.
109 lines
3.3 KiB
Python
Executable File
109 lines
3.3 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 inspect
|
|
import os
|
|
import shutil
|
|
import sys
|
|
import tempfile
|
|
import unittest
|
|
|
|
|
|
# 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())"""
|
|
|
|
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())"""
|
|
|
|
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
|
|
|
|
|
|
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 '{}'".format(test_data)
|
|
setattr(test_class, 'test_{}'.format(i), stub_test)
|
|
|
|
|
|
def setup_aa(aa):
|
|
confdir = os.getenv('__AA_CONFDIR')
|
|
if confdir:
|
|
aa.init_aa(confdir=confdir)
|
|
else:
|
|
aa.init_aa()
|
|
|
|
|
|
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()
|