mirror of
https://gitlab.com/apparmor/apparmor.git
synced 2025-03-04 16:35:02 +01:00
a2d3a382a8
Multiple flags can be given as string (will be split) or as array. Also add some tests confirming that everything works as expected.
97 lines
4.7 KiB
Python
97 lines
4.7 KiB
Python
#! /usr/bin/python3
|
|
# ------------------------------------------------------------------
|
|
#
|
|
# Copyright (C) 2017 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 published by the Free Software Foundation.
|
|
#
|
|
# ------------------------------------------------------------------
|
|
|
|
import unittest
|
|
from common_test import AATest, setup_all_loops
|
|
|
|
from apparmor.common import AppArmorBug
|
|
from apparmor.profile_storage import ProfileStorage, add_or_remove_flag, split_flags, var_transform
|
|
|
|
class TestUnknownKey(AATest):
|
|
def AASetup(self):
|
|
self.storage = ProfileStorage('/test/foo', 'hat', 'TEST')
|
|
|
|
def test_read(self):
|
|
with self.assertRaises(AppArmorBug):
|
|
self.storage['foo']
|
|
|
|
def test_get(self):
|
|
with self.assertRaises(AppArmorBug):
|
|
self.storage.get('foo')
|
|
|
|
def test_get_with_fallback(self):
|
|
with self.assertRaises(AppArmorBug):
|
|
self.storage.get('foo', 'bar')
|
|
|
|
def test_set(self):
|
|
with self.assertRaises(AppArmorBug):
|
|
self.storage['foo'] = 'bar'
|
|
|
|
class AaTest_add_or_remove_flag(AATest):
|
|
tests = [
|
|
# existing flag(s) flag to change add or remove? expected flags
|
|
([ [], 'complain', True ], ['complain'] ),
|
|
([ [], 'complain', False ], [] ),
|
|
([ ['complain'], 'complain', True ], ['complain'] ),
|
|
([ ['complain'], 'complain', False ], [] ),
|
|
([ [], 'audit', True ], ['audit'] ),
|
|
([ [], 'audit', False ], [] ),
|
|
([ ['complain'], 'audit', True ], ['audit', 'complain'] ),
|
|
([ ['complain'], 'audit', False ], ['complain'] ),
|
|
([ '', 'audit', True ], ['audit'] ),
|
|
([ None, 'audit', False ], [] ),
|
|
([ 'complain', 'audit', True ], ['audit', 'complain'] ),
|
|
([ ' complain ', 'audit', False ], ['complain'] ),
|
|
([ 'audit complain', ['audit', 'complain'], False ], [] ),
|
|
([ 'audit complain', 'audit complain', False ], [] ),
|
|
([ 'audit complain', ['audit', 'enforce'], False ], ['complain'] ),
|
|
([ 'audit complain', 'audit enforce', False ], ['complain'] ),
|
|
([ '', ['audit', 'complain'], True ], ['audit', 'complain'] ),
|
|
([ '', 'audit complain', True ], ['audit', 'complain'] ),
|
|
([ 'audit', ['audit', 'enforce'], True ], ['audit', 'enforce'] ),
|
|
([ 'audit', 'audit enforce', True ], ['audit', 'enforce'] ),
|
|
]
|
|
|
|
def _run_test(self, params, expected):
|
|
new_flags = add_or_remove_flag(params[0], params[1], params[2])
|
|
self.assertEqual(new_flags, expected)
|
|
|
|
class AaTest_split_flags(AATest):
|
|
tests = [
|
|
(None , [] ),
|
|
('' , [] ),
|
|
(' ' , [] ),
|
|
(' , ' , [] ),
|
|
('complain' , ['complain'] ),
|
|
(' complain attach_disconnected' , ['attach_disconnected', 'complain'] ),
|
|
(' complain , attach_disconnected' , ['attach_disconnected', 'complain'] ),
|
|
(' complain , , audit , , ' , ['audit', 'complain'] ),
|
|
]
|
|
|
|
def _run_test(self, params, expected):
|
|
split = split_flags(params)
|
|
self.assertEqual(split, expected)
|
|
|
|
class AaTest_var_transform(AATest):
|
|
tests = [
|
|
(['foo', ''], '"" foo' ),
|
|
(['foo', 'bar'], 'bar foo' ),
|
|
([''], '""' ),
|
|
(['bar baz', 'foo'], '"bar baz" foo' ),
|
|
]
|
|
|
|
def _run_test(self, params, expected):
|
|
self.assertEqual(var_transform(params), expected)
|
|
|
|
|
|
setup_all_loops(__name__)
|
|
if __name__ == '__main__':
|
|
unittest.main(verbosity=1)
|