mirror of
https://github.com/samba-team/samba.git
synced 2025-01-11 05:18:09 +03:00
bb88292cee
1. Merge tests for different formats into a for loop, make it easy to read and extend. 2. Add test for invalid formats. Signed-off-by: Joe Guo <joeg@catalyst.net.nz> Reviewed-by: Andrew Bartlett <abartlet@samba.org> Reviewed-by: Douglas Bagnall <douglas.bagnall@catalyst.net.nz>
108 lines
3.5 KiB
Python
108 lines
3.5 KiB
Python
# Unix SMB/CIFS implementation.
|
|
# Copyright (C) Jelmer Vernooij <jelmer@samba.org> 2007
|
|
#
|
|
# 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.dcerpc.misc."""
|
|
|
|
from samba.dcerpc import misc
|
|
import samba.tests
|
|
from samba.compat import PY3
|
|
|
|
text1 = "76f53846-a7c2-476a-ae2c-20e2b80d7b34"
|
|
text2 = "344edffa-330a-4b39-b96e-2c34da52e8b1"
|
|
text3 = "00112233-4455-6677-8899-aabbccddeeff"
|
|
|
|
|
|
if PY3:
|
|
# cmp() exists only in Python 2
|
|
def cmp(a, b):
|
|
return (a > b) - (a < b)
|
|
|
|
|
|
class GUIDTests(samba.tests.TestCase):
|
|
|
|
def test_str(self):
|
|
guid = misc.GUID(text1)
|
|
self.assertEquals(text1, str(guid))
|
|
|
|
def test_repr(self):
|
|
guid = misc.GUID(text1)
|
|
self.assertEquals("GUID('%s')" % text1, repr(guid))
|
|
|
|
def test_compare_different(self):
|
|
guid1 = misc.GUID(text1)
|
|
guid2 = misc.GUID(text2)
|
|
self.assertFalse(guid1 == guid2)
|
|
self.assertGreater(guid1, guid2)
|
|
self.assertTrue(cmp(guid1, guid2) > 0)
|
|
|
|
def test_compare_same(self):
|
|
guid1 = misc.GUID(text1)
|
|
guid2 = misc.GUID(text1)
|
|
self.assertTrue(guid1 == guid2)
|
|
self.assertEquals(guid1, guid2)
|
|
self.assertEquals(0, cmp(guid1, guid2))
|
|
|
|
def test_valid_formats(self):
|
|
fmts = [
|
|
"00112233-4455-6677-8899-aabbccddeeff", # 36
|
|
b"00112233-4455-6677-8899-aabbccddeeff", # 36 as bytes
|
|
"{00112233-4455-6677-8899-aabbccddeeff}", # 38
|
|
|
|
"33221100554477668899aabbccddeeff", # 32
|
|
b"33221100554477668899aabbccddeeff", # 32 as bytes
|
|
|
|
# 16 as hex bytes
|
|
b"\x33\x22\x11\x00\x55\x44\x77\x66\x88\x99\xaa\xbb\xcc\xdd\xee\xff"
|
|
]
|
|
for fmt in fmts:
|
|
guid = misc.GUID(fmt)
|
|
self.assertEquals(text3, str(guid))
|
|
|
|
def test_invalid_formats(self):
|
|
fmts = [
|
|
"00112233-4455-6677-8899-aabbccddee", # 34
|
|
"{33221100554477668899aabbccddeeff}",
|
|
"33221100554477668899aabbccddee", # 30
|
|
"\\x33\\x22\\x11\\x00\\x55\\x44\\x77\\x66\\x88\\x99\\xaa\\xbb\\xcc\\xdd\\xee\\xff",
|
|
r"\x33\x22\x11\x00\x55\x44\x77\x66\x88\x99\xaa\xbb\xcc\xdd\xee\xff",
|
|
]
|
|
for fmt in fmts:
|
|
try:
|
|
misc.GUID(fmt)
|
|
except samba.NTSTATUSError:
|
|
# invalid formats should get this error
|
|
continue
|
|
else:
|
|
# otherwise, test fail
|
|
self.fail()
|
|
|
|
|
|
class PolicyHandleTests(samba.tests.TestCase):
|
|
|
|
def test_init(self):
|
|
x = misc.policy_handle(text1, 1)
|
|
self.assertEquals(1, x.handle_type)
|
|
self.assertEquals(text1, str(x.uuid))
|
|
|
|
def test_repr(self):
|
|
x = misc.policy_handle(text1, 42)
|
|
self.assertEquals("policy_handle(%d, '%s')" % (42, text1), repr(x))
|
|
|
|
def test_str(self):
|
|
x = misc.policy_handle(text1, 42)
|
|
self.assertEquals("%d, %s" % (42, text1), str(x))
|