python/samba/tests/s3idmapdb.py

   1 # Unix SMB/CIFS implementation.
   2 # Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2007
   3 #
   4 # This program is free software; you can redistribute it and/or modify
   5 # it under the terms of the GNU General Public License as published by
   6 # the Free Software Foundation; either version 3 of the License, or
   7 # (at your option) any later version.
   8 #
   9 # This program is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 # GNU General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU General Public License
  15 # along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16 #
  17
  18 """Tests for samba.samba3."""
  19
  20 from samba.samba3 import IdmapDatabase
  21 from samba.tests import TestCase
  22 import os
  23
  24 for p in ["../../../../../testdata/samba3", "../../../../testdata/samba3"]:
  25     DATADIR = os.path.join(os.path.dirname(__file__), p)
  26     if os.path.exists(DATADIR):
  27         break
  28
  29
  30 class IdmapDbTestCase(TestCase):
  31
  32     def setUp(self):
  33         super().setUp()
  34         self.idmapdb = IdmapDatabase(os.path.join(DATADIR,
  35                                                   "winbindd_idmap"))
  36
  37     def test_user_hwm(self):
  38         self.assertEqual(10000, self.idmapdb.get_user_hwm())
  39
  40     def test_group_hwm(self):
  41         self.assertEqual(10002, self.idmapdb.get_group_hwm())
  42
  43     def test_uids(self):
  44         self.assertEqual(1, len(list(self.idmapdb.uids())))
  45
  46     def test_gids(self):
  47         self.assertEqual(3, len(list(self.idmapdb.gids())))
  48
  49     def test_get_user_sid(self):
  50         self.assertEqual(b"S-1-5-21-58189338-3053988021-627566699-501", self.idmapdb.get_user_sid(65534))
  51
  52     def test_get_group_sid(self):
  53         self.assertEqual(b"S-1-5-21-2447931902-1787058256-3961074038-3007", self.idmapdb.get_group_sid(10001))
  54
  55     def tearDown(self):
  56         self.idmapdb.close()
  57         super().tearDown()