python/samba/tests/common.py

   1 # Unix SMB/CIFS implementation. Tests for common.py routines
   2 # Copyright (C) Andrew Tridgell 2011
   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.common"""
  19
  20 import samba
  21 import os
  22 import samba.tests
  23 from samba.common import normalise_int32
  24 from samba.samdb import dsdb_Dn
  25
  26
  27 class CommonTests(samba.tests.TestCaseInTempDir):
  28
  29     def test_normalise_int32(self):
  30         self.assertEqual('17', normalise_int32(17))
  31         self.assertEqual('17', normalise_int32('17'))
  32         self.assertEqual('-123', normalise_int32('-123'))
  33         self.assertEqual('-1294967296', normalise_int32('3000000000'))
  34
  35     def test_dsdb_Dn_binary(self):
  36         url = self.tempdir + "/test_dsdb_Dn_binary.ldb"
  37         sam = samba.Ldb(url=url)
  38         dn1 = dsdb_Dn(sam, "DC=foo,DC=bar")
  39         dn2 = dsdb_Dn(sam, "B:8:0000000D:<GUID=b3f0ec29-17f4-452a-b002-963e1909d101>;DC=samba,DC=example,DC=com")
  40         self.assertEqual(dn2.binary, "0000000D")
  41         self.assertEqual(13, dn2.get_binary_integer())
  42         os.unlink(url)
  43
  44     def test_dsdb_Dn_sorted(self):
  45         url = self.tempdir + "/test_dsdb_Dn_sorted.ldb"
  46         sam = samba.Ldb(url=url)
  47         try:
  48             dn1 = dsdb_Dn(sam, "B:8:0000000D:<GUID=b3f0ec29-17f4-452a-b002-963e1909d101>;OU=dn1,DC=samba,DC=example,DC=com")
  49             dn2 = dsdb_Dn(sam, "B:8:0000000C:<GUID=b3f0ec29-17f4-452a-b002-963e1909d101>;OU=dn1,DC=samba,DC=example,DC=com")
  50             dn3 = dsdb_Dn(sam, "B:8:0000000F:<GUID=00000000-17f4-452a-b002-963e1909d101>;OU=dn3,DC=samba,DC=example,DC=com")
  51             dn4 = dsdb_Dn(sam, "B:8:00000000:<GUID=ffffffff-17f4-452a-b002-963e1909d101>;OU=dn4,DC=samba,DC=example,DC=com")
  52             dn5 = dsdb_Dn(sam, "<GUID=ffffffff-27f4-452a-b002-963e1909d101>;OU=dn5,DC=samba,DC=example,DC=com")
  53             dn6 = dsdb_Dn(sam, "<GUID=00000000-27f4-452a-b002-963e1909d101>;OU=dn6,DC=samba,DC=example,DC=com")
  54             unsorted_links14 = [dn1, dn2, dn3, dn4]
  55             sorted_vals14 = [str(dn) for dn in sorted(unsorted_links14)]
  56             self.assertEqual(sorted_vals14[0], str(dn3))
  57             self.assertEqual(sorted_vals14[1], str(dn2))
  58             self.assertEqual(sorted_vals14[2], str(dn1))
  59             self.assertEqual(sorted_vals14[3], str(dn4))
  60             unsorted_links56 = [dn5, dn6]
  61             sorted_vals56 = [str(dn) for dn in sorted(unsorted_links56)]
  62             self.assertEqual(sorted_vals56[0], str(dn6))
  63             self.assertEqual(sorted_vals56[1], str(dn5))
  64         finally:
  65             del sam
  66             os.unlink(url)