1
0
mirror of https://github.com/samba-team/samba.git synced 2024-12-24 21:34:56 +03:00
samba-mirror/python/samba/tests/dsdb_api.py
Andrew Bartlett fb6c0b9e2a pydsdb: Add API to return strings of known UF_ flags
Signed-off-by: Andrew Bartlett <abartlet@samba.org>
Reviewed-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
2021-09-02 05:03:31 +00:00

58 lines
1.9 KiB
Python

# Unix SMB/CIFS implementation. Tests for dsdb
# Copyright (C) Andrew Bartlett <abartlet@samba.org> 2021
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 3 of the License, or
# (at your option) any later version.
#
# 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.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
#
"""Tests for samba.dsdb."""
from samba.tests import TestCase, DynamicTestCase
from samba.dsdb import user_account_control_flag_bit_to_string
import samba
@DynamicTestCase
class DsdbFlagTests(TestCase):
@classmethod
def setUpDynamicTestCases(cls):
for x in dir(samba.dsdb):
if x.startswith("UF_"):
cls.generate_dynamic_test("test",
x,
x,
getattr(samba.dsdb, x))
def _test_with_args(self, uf_string, uf_bit):
self.assertEqual(user_account_control_flag_bit_to_string(uf_bit),
uf_string)
def test_not_a_flag(self):
self.assertRaises(KeyError,
user_account_control_flag_bit_to_string,
0xabcdef)
def test_too_long(self):
self.assertRaises(OverflowError,
user_account_control_flag_bit_to_string,
0xabcdefffff)
def test_way_too_long(self):
self.assertRaises(OverflowError,
user_account_control_flag_bit_to_string,
0xabcdeffffffffffff)