2 # Simple tests for the ldb python bindings.
3 # Copyright (C) 2007 Jelmer Vernooij <jelmer@samba.org>
6 from unittest import TestCase
14 dir_prefix = os.path.join(os.environ["SELFTEST_PREFIX"], "tmp")
17 return tempfile.mktemp(dir=dir_prefix)
20 class NoContextTests(TestCase):
22 def test_valid_attr_name(self):
23 self.assertTrue(ldb.valid_attr_name("foo"))
24 self.assertFalse(ldb.valid_attr_name("24foo"))
26 def test_timestring(self):
27 self.assertEqual("19700101000000.0Z", ldb.timestring(0))
28 self.assertEqual("20071119191012.0Z", ldb.timestring(1195499412))
30 def test_string_to_time(self):
31 self.assertEqual(0, ldb.string_to_time("19700101000000.0Z"))
32 self.assertEqual(1195499412, ldb.string_to_time("20071119191012.0Z"))
34 def test_binary_encode(self):
35 encoded = ldb.binary_encode('test\\x')
36 decoded = ldb.binary_decode(encoded)
37 self.assertEqual(decoded, 'test\\x')
39 class SimpleLdb(TestCase):
41 def test_connect(self):
44 def test_connect_none(self):
47 def test_connect_later(self):
53 self.assertTrue(repr(x).startswith("<ldb connection"))
55 def test_set_create_perms(self):
57 x.set_create_perms(0o600)
59 def test_modules_none(self):
61 self.assertEqual([], x.modules())
63 def test_modules_tdb(self):
64 x = ldb.Ldb(filename())
65 self.assertEqual("[<ldb module 'tdb'>]", repr(x.modules()))
67 def test_search(self):
68 l = ldb.Ldb(filename())
69 self.assertEqual(len(l.search()), 0)
71 def test_search_controls(self):
72 l = ldb.Ldb(filename())
73 self.assertEqual(len(l.search(controls=["paged_results:0:5"])), 0)
75 def test_search_attrs(self):
76 l = ldb.Ldb(filename())
77 self.assertEqual(len(l.search(ldb.Dn(l, ""), ldb.SCOPE_SUBTREE, "(dc=*)", ["dc"])), 0)
79 def test_search_string_dn(self):
80 l = ldb.Ldb(filename())
81 self.assertEqual(len(l.search("", ldb.SCOPE_SUBTREE, "(dc=*)", ["dc"])), 0)
83 def test_search_attr_string(self):
84 l = ldb.Ldb(filename())
85 self.assertRaises(TypeError, l.search, attrs="dc")
87 def test_opaque(self):
88 l = ldb.Ldb(filename())
89 l.set_opaque("my_opaque", l)
90 self.assertTrue(l.get_opaque("my_opaque") is not None)
91 self.assertEqual(None, l.get_opaque("unknown"))
93 def test_search_scope_base(self):
94 l = ldb.Ldb(filename())
95 self.assertEqual(len(l.search(ldb.Dn(l, "dc=foo1"),
96 ldb.SCOPE_ONELEVEL)), 0)
98 def test_delete(self):
99 l = ldb.Ldb(filename())
100 self.assertRaises(ldb.LdbError, lambda: l.delete(ldb.Dn(l, "dc=foo2")))
102 def test_delete_w_unhandled_ctrl(self):
103 l = ldb.Ldb(filename())
105 m.dn = ldb.Dn(l, "dc=foo1")
108 self.assertRaises(ldb.LdbError, lambda: l.delete(m.dn, ["search_options:1:2"]))
111 def test_contains(self):
114 self.assertFalse(ldb.Dn(l, "dc=foo3") in l)
117 m.dn = ldb.Dn(l, "dc=foo3")
121 self.assertTrue(ldb.Dn(l, "dc=foo3") in l)
125 def test_get_config_basedn(self):
126 l = ldb.Ldb(filename())
127 self.assertEqual(None, l.get_config_basedn())
129 def test_get_root_basedn(self):
130 l = ldb.Ldb(filename())
131 self.assertEqual(None, l.get_root_basedn())
133 def test_get_schema_basedn(self):
134 l = ldb.Ldb(filename())
135 self.assertEqual(None, l.get_schema_basedn())
137 def test_get_default_basedn(self):
138 l = ldb.Ldb(filename())
139 self.assertEqual(None, l.get_default_basedn())
142 l = ldb.Ldb(filename())
144 m.dn = ldb.Dn(l, "dc=foo4")
146 self.assertEqual(len(l.search()), 0)
149 self.assertEqual(len(l.search()), 1)
151 l.delete(ldb.Dn(l, "dc=foo4"))
153 def test_add_w_unhandled_ctrl(self):
154 l = ldb.Ldb(filename())
156 m.dn = ldb.Dn(l, "dc=foo4")
158 self.assertEqual(len(l.search()), 0)
159 self.assertRaises(ldb.LdbError, lambda: l.add(m,["search_options:1:2"]))
161 def test_add_dict(self):
162 l = ldb.Ldb(filename())
163 m = {"dn": ldb.Dn(l, "dc=foo5"),
165 self.assertEqual(len(l.search()), 0)
168 self.assertEqual(len(l.search()), 1)
170 l.delete(ldb.Dn(l, "dc=foo5"))
172 def test_add_dict_string_dn(self):
173 l = ldb.Ldb(filename())
174 m = {"dn": "dc=foo6", "bla": "bla"}
175 self.assertEqual(len(l.search()), 0)
178 self.assertEqual(len(l.search()), 1)
180 l.delete(ldb.Dn(l, "dc=foo6"))
182 def test_rename(self):
183 l = ldb.Ldb(filename())
185 m.dn = ldb.Dn(l, "dc=foo7")
187 self.assertEqual(len(l.search()), 0)
190 l.rename(ldb.Dn(l, "dc=foo7"), ldb.Dn(l, "dc=bar"))
191 self.assertEqual(len(l.search()), 1)
193 l.delete(ldb.Dn(l, "dc=bar"))
195 def test_rename_string_dns(self):
196 l = ldb.Ldb(filename())
198 m.dn = ldb.Dn(l, "dc=foo8")
200 self.assertEqual(len(l.search()), 0)
202 self.assertEqual(len(l.search()), 1)
204 l.rename("dc=foo8", "dc=bar")
205 self.assertEqual(len(l.search()), 1)
207 l.delete(ldb.Dn(l, "dc=bar"))
209 def test_modify_delete(self):
210 l = ldb.Ldb(filename())
212 m.dn = ldb.Dn(l, "dc=modifydelete")
215 rm = l.search(m.dn)[0]
216 self.assertEqual(["1234"], list(rm["bla"]))
219 m.dn = ldb.Dn(l, "dc=modifydelete")
220 m["bla"] = ldb.MessageElement([], ldb.FLAG_MOD_DELETE, "bla")
221 self.assertEqual(ldb.FLAG_MOD_DELETE, m["bla"].flags())
223 rm = l.search(m.dn)[0]
224 self.assertEqual(1, len(rm))
225 rm = l.search(m.dn, attrs=["bla"])
226 self.assertEqual(0, len(rm))
228 l.delete(ldb.Dn(l, "dc=modifydelete"))
230 def test_modify_add(self):
231 l = ldb.Ldb(filename())
233 m.dn = ldb.Dn(l, "dc=add")
238 m.dn = ldb.Dn(l, "dc=add")
239 m["bla"] = ldb.MessageElement(["456"], ldb.FLAG_MOD_ADD, "bla")
240 self.assertEqual(ldb.FLAG_MOD_ADD, m["bla"].flags())
242 rm = l.search(m.dn)[0]
243 self.assertEqual(2, len(rm))
244 self.assertEqual(["1234", "456"], list(rm["bla"]))
246 l.delete(ldb.Dn(l, "dc=add"))
248 def test_modify_replace(self):
249 l = ldb.Ldb(filename())
251 m.dn = ldb.Dn(l, "dc=modify2")
252 m["bla"] = ["1234", "456"]
256 m.dn = ldb.Dn(l, "dc=modify2")
257 m["bla"] = ldb.MessageElement(["789"], ldb.FLAG_MOD_REPLACE, "bla")
258 self.assertEqual(ldb.FLAG_MOD_REPLACE, m["bla"].flags())
260 rm = l.search(m.dn)[0]
261 self.assertEqual(2, len(rm))
262 self.assertEqual(["789"], list(rm["bla"]))
263 rm = l.search(m.dn, attrs=["bla"])[0]
264 self.assertEqual(1, len(rm))
266 l.delete(ldb.Dn(l, "dc=modify2"))
268 def test_modify_flags_change(self):
269 l = ldb.Ldb(filename())
271 m.dn = ldb.Dn(l, "dc=add")
276 m.dn = ldb.Dn(l, "dc=add")
277 m["bla"] = ldb.MessageElement(["456"], ldb.FLAG_MOD_ADD, "bla")
278 self.assertEqual(ldb.FLAG_MOD_ADD, m["bla"].flags())
280 rm = l.search(m.dn)[0]
281 self.assertEqual(2, len(rm))
282 self.assertEqual(["1234", "456"], list(rm["bla"]))
284 # Now create another modify, but switch the flags before we do it
285 m["bla"] = ldb.MessageElement(["456"], ldb.FLAG_MOD_ADD, "bla")
286 m["bla"].set_flags(ldb.FLAG_MOD_DELETE)
288 rm = l.search(m.dn, attrs=["bla"])[0]
289 self.assertEqual(1, len(rm))
290 self.assertEqual(["1234"], list(rm["bla"]))
292 l.delete(ldb.Dn(l, "dc=add"))
294 def test_transaction_commit(self):
295 l = ldb.Ldb(filename())
296 l.transaction_start()
297 m = ldb.Message(ldb.Dn(l, "dc=foo9"))
300 l.transaction_commit()
303 def test_transaction_cancel(self):
304 l = ldb.Ldb(filename())
305 l.transaction_start()
306 m = ldb.Message(ldb.Dn(l, "dc=foo10"))
309 l.transaction_cancel()
310 self.assertEqual(0, len(l.search(ldb.Dn(l, "dc=foo10"))))
312 def test_set_debug(self):
313 def my_report_fn(level, text):
315 l = ldb.Ldb(filename())
316 l.set_debug(my_report_fn)
318 def test_zero_byte_string(self):
319 """Testing we do not get trapped in the \0 byte in a property string."""
320 l = ldb.Ldb(filename())
323 "objectclass" : "user",
324 "cN" : "LDAPtestUSER",
325 "givenname" : "ldap",
326 "displayname" : "foo\0bar",
328 res = l.search(expression="(dn=dc=somedn)")
329 self.assertEqual("foo\0bar", res[0]["displayname"][0])
331 def test_no_crash_broken_expr(self):
332 l = ldb.Ldb(filename())
333 self.assertRaises(ldb.LdbError,lambda: l.search("", ldb.SCOPE_SUBTREE, "&(dc=*)(dn=*)", ["dc"]))
336 class DnTests(TestCase):
339 super(DnTests, self).setUp()
340 self.ldb = ldb.Ldb(filename())
342 def test_set_dn_invalid(self):
346 self.assertRaises(TypeError, assign)
349 x = ldb.Dn(self.ldb, "dc=foo11,bar=bloe")
350 y = ldb.Dn(self.ldb, "dc=foo11,bar=bloe")
351 self.assertEqual(x, y)
352 y = ldb.Dn(self.ldb, "dc=foo11,bar=blie")
353 self.assertNotEqual(x, y)
356 x = ldb.Dn(self.ldb, "dc=foo12,bar=bloe")
357 self.assertEqual(x.__str__(), "dc=foo12,bar=bloe")
360 x = ldb.Dn(self.ldb, "dc=foo13,bla=blie")
361 self.assertEqual(x.__repr__(), "Dn('dc=foo13,bla=blie')")
363 def test_get_casefold(self):
364 x = ldb.Dn(self.ldb, "dc=foo14,bar=bloe")
365 self.assertEqual(x.get_casefold(), "DC=FOO14,BAR=bloe")
367 def test_validate(self):
368 x = ldb.Dn(self.ldb, "dc=foo15,bar=bloe")
369 self.assertTrue(x.validate())
371 def test_parent(self):
372 x = ldb.Dn(self.ldb, "dc=foo16,bar=bloe")
373 self.assertEqual("bar=bloe", x.parent().__str__())
375 def test_parent_nonexistent(self):
376 x = ldb.Dn(self.ldb, "@BLA")
377 self.assertEqual(None, x.parent())
379 def test_is_valid(self):
380 x = ldb.Dn(self.ldb, "dc=foo18,dc=bloe")
381 self.assertTrue(x.is_valid())
382 x = ldb.Dn(self.ldb, "")
383 self.assertTrue(x.is_valid())
385 def test_is_special(self):
386 x = ldb.Dn(self.ldb, "dc=foo19,bar=bloe")
387 self.assertFalse(x.is_special())
388 x = ldb.Dn(self.ldb, "@FOOBAR")
389 self.assertTrue(x.is_special())
391 def test_check_special(self):
392 x = ldb.Dn(self.ldb, "dc=foo20,bar=bloe")
393 self.assertFalse(x.check_special("FOOBAR"))
394 x = ldb.Dn(self.ldb, "@FOOBAR")
395 self.assertTrue(x.check_special("@FOOBAR"))
398 x = ldb.Dn(self.ldb, "dc=foo21,bar=bloe")
399 self.assertEqual(2, len(x))
400 x = ldb.Dn(self.ldb, "dc=foo21")
401 self.assertEqual(1, len(x))
403 def test_add_child(self):
404 x = ldb.Dn(self.ldb, "dc=foo22,bar=bloe")
405 self.assertTrue(x.add_child(ldb.Dn(self.ldb, "bla=bloe")))
406 self.assertEqual("bla=bloe,dc=foo22,bar=bloe", x.__str__())
408 def test_add_base(self):
409 x = ldb.Dn(self.ldb, "dc=foo23,bar=bloe")
410 base = ldb.Dn(self.ldb, "bla=bloe")
411 self.assertTrue(x.add_base(base))
412 self.assertEqual("dc=foo23,bar=bloe,bla=bloe", x.__str__())
414 def test_add_child_str(self):
415 x = ldb.Dn(self.ldb, "dc=foo22,bar=bloe")
416 self.assertTrue(x.add_child("bla=bloe"))
417 self.assertEquals("bla=bloe,dc=foo22,bar=bloe", x.__str__())
419 def test_add_base_str(self):
420 x = ldb.Dn(self.ldb, "dc=foo23,bar=bloe")
422 self.assertTrue(x.add_base(base))
423 self.assertEquals("dc=foo23,bar=bloe,bla=bloe", x.__str__())
426 x = ldb.Dn(self.ldb, "dc=foo24")
427 y = ldb.Dn(self.ldb, "bar=bla")
428 self.assertEqual("dc=foo24,bar=bla", str(x + y))
430 def test_remove_base_components(self):
431 x = ldb.Dn(self.ldb, "dc=foo24,dc=samba,dc=org")
432 x.remove_base_components(len(x)-1)
433 self.assertEqual("dc=foo24", str(x))
435 def test_parse_ldif(self):
436 msgs = self.ldb.parse_ldif("dn: foo=bar\n")
438 self.assertEqual("foo=bar", str(msg[1].dn))
439 self.assertTrue(isinstance(msg[1], ldb.Message))
440 ldif = self.ldb.write_ldif(msg[1], ldb.CHANGETYPE_NONE)
441 self.assertEqual("dn: foo=bar\n\n", ldif)
443 def test_parse_ldif_more(self):
444 msgs = self.ldb.parse_ldif("dn: foo=bar\n\n\ndn: bar=bar")
446 self.assertEqual("foo=bar", str(msg[1].dn))
448 self.assertEqual("bar=bar", str(msg[1].dn))
450 def test_canonical_string(self):
451 x = ldb.Dn(self.ldb, "dc=foo25,bar=bloe")
452 self.assertEqual("/bloe/foo25", x.canonical_str())
454 def test_canonical_ex_string(self):
455 x = ldb.Dn(self.ldb, "dc=foo26,bar=bloe")
456 self.assertEqual("/bloe\nfoo26", x.canonical_ex_str())
458 def test_ldb_is_child_of(self):
459 """Testing ldb_dn_compare_dn"""
460 dn1 = ldb.Dn(self.ldb, "dc=base")
461 dn2 = ldb.Dn(self.ldb, "cn=foo,dc=base")
462 dn3 = ldb.Dn(self.ldb, "cn=bar,dc=base")
463 dn4 = ldb.Dn(self.ldb, "cn=baz,cn=bar,dc=base")
465 self.assertTrue(dn2.is_child_of(dn1))
466 self.assertTrue(dn4.is_child_of(dn1))
467 self.assertTrue(dn4.is_child_of(dn3))
468 self.assertFalse(dn3.is_child_of(dn2))
469 self.assertFalse(dn1.is_child_of(dn4))
471 def test_ldb_is_child_of_str(self):
472 """Testing ldb_dn_compare_dn"""
474 dn2_str = "cn=foo,dc=base"
475 dn3_str = "cn=bar,dc=base"
476 dn4_str = "cn=baz,cn=bar,dc=base"
478 dn1 = ldb.Dn(self.ldb, dn1_str)
479 dn2 = ldb.Dn(self.ldb, dn2_str)
480 dn3 = ldb.Dn(self.ldb, dn3_str)
481 dn4 = ldb.Dn(self.ldb, dn4_str)
483 self.assertTrue(dn2.is_child_of(dn1_str))
484 self.assertTrue(dn4.is_child_of(dn1_str))
485 self.assertTrue(dn4.is_child_of(dn3_str))
486 self.assertFalse(dn3.is_child_of(dn2_str))
487 self.assertFalse(dn1.is_child_of(dn4_str))
489 class LdbMsgTests(TestCase):
492 super(LdbMsgTests, self).setUp()
493 self.msg = ldb.Message()
495 def test_init_dn(self):
496 self.msg = ldb.Message(ldb.Dn(ldb.Ldb(), "dc=foo27"))
497 self.assertEqual("dc=foo27", str(self.msg.dn))
499 def test_iter_items(self):
500 self.assertEqual(0, len(self.msg.items()))
501 self.msg.dn = ldb.Dn(ldb.Ldb(filename()), "dc=foo28")
502 self.assertEqual(1, len(self.msg.items()))
505 self.msg.dn = ldb.Dn(ldb.Ldb(filename()), "dc=foo29")
506 self.msg["dc"] = "foo"
507 self.assertIn(repr(self.msg), [
508 "Message({'dn': Dn('dc=foo29'), 'dc': MessageElement(['foo'])})",
509 "Message({'dc': MessageElement(['foo']), 'dn': Dn('dc=foo29')})",
513 self.assertEqual(0, len(self.msg))
515 def test_notpresent(self):
516 self.assertRaises(KeyError, lambda: self.msg["foo"])
522 self.msg.add(ldb.MessageElement(["456"], ldb.FLAG_MOD_ADD, "bla"))
524 def test_elements_empty(self):
525 self.assertEqual([], self.msg.elements())
527 def test_elements(self):
528 el = ldb.MessageElement(["456"], ldb.FLAG_MOD_ADD, "bla")
530 self.assertEqual([el], self.msg.elements())
532 def test_add_value(self):
533 self.assertEqual(0, len(self.msg))
534 self.msg["foo"] = ["foo"]
535 self.assertEqual(1, len(self.msg))
537 def test_add_value_multiple(self):
538 self.assertEqual(0, len(self.msg))
539 self.msg["foo"] = ["foo", "bla"]
540 self.assertEqual(1, len(self.msg))
541 self.assertEqual(["foo", "bla"], list(self.msg["foo"]))
543 def test_set_value(self):
544 self.msg["foo"] = ["fool"]
545 self.assertEqual(["fool"], list(self.msg["foo"]))
546 self.msg["foo"] = ["bar"]
547 self.assertEqual(["bar"], list(self.msg["foo"]))
550 self.msg.dn = ldb.Dn(ldb.Ldb(filename()), "@BASEINFO")
551 self.msg["foo"] = ["bla"]
552 self.msg["bar"] = ["bla"]
553 self.assertEqual(["dn", "foo", "bar"], self.msg.keys())
556 self.msg.dn = ldb.Dn(ldb.Ldb(filename()), "@BASEINFO")
557 self.assertEqual("@BASEINFO", self.msg.dn.__str__())
559 def test_get_dn(self):
560 self.msg.dn = ldb.Dn(ldb.Ldb(filename()), "@BASEINFO")
561 self.assertEqual("@BASEINFO", self.msg.get("dn").__str__())
563 def test_get_invalid(self):
564 self.msg.dn = ldb.Dn(ldb.Ldb(filename()), "@BASEINFO")
565 self.assertRaises(TypeError, self.msg.get, 42)
567 def test_get_other(self):
568 self.msg["foo"] = ["bar"]
569 self.assertEqual("bar", self.msg.get("foo")[0])
570 self.assertEqual("bar", self.msg.get("foo", idx=0))
571 self.assertEqual(None, self.msg.get("foo", idx=1))
572 self.assertEqual("", self.msg.get("foo", default='', idx=1))
574 def test_get_default(self):
575 self.assertEqual(None, self.msg.get("tatayoyo", idx=0))
576 self.assertEqual("anniecordie", self.msg.get("tatayoyo", "anniecordie"))
578 def test_get_unknown(self):
579 self.assertEqual(None, self.msg.get("lalalala"))
581 def test_msg_diff(self):
583 msgs = l.parse_ldif("dn: foo=bar\nfoo: bar\nbaz: do\n\ndn: foo=bar\nfoo: bar\nbaz: dont\n")
586 msgdiff = l.msg_diff(msg1, msg2)
587 self.assertEqual("foo=bar", msgdiff.get("dn").__str__())
588 self.assertRaises(KeyError, lambda: msgdiff["foo"])
589 self.assertEqual(1, len(msgdiff))
591 def test_equal_empty(self):
594 self.assertEqual(msg1, msg2)
596 def test_equal_simplel(self):
597 db = ldb.Ldb(filename())
599 msg1.dn = ldb.Dn(db, "foo=bar")
601 msg2.dn = ldb.Dn(db, "foo=bar")
602 self.assertEqual(msg1, msg2)
605 self.assertEqual(msg1, msg2)
607 self.assertNotEqual(msg1, msg2)
610 def test_from_dict(self):
611 rec = {"dn": "dc=fromdict",
612 "a1": ["a1-val1", "a1-val1"]}
614 # check different types of input Flags
615 for flags in [ldb.FLAG_MOD_ADD, ldb.FLAG_MOD_REPLACE, ldb.FLAG_MOD_DELETE]:
616 m = ldb.Message.from_dict(l, rec, flags)
617 self.assertEqual(rec["a1"], list(m["a1"]))
618 self.assertEqual(flags, m["a1"].flags())
620 self.assertRaises(TypeError, ldb.Message.from_dict, dict(), rec, ldb.FLAG_MOD_REPLACE)
621 self.assertRaises(TypeError, ldb.Message.from_dict, l, list(), ldb.FLAG_MOD_REPLACE)
622 self.assertRaises(ValueError, ldb.Message.from_dict, l, rec, 0)
623 # Message.from_dict expects dictionary with 'dn'
624 err_rec = {"a1": ["a1-val1", "a1-val1"]}
625 self.assertRaises(TypeError, ldb.Message.from_dict, l, err_rec, ldb.FLAG_MOD_REPLACE)
627 def test_copy_add_message_element(self):
629 m["1"] = ldb.MessageElement(["val 111"], ldb.FLAG_MOD_ADD, "1")
630 m["2"] = ldb.MessageElement(["val 222"], ldb.FLAG_MOD_ADD, "2")
634 self.assertEqual(mto["1"], m["1"])
635 self.assertEqual(mto["2"], m["2"])
639 self.assertEqual(mto["1"], m["1"])
640 self.assertEqual(mto["2"], m["2"])
643 class MessageElementTests(TestCase):
645 def test_cmp_element(self):
646 x = ldb.MessageElement(["foo"])
647 y = ldb.MessageElement(["foo"])
648 z = ldb.MessageElement(["bzr"])
649 self.assertEqual(x, y)
650 self.assertNotEqual(x, z)
652 def test_create_iterable(self):
653 x = ldb.MessageElement(["foo"])
654 self.assertEqual(["foo"], list(x))
657 x = ldb.MessageElement(["foo"])
658 self.assertEqual("MessageElement(['foo'])", repr(x))
659 x = ldb.MessageElement(["foo", "bla"])
660 self.assertEqual(2, len(x))
661 self.assertEqual("MessageElement(['foo','bla'])", repr(x))
663 def test_get_item(self):
664 x = ldb.MessageElement(["foo", "bar"])
665 self.assertEqual("foo", x[0])
666 self.assertEqual("bar", x[1])
667 self.assertEqual("bar", x[-1])
668 self.assertRaises(IndexError, lambda: x[45])
671 x = ldb.MessageElement(["foo", "bar"])
672 self.assertEqual(2, len(x))
675 x = ldb.MessageElement(["foo", "bar"])
676 y = ldb.MessageElement(["foo", "bar"])
677 self.assertEqual(y, x)
678 x = ldb.MessageElement(["foo"])
679 self.assertNotEqual(y, x)
680 y = ldb.MessageElement(["foo"])
681 self.assertEqual(y, x)
683 def test_extended(self):
684 el = ldb.MessageElement(["456"], ldb.FLAG_MOD_ADD, "bla")
685 self.assertEqual("MessageElement(['456'])", repr(el))
688 class ModuleTests(TestCase):
690 def test_register_module(self):
693 ldb.register_module(ExampleModule)
695 def test_use_module(self):
700 def __init__(self, ldb, next):
704 def search(self, *args, **kwargs):
705 return self.next.search(*args, **kwargs)
707 def request(self, *args, **kwargs):
711 ldb.register_module(ExampleModule)
712 if os.path.exists(name):
715 l.add({"dn": "@MODULES", "@LIST": "bla"})
716 self.assertEqual([], ops)
718 self.assertEqual(["init"], ops)
720 class LdbResultTests(TestCase):
723 super(LdbResultTests, self).setUp()
726 if os.path.exists(name):
728 self.l = ldb.Ldb(name)
729 self.l.add({"dn": "DC=SAMBA,DC=ORG", "name": "samba.org"})
730 self.l.add({"dn": "OU=ADMIN,DC=SAMBA,DC=ORG", "name": "Admins"})
731 self.l.add({"dn": "OU=USERS,DC=SAMBA,DC=ORG", "name": "Users"})
732 self.l.add({"dn": "OU=OU1,DC=SAMBA,DC=ORG", "name": "OU #1"})
733 self.l.add({"dn": "OU=OU2,DC=SAMBA,DC=ORG", "name": "OU #2"})
734 self.l.add({"dn": "OU=OU3,DC=SAMBA,DC=ORG", "name": "OU #3"})
735 self.l.add({"dn": "OU=OU4,DC=SAMBA,DC=ORG", "name": "OU #4"})
736 self.l.add({"dn": "OU=OU5,DC=SAMBA,DC=ORG", "name": "OU #5"})
737 self.l.add({"dn": "OU=OU6,DC=SAMBA,DC=ORG", "name": "OU #6"})
738 self.l.add({"dn": "OU=OU7,DC=SAMBA,DC=ORG", "name": "OU #7"})
739 self.l.add({"dn": "OU=OU8,DC=SAMBA,DC=ORG", "name": "OU #8"})
740 self.l.add({"dn": "OU=OU9,DC=SAMBA,DC=ORG", "name": "OU #9"})
741 self.l.add({"dn": "OU=OU10,DC=SAMBA,DC=ORG", "name": "OU #10"})
744 super(LdbResultTests, self).tearDown()
745 if os.path.exists(self.name):
748 def test_return_type(self):
749 res = self.l.search()
750 self.assertEqual(str(res), "<ldb result>")
752 def test_get_msgs(self):
753 res = self.l.search()
756 def test_get_controls(self):
757 res = self.l.search()
760 def test_get_referals(self):
761 res = self.l.search()
764 def test_iter_msgs(self):
766 for l in self.l.search().msgs:
767 if str(l.dn) == "OU=OU10,DC=SAMBA,DC=ORG":
769 self.assertTrue(found)
771 def test_iter_msgs_count(self):
772 self.assertTrue(self.l.search().count > 0)
773 # 13 objects has been added to the DC=SAMBA, DC=ORG
774 self.assertEqual(self.l.search(base="DC=SAMBA,DC=ORG").count, 13)
776 def test_iter_controls(self):
777 res = self.l.search().controls
780 def test_create_control(self):
781 self.assertRaises(ValueError, ldb.Control, self.l, "tatayoyo:0")
782 c = ldb.Control(self.l, "relax:1")
783 self.assertEqual(c.critical, True)
784 self.assertEqual(c.oid, "1.3.6.1.4.1.4203.666.5.12")
786 def test_iter_refs(self):
787 res = self.l.search().referals
790 def test_iter_as_sequence_msgs(self):
792 res = self.l.search().msgs
794 for i in range(0, len(res)):
796 if str(l.dn) == "OU=OU10,DC=SAMBA,DC=ORG":
798 self.assertTrue(found)
800 def test_iter_as_sequence(self):
802 res = self.l.search()
804 for i in range(0, len(res)):
806 if str(l.dn) == "OU=OU10,DC=SAMBA,DC=ORG":
808 self.assertTrue(found)
811 class BadTypeTests(TestCase):
812 def test_control(self):
814 self.assertRaises(TypeError, ldb.Control, '<bad type>', 'relax:1')
815 self.assertRaises(TypeError, ldb.Control, ldb, 1234)
817 def test_modify(self):
819 dn = ldb.Dn(l, 'a=b')
821 self.assertRaises(TypeError, l.modify, '<bad type>')
822 self.assertRaises(TypeError, l.modify, m, '<bad type>')
826 dn = ldb.Dn(l, 'a=b')
828 self.assertRaises(TypeError, l.add, '<bad type>')
829 self.assertRaises(TypeError, l.add, m, '<bad type>')
831 def test_delete(self):
833 dn = ldb.Dn(l, 'a=b')
834 self.assertRaises(TypeError, l.add, '<bad type>')
835 self.assertRaises(TypeError, l.add, dn, '<bad type>')
837 def test_rename(self):
839 dn = ldb.Dn(l, 'a=b')
840 self.assertRaises(TypeError, l.add, '<bad type>', dn)
841 self.assertRaises(TypeError, l.add, dn, '<bad type>')
842 self.assertRaises(TypeError, l.add, dn, dn, '<bad type>')
844 def test_search(self):
846 self.assertRaises(TypeError, l.search, base=1234)
847 self.assertRaises(TypeError, l.search, scope='<bad type>')
848 self.assertRaises(TypeError, l.search, expression=1234)
849 self.assertRaises(TypeError, l.search, attrs='<bad type>')
850 self.assertRaises(TypeError, l.search, controls='<bad type>')
853 class VersionTests(TestCase):
855 def test_version(self):
856 self.assertTrue(isinstance(ldb.__version__, str))
859 if __name__ == '__main__':
861 unittest.TestProgram()