#!/usr/bin/env python # Simple tests for the ldb python bindings. # Copyright (C) 2007 Jelmer Vernooij import os from unittest import TestCase import ldb def filename(): import tempfile try: dir_prefix = os.path.join(os.environ["SELFTEST_PREFIX"], "tmp") except KeyError: dir_prefix = None return tempfile.mktemp(dir=dir_prefix) class NoContextTests(TestCase): def test_valid_attr_name(self): self.assertTrue(ldb.valid_attr_name("foo")) self.assertFalse(ldb.valid_attr_name("24foo")) def test_timestring(self): self.assertEquals("19700101000000.0Z", ldb.timestring(0)) self.assertEquals("20071119191012.0Z", ldb.timestring(1195499412)) def test_string_to_time(self): self.assertEquals(0, ldb.string_to_time("19700101000000.0Z")) self.assertEquals(1195499412, ldb.string_to_time("20071119191012.0Z")) def test_binary_encode(self): encoded = ldb.binary_encode('test\\x') decoded = ldb.binary_decode(encoded) self.assertEquals(decoded, 'test\\x') class SimpleLdb(TestCase): def test_connect(self): ldb.Ldb(filename()) def test_connect_none(self): ldb.Ldb() def test_connect_later(self): x = ldb.Ldb() x.connect(filename()) def test_repr(self): x = ldb.Ldb() self.assertTrue(repr(x).startswith("]", repr(x.modules())) def test_search(self): l = ldb.Ldb(filename()) self.assertEquals(len(l.search()), 0) def test_search_controls(self): l = ldb.Ldb(filename()) self.assertEquals(len(l.search(controls=["paged_results:0:5"])), 0) def test_search_attrs(self): l = ldb.Ldb(filename()) self.assertEquals(len(l.search(ldb.Dn(l, ""), ldb.SCOPE_SUBTREE, "(dc=*)", ["dc"])), 0) def test_search_string_dn(self): l = ldb.Ldb(filename()) self.assertEquals(len(l.search("", ldb.SCOPE_SUBTREE, "(dc=*)", ["dc"])), 0) def test_search_attr_string(self): l = ldb.Ldb(filename()) self.assertRaises(TypeError, l.search, attrs="dc") def test_opaque(self): l = ldb.Ldb(filename()) l.set_opaque("my_opaque", l) self.assertTrue(l.get_opaque("my_opaque") is not None) self.assertEquals(None, l.get_opaque("unknown")) def test_search_scope_base(self): l = ldb.Ldb(filename()) self.assertEquals(len(l.search(ldb.Dn(l, "dc=foo1"), ldb.SCOPE_ONELEVEL)), 0) def test_delete(self): l = ldb.Ldb(filename()) self.assertRaises(ldb.LdbError, lambda: l.delete(ldb.Dn(l, "dc=foo2"))) def test_delete_w_unhandled_ctrl(self): l = ldb.Ldb(filename()) m = ldb.Message() m.dn = ldb.Dn(l, "dc=foo1") m["b"] = ["a"] l.add(m) self.assertRaises(ldb.LdbError, lambda: l.delete(m.dn, ["search_options:1:2"])) l.delete(m.dn) def test_contains(self): name = filename() l = ldb.Ldb(name) self.assertFalse(ldb.Dn(l, "dc=foo3") in l) l = ldb.Ldb(name) m = ldb.Message() m.dn = ldb.Dn(l, "dc=foo3") m["b"] = ["a"] l.add(m) try: self.assertTrue(ldb.Dn(l, "dc=foo3") in l) finally: l.delete(m.dn) def test_get_config_basedn(self): l = ldb.Ldb(filename()) self.assertEquals(None, l.get_config_basedn()) def test_get_root_basedn(self): l = ldb.Ldb(filename()) self.assertEquals(None, l.get_root_basedn()) def test_get_schema_basedn(self): l = ldb.Ldb(filename()) self.assertEquals(None, l.get_schema_basedn()) def test_get_default_basedn(self): l = ldb.Ldb(filename()) self.assertEquals(None, l.get_default_basedn()) def test_add(self): l = ldb.Ldb(filename()) m = ldb.Message() m.dn = ldb.Dn(l, "dc=foo4") m["bla"] = "bla" self.assertEquals(len(l.search()), 0) l.add(m) try: self.assertEquals(len(l.search()), 1) finally: l.delete(ldb.Dn(l, "dc=foo4")) def test_add_w_unhandled_ctrl(self): l = ldb.Ldb(filename()) m = ldb.Message() m.dn = ldb.Dn(l, "dc=foo4") m["bla"] = "bla" self.assertEquals(len(l.search()), 0) self.assertRaises(ldb.LdbError, lambda: l.add(m,["search_options:1:2"])) def test_add_dict(self): l = ldb.Ldb(filename()) m = {"dn": ldb.Dn(l, "dc=foo5"), "bla": "bla"} self.assertEquals(len(l.search()), 0) l.add(m) try: self.assertEquals(len(l.search()), 1) finally: l.delete(ldb.Dn(l, "dc=foo5")) def test_add_dict_string_dn(self): l = ldb.Ldb(filename()) m = {"dn": "dc=foo6", "bla": "bla"} self.assertEquals(len(l.search()), 0) l.add(m) try: self.assertEquals(len(l.search()), 1) finally: l.delete(ldb.Dn(l, "dc=foo6")) def test_rename(self): l = ldb.Ldb(filename()) m = ldb.Message() m.dn = ldb.Dn(l, "dc=foo7") m["bla"] = "bla" self.assertEquals(len(l.search()), 0) l.add(m) try: l.rename(ldb.Dn(l, "dc=foo7"), ldb.Dn(l, "dc=bar")) self.assertEquals(len(l.search()), 1) finally: l.delete(ldb.Dn(l, "dc=bar")) def test_rename_string_dns(self): l = ldb.Ldb(filename()) m = ldb.Message() m.dn = ldb.Dn(l, "dc=foo8") m["bla"] = "bla" self.assertEquals(len(l.search()), 0) l.add(m) self.assertEquals(len(l.search()), 1) try: l.rename("dc=foo8", "dc=bar") self.assertEquals(len(l.search()), 1) finally: l.delete(ldb.Dn(l, "dc=bar")) def test_modify_delete(self): l = ldb.Ldb(filename()) m = ldb.Message() m.dn = ldb.Dn(l, "dc=modifydelete") m["bla"] = ["1234"] l.add(m) rm = l.search(m.dn)[0] self.assertEquals(["1234"], list(rm["bla"])) try: m = ldb.Message() m.dn = ldb.Dn(l, "dc=modifydelete") m["bla"] = ldb.MessageElement([], ldb.FLAG_MOD_DELETE, "bla") self.assertEquals(ldb.FLAG_MOD_DELETE, m["bla"].flags()) l.modify(m) rm = l.search(m.dn)[0] self.assertEquals(1, len(rm)) rm = l.search(m.dn, attrs=["bla"]) self.assertEquals(0, len(rm)) finally: l.delete(ldb.Dn(l, "dc=modifydelete")) def test_modify_add(self): l = ldb.Ldb(filename()) m = ldb.Message() m.dn = ldb.Dn(l, "dc=add") m["bla"] = ["1234"] l.add(m) try: m = ldb.Message() m.dn = ldb.Dn(l, "dc=add") m["bla"] = ldb.MessageElement(["456"], ldb.FLAG_MOD_ADD, "bla") self.assertEquals(ldb.FLAG_MOD_ADD, m["bla"].flags()) l.modify(m) rm = l.search(m.dn)[0] self.assertEquals(2, len(rm)) self.assertEquals(["1234", "456"], list(rm["bla"])) finally: l.delete(ldb.Dn(l, "dc=add")) def test_modify_replace(self): l = ldb.Ldb(filename()) m = ldb.Message() m.dn = ldb.Dn(l, "dc=modify2") m["bla"] = ["1234", "456"] l.add(m) try: m = ldb.Message() m.dn = ldb.Dn(l, "dc=modify2") m["bla"] = ldb.MessageElement(["789"], ldb.FLAG_MOD_REPLACE, "bla") self.assertEquals(ldb.FLAG_MOD_REPLACE, m["bla"].flags()) l.modify(m) rm = l.search(m.dn)[0] self.assertEquals(2, len(rm)) self.assertEquals(["789"], list(rm["bla"])) rm = l.search(m.dn, attrs=["bla"])[0] self.assertEquals(1, len(rm)) finally: l.delete(ldb.Dn(l, "dc=modify2")) def test_modify_flags_change(self): l = ldb.Ldb(filename()) m = ldb.Message() m.dn = ldb.Dn(l, "dc=add") m["bla"] = ["1234"] l.add(m) try: m = ldb.Message() m.dn = ldb.Dn(l, "dc=add") m["bla"] = ldb.MessageElement(["456"], ldb.FLAG_MOD_ADD, "bla") self.assertEquals(ldb.FLAG_MOD_ADD, m["bla"].flags()) l.modify(m) rm = l.search(m.dn)[0] self.assertEquals(2, len(rm)) self.assertEquals(["1234", "456"], list(rm["bla"])) # Now create another modify, but switch the flags before we do it m["bla"] = ldb.MessageElement(["456"], ldb.FLAG_MOD_ADD, "bla") m["bla"].set_flags(ldb.FLAG_MOD_DELETE) l.modify(m) rm = l.search(m.dn, attrs=["bla"])[0] self.assertEquals(1, len(rm)) self.assertEquals(["1234"], list(rm["bla"])) finally: l.delete(ldb.Dn(l, "dc=add")) def test_transaction_commit(self): l = ldb.Ldb(filename()) l.transaction_start() m = ldb.Message(ldb.Dn(l, "dc=foo9")) m["foo"] = ["bar"] l.add(m) l.transaction_commit() l.delete(m.dn) def test_transaction_cancel(self): l = ldb.Ldb(filename()) l.transaction_start() m = ldb.Message(ldb.Dn(l, "dc=foo10")) m["foo"] = ["bar"] l.add(m) l.transaction_cancel() self.assertEquals(0, len(l.search(ldb.Dn(l, "dc=foo10")))) def test_set_debug(self): def my_report_fn(level, text): pass l = ldb.Ldb(filename()) l.set_debug(my_report_fn) def test_zero_byte_string(self): """Testing we do not get trapped in the \0 byte in a property string.""" l = ldb.Ldb(filename()) l.add({ "dn" : "dc=somedn", "objectclass" : "user", "cN" : "LDAPtestUSER", "givenname" : "ldap", "displayname" : "foo\0bar", }) res = l.search(expression="(dn=dc=somedn)") self.assertEquals("foo\0bar", res[0]["displayname"][0]) def test_no_crash_broken_expr(self): l = ldb.Ldb(filename()) self.assertRaises(ldb.LdbError,lambda: l.search("", ldb.SCOPE_SUBTREE, "&(dc=*)(dn=*)", ["dc"])) class DnTests(TestCase): def setUp(self): super(DnTests, self).setUp() self.ldb = ldb.Ldb(filename()) def test_set_dn_invalid(self): x = ldb.Message() def assign(): x.dn = "astring" self.assertRaises(TypeError, assign) def test_eq(self): x = ldb.Dn(self.ldb, "dc=foo11,bar=bloe") y = ldb.Dn(self.ldb, "dc=foo11,bar=bloe") self.assertEquals(x, y) y = ldb.Dn(self.ldb, "dc=foo11,bar=blie") self.assertNotEquals(x, y) def test_str(self): x = ldb.Dn(self.ldb, "dc=foo12,bar=bloe") self.assertEquals(x.__str__(), "dc=foo12,bar=bloe") def test_repr(self): x = ldb.Dn(self.ldb, "dc=foo13,bla=blie") self.assertEquals(x.__repr__(), "Dn('dc=foo13,bla=blie')") def test_get_casefold(self): x = ldb.Dn(self.ldb, "dc=foo14,bar=bloe") self.assertEquals(x.get_casefold(), "DC=FOO14,BAR=bloe") def test_validate(self): x = ldb.Dn(self.ldb, "dc=foo15,bar=bloe") self.assertTrue(x.validate()) def test_parent(self): x = ldb.Dn(self.ldb, "dc=foo16,bar=bloe") self.assertEquals("bar=bloe", x.parent().__str__()) def test_parent_nonexistent(self): x = ldb.Dn(self.ldb, "@BLA") self.assertEquals(None, x.parent()) def test_is_valid(self): x = ldb.Dn(self.ldb, "dc=foo18,dc=bloe") self.assertTrue(x.is_valid()) x = ldb.Dn(self.ldb, "") self.assertTrue(x.is_valid()) def test_is_special(self): x = ldb.Dn(self.ldb, "dc=foo19,bar=bloe") self.assertFalse(x.is_special()) x = ldb.Dn(self.ldb, "@FOOBAR") self.assertTrue(x.is_special()) def test_check_special(self): x = ldb.Dn(self.ldb, "dc=foo20,bar=bloe") self.assertFalse(x.check_special("FOOBAR")) x = ldb.Dn(self.ldb, "@FOOBAR") self.assertTrue(x.check_special("@FOOBAR")) def test_len(self): x = ldb.Dn(self.ldb, "dc=foo21,bar=bloe") self.assertEquals(2, len(x)) x = ldb.Dn(self.ldb, "dc=foo21") self.assertEquals(1, len(x)) def test_add_child(self): x = ldb.Dn(self.ldb, "dc=foo22,bar=bloe") self.assertTrue(x.add_child(ldb.Dn(self.ldb, "bla=bloe"))) self.assertEquals("bla=bloe,dc=foo22,bar=bloe", x.__str__()) def test_add_base(self): x = ldb.Dn(self.ldb, "dc=foo23,bar=bloe") base = ldb.Dn(self.ldb, "bla=bloe") self.assertTrue(x.add_base(base)) self.assertEquals("dc=foo23,bar=bloe,bla=bloe", x.__str__()) def test_add(self): x = ldb.Dn(self.ldb, "dc=foo24") y = ldb.Dn(self.ldb, "bar=bla") self.assertEquals("dc=foo24,bar=bla", str(x + y)) def test_remove_base_components(self): x = ldb.Dn(self.ldb, "dc=foo24,dc=samba,dc=org") x.remove_base_components(len(x)-1) self.assertEquals("dc=foo24", str(x)) def test_parse_ldif(self): msgs = self.ldb.parse_ldif("dn: foo=bar\n") msg = msgs.next() self.assertEquals("foo=bar", str(msg[1].dn)) self.assertTrue(isinstance(msg[1], ldb.Message)) ldif = self.ldb.write_ldif(msg[1], ldb.CHANGETYPE_NONE) self.assertEquals("dn: foo=bar\n\n", ldif) def test_parse_ldif_more(self): msgs = self.ldb.parse_ldif("dn: foo=bar\n\n\ndn: bar=bar") msg = msgs.next() self.assertEquals("foo=bar", str(msg[1].dn)) msg = msgs.next() self.assertEquals("bar=bar", str(msg[1].dn)) def test_canonical_string(self): x = ldb.Dn(self.ldb, "dc=foo25,bar=bloe") self.assertEquals("/bloe/foo25", x.canonical_str()) def test_canonical_ex_string(self): x = ldb.Dn(self.ldb, "dc=foo26,bar=bloe") self.assertEquals("/bloe\nfoo26", x.canonical_ex_str()) def test_ldb_is_child_of(self): """Testing ldb_dn_compare_dn""" dn1 = ldb.Dn(self.ldb, "dc=base") dn2 = ldb.Dn(self.ldb, "cn=foo,dc=base") dn3 = ldb.Dn(self.ldb, "cn=bar,dc=base") dn4 = ldb.Dn(self.ldb, "cn=baz,cn=bar,dc=base") self.assertTrue(dn2.is_child_of(dn1)) self.assertTrue(dn4.is_child_of(dn1)) self.assertTrue(dn4.is_child_of(dn3)) self.assertFalse(dn3.is_child_of(dn2)) self.assertFalse(dn1.is_child_of(dn4)) class LdbMsgTests(TestCase): def setUp(self): super(LdbMsgTests, self).setUp() self.msg = ldb.Message() def test_init_dn(self): self.msg = ldb.Message(ldb.Dn(ldb.Ldb(), "dc=foo27")) self.assertEquals("dc=foo27", str(self.msg.dn)) def test_iter_items(self): self.assertEquals(0, len(self.msg.items())) self.msg.dn = ldb.Dn(ldb.Ldb(filename()), "dc=foo28") self.assertEquals(1, len(self.msg.items())) def test_repr(self): self.msg.dn = ldb.Dn(ldb.Ldb(filename()), "dc=foo29") self.msg["dc"] = "foo" self.assertEquals("Message({'dn': Dn('dc=foo29'), 'dc': MessageElement(['foo'])})", repr(self.msg)) def test_len(self): self.assertEquals(0, len(self.msg)) def test_notpresent(self): self.assertRaises(KeyError, lambda: self.msg["foo"]) def test_del(self): del self.msg["foo"] def test_add(self): self.msg.add(ldb.MessageElement(["456"], ldb.FLAG_MOD_ADD, "bla")) def test_elements_empty(self): self.assertEquals([], self.msg.elements()) def test_elements(self): el = ldb.MessageElement(["456"], ldb.FLAG_MOD_ADD, "bla") self.msg.add(el) self.assertEquals([el], self.msg.elements()) def test_add_value(self): self.assertEquals(0, len(self.msg)) self.msg["foo"] = ["foo"] self.assertEquals(1, len(self.msg)) def test_add_value_multiple(self): self.assertEquals(0, len(self.msg)) self.msg["foo"] = ["foo", "bla"] self.assertEquals(1, len(self.msg)) self.assertEquals(["foo", "bla"], list(self.msg["foo"])) def test_set_value(self): self.msg["foo"] = ["fool"] self.assertEquals(["fool"], list(self.msg["foo"])) self.msg["foo"] = ["bar"] self.assertEquals(["bar"], list(self.msg["foo"])) def test_keys(self): self.msg.dn = ldb.Dn(ldb.Ldb(filename()), "@BASEINFO") self.msg["foo"] = ["bla"] self.msg["bar"] = ["bla"] self.assertEquals(["dn", "foo", "bar"], self.msg.keys()) def test_dn(self): self.msg.dn = ldb.Dn(ldb.Ldb(filename()), "@BASEINFO") self.assertEquals("@BASEINFO", self.msg.dn.__str__()) def test_get_dn(self): self.msg.dn = ldb.Dn(ldb.Ldb(filename()), "@BASEINFO") self.assertEquals("@BASEINFO", self.msg.get("dn").__str__()) def test_get_invalid(self): self.msg.dn = ldb.Dn(ldb.Ldb(filename()), "@BASEINFO") self.assertRaises(TypeError, self.msg.get, 42) def test_get_other(self): self.msg["foo"] = ["bar"] self.assertEquals("bar", self.msg.get("foo")[0]) self.assertEquals("bar", self.msg.get("foo", idx=0)) self.assertEquals(None, self.msg.get("foo", idx=1)) self.assertEquals("", self.msg.get("foo", default='', idx=1)) def test_get_default(self): self.assertEquals(None, self.msg.get("tatayoyo", idx=0)) self.assertEquals("anniecordie", self.msg.get("tatayoyo", "anniecordie")) def test_get_unknown(self): self.assertEquals(None, self.msg.get("lalalala")) def test_msg_diff(self): l = ldb.Ldb() msgs = l.parse_ldif("dn: foo=bar\nfoo: bar\nbaz: do\n\ndn: foo=bar\nfoo: bar\nbaz: dont\n") msg1 = msgs.next()[1] msg2 = msgs.next()[1] msgdiff = l.msg_diff(msg1, msg2) self.assertEquals("foo=bar", msgdiff.get("dn").__str__()) self.assertRaises(KeyError, lambda: msgdiff["foo"]) self.assertEquals(1, len(msgdiff)) def test_equal_empty(self): msg1 = ldb.Message() msg2 = ldb.Message() self.assertEquals(msg1, msg2) def test_equal_simplel(self): db = ldb.Ldb(filename()) msg1 = ldb.Message() msg1.dn = ldb.Dn(db, "foo=bar") msg2 = ldb.Message() msg2.dn = ldb.Dn(db, "foo=bar") self.assertEquals(msg1, msg2) msg1['foo'] = 'bar' msg2['foo'] = 'bar' self.assertEquals(msg1, msg2) msg2['foo'] = 'blie' self.assertNotEquals(msg1, msg2) msg2['foo'] = 'blie' def test_from_dict(self): rec = {"dn": "dc=fromdict", "a1": ["a1-val1", "a1-val1"]} l = ldb.Ldb() # check different types of input Flags for flags in [ldb.FLAG_MOD_ADD, ldb.FLAG_MOD_REPLACE, ldb.FLAG_MOD_DELETE]: m = ldb.Message.from_dict(l, rec, flags) self.assertEquals(rec["a1"], list(m["a1"])) self.assertEquals(flags, m["a1"].flags()) # check input params self.assertRaises(TypeError, ldb.Message.from_dict, dict(), rec, ldb.FLAG_MOD_REPLACE) self.assertRaises(TypeError, ldb.Message.from_dict, l, list(), ldb.FLAG_MOD_REPLACE) self.assertRaises(ValueError, ldb.Message.from_dict, l, rec, 0) # Message.from_dict expects dictionary with 'dn' err_rec = {"a1": ["a1-val1", "a1-val1"]} self.assertRaises(TypeError, ldb.Message.from_dict, l, err_rec, ldb.FLAG_MOD_REPLACE) class MessageElementTests(TestCase): def test_cmp_element(self): x = ldb.MessageElement(["foo"]) y = ldb.MessageElement(["foo"]) z = ldb.MessageElement(["bzr"]) self.assertEquals(x, y) self.assertNotEquals(x, z) def test_create_iterable(self): x = ldb.MessageElement(["foo"]) self.assertEquals(["foo"], list(x)) def test_repr(self): x = ldb.MessageElement(["foo"]) self.assertEquals("MessageElement(['foo'])", repr(x)) x = ldb.MessageElement(["foo", "bla"]) self.assertEquals(2, len(x)) self.assertEquals("MessageElement(['foo','bla'])", repr(x)) def test_get_item(self): x = ldb.MessageElement(["foo", "bar"]) self.assertEquals("foo", x[0]) self.assertEquals("bar", x[1]) self.assertEquals("bar", x[-1]) self.assertRaises(IndexError, lambda: x[45]) def test_len(self): x = ldb.MessageElement(["foo", "bar"]) self.assertEquals(2, len(x)) def test_eq(self): x = ldb.MessageElement(["foo", "bar"]) y = ldb.MessageElement(["foo", "bar"]) self.assertEquals(y, x) x = ldb.MessageElement(["foo"]) self.assertNotEquals(y, x) y = ldb.MessageElement(["foo"]) self.assertEquals(y, x) def test_extended(self): el = ldb.MessageElement(["456"], ldb.FLAG_MOD_ADD, "bla") self.assertEquals("MessageElement(['456'])", repr(el)) class ModuleTests(TestCase): def test_register_module(self): class ExampleModule: name = "example" ldb.register_module(ExampleModule) def test_use_module(self): ops = [] class ExampleModule: name = "bla" def __init__(self, ldb, next): ops.append("init") self.next = next def search(self, *args, **kwargs): return self.next.search(*args, **kwargs) def request(self, *args, **kwargs): pass name = filename() ldb.register_module(ExampleModule) if os.path.exists(name): os.unlink(name) l = ldb.Ldb(name) l.add({"dn": "@MODULES", "@LIST": "bla"}) self.assertEquals([], ops) l = ldb.Ldb(name) self.assertEquals(["init"], ops) class LdbResultTests(TestCase): def setUp(self): super(LdbResultTests, self).setUp() name = filename() self.name = name if os.path.exists(name): os.unlink(name) self.l = ldb.Ldb(name) self.l.add({"dn": "DC=SAMBA,DC=ORG", "name": "samba.org"}) self.l.add({"dn": "OU=ADMIN,DC=SAMBA,DC=ORG", "name": "Admins"}) self.l.add({"dn": "OU=USERS,DC=SAMBA,DC=ORG", "name": "Users"}) self.l.add({"dn": "OU=OU1,DC=SAMBA,DC=ORG", "name": "OU #1"}) self.l.add({"dn": "OU=OU2,DC=SAMBA,DC=ORG", "name": "OU #2"}) self.l.add({"dn": "OU=OU3,DC=SAMBA,DC=ORG", "name": "OU #3"}) self.l.add({"dn": "OU=OU4,DC=SAMBA,DC=ORG", "name": "OU #4"}) self.l.add({"dn": "OU=OU5,DC=SAMBA,DC=ORG", "name": "OU #5"}) self.l.add({"dn": "OU=OU6,DC=SAMBA,DC=ORG", "name": "OU #6"}) self.l.add({"dn": "OU=OU7,DC=SAMBA,DC=ORG", "name": "OU #7"}) self.l.add({"dn": "OU=OU8,DC=SAMBA,DC=ORG", "name": "OU #8"}) self.l.add({"dn": "OU=OU9,DC=SAMBA,DC=ORG", "name": "OU #9"}) self.l.add({"dn": "OU=OU10,DC=SAMBA,DC=ORG", "name": "OU #10"}) def tearDown(self): super(LdbResultTests, self).tearDown() if os.path.exists(self.name): os.unlink(self.name) def test_return_type(self): res = self.l.search() self.assertEquals(str(res), "") def test_get_msgs(self): res = self.l.search() list = res.msgs def test_get_controls(self): res = self.l.search() list = res.controls def test_get_referals(self): res = self.l.search() list = res.referals def test_iter_msgs(self): found = False for l in self.l.search().msgs: if str(l.dn) == "OU=OU10,DC=SAMBA,DC=ORG": found = True self.assertTrue(found) def test_iter_msgs_count(self): self.assertTrue(self.l.search().count > 0) # 13 objects has been added to the DC=SAMBA, DC=ORG self.assertEqual(self.l.search(base="DC=SAMBA,DC=ORG").count, 13) def test_iter_controls(self): res = self.l.search().controls it = iter(res) def test_create_control(self): self.assertRaises(ValueError, ldb.Control, self.l, "tatayoyo:0") c = ldb.Control(self.l, "relax:1") self.assertEquals(c.critical, True) self.assertEquals(c.oid, "1.3.6.1.4.1.4203.666.5.12") def test_iter_refs(self): res = self.l.search().referals it = iter(res) def test_iter_as_sequence_msgs(self): found = False res = self.l.search().msgs for i in range(0, len(res)): l = res[i] if str(l.dn) == "OU=OU10,DC=SAMBA,DC=ORG": found = True self.assertTrue(found) def test_iter_as_sequence(self): found = False res = self.l.search() for i in range(0, len(res)): l = res[i] if str(l.dn) == "OU=OU10,DC=SAMBA,DC=ORG": found = True self.assertTrue(found) class VersionTests(TestCase): def test_version(self): self.assertTrue(isinstance(ldb.__version__, str)) if __name__ == '__main__': import unittest unittest.TestProgram()