PEP8: fix E401: multiple imports on one line
[amitay/samba.git] / 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 *
24 from samba.samdb import SamDB
25
26
27 class CommonTests(samba.tests.TestCaseInTempDir):
28
29     def test_normalise_int32(self):
30         self.assertEquals('17', normalise_int32(17))
31         self.assertEquals('17', normalise_int32('17'))
32         self.assertEquals('-123', normalise_int32('-123'))
33         self.assertEquals('-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.assertEquals(dn2.binary, "0000000D")
41         self.assertEquals(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.assertEquals(sorted_vals14[0], str(dn3))
57             self.assertEquals(sorted_vals14[1], str(dn2))
58             self.assertEquals(sorted_vals14[2], str(dn1))
59             self.assertEquals(sorted_vals14[3], str(dn4))
60             unsorted_links56 = [dn5, dn6]
61             sorted_vals56 = [str(dn) for dn in sorted(unsorted_links56)]
62             self.assertEquals(sorted_vals56[0], str(dn6))
63             self.assertEquals(sorted_vals56[1], str(dn5))
64         finally:
65             del sam
66             os.unlink(url)