Add some unit tests
This commit is contained in:
parent
5fac716fb5
commit
c876b63acf
|
@ -0,0 +1,95 @@
|
|||
import ldap, unittest
|
||||
import slapd
|
||||
|
||||
from ldap.ldapobject import LDAPObject
|
||||
|
||||
server = None
|
||||
|
||||
class TestSearch(unittest.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
global server
|
||||
if server is None:
|
||||
server = slapd.Slapd()
|
||||
server.start()
|
||||
base = server.get_dn_suffix()
|
||||
|
||||
# insert some Foo* objects via ldapadd
|
||||
server.ldapadd("\n".join([
|
||||
"dn: cn=Foo1,"+base,
|
||||
"objectClass: organizationalRole",
|
||||
"cn: Foo1",
|
||||
"",
|
||||
"dn: cn=Foo2,"+base,
|
||||
"objectClass: organizationalRole",
|
||||
"cn: Foo2",
|
||||
"",
|
||||
"dn: cn=Foo3,"+base,
|
||||
"objectClass: organizationalRole",
|
||||
"cn: Foo3",
|
||||
"",
|
||||
"dn: ou=Container,"+base,
|
||||
"objectClass: organizationalUnit",
|
||||
"ou: Container",
|
||||
"",
|
||||
"dn: cn=Foo4,ou=Container,"+base,
|
||||
"objectClass: organizationalRole",
|
||||
"cn: Foo4",
|
||||
"",
|
||||
])+"\n")
|
||||
|
||||
l = LDAPObject(server.get_url())
|
||||
l.protocol_version = 3
|
||||
l.set_option(ldap.OPT_REFERRALS,0)
|
||||
l.simple_bind_s(server.get_root_dn(),
|
||||
server.get_root_password())
|
||||
self.ldap = l
|
||||
self.server = server
|
||||
|
||||
def test_search_subtree(self):
|
||||
base = self.server.get_dn_suffix()
|
||||
l = self.ldap
|
||||
|
||||
result = l.search_s(base, ldap.SCOPE_SUBTREE, '(cn=Foo*)', ['*'])
|
||||
result.sort()
|
||||
self.assertEquals(result,
|
||||
[('cn=Foo1,'+base,
|
||||
{'cn': ['Foo1'], 'objectClass': ['organizationalRole']}),
|
||||
('cn=Foo2,'+base,
|
||||
{'cn': ['Foo2'], 'objectClass': ['organizationalRole']}),
|
||||
('cn=Foo3,'+base,
|
||||
{'cn': ['Foo3'], 'objectClass': ['organizationalRole']}),
|
||||
('cn=Foo4,ou=Container,'+base,
|
||||
{'cn': ['Foo4'], 'objectClass': ['organizationalRole']}),
|
||||
]
|
||||
)
|
||||
|
||||
def test_search_onelevel(self):
|
||||
base = self.server.get_dn_suffix()
|
||||
l = self.ldap
|
||||
|
||||
result = l.search_s(base, ldap.SCOPE_ONELEVEL, '(cn=Foo*)', ['*'])
|
||||
result.sort()
|
||||
self.assertEquals(result,
|
||||
[('cn=Foo1,'+base,
|
||||
{'cn': ['Foo1'], 'objectClass': ['organizationalRole']}),
|
||||
('cn=Foo2,'+base,
|
||||
{'cn': ['Foo2'], 'objectClass': ['organizationalRole']}),
|
||||
('cn=Foo3,'+base,
|
||||
{'cn': ['Foo3'], 'objectClass': ['organizationalRole']}),
|
||||
]
|
||||
)
|
||||
|
||||
def test_search_oneattr(self):
|
||||
base = self.server.get_dn_suffix()
|
||||
l = self.ldap
|
||||
|
||||
result = l.search_s(base, ldap.SCOPE_SUBTREE, '(cn=Foo4)', ['cn'])
|
||||
result.sort()
|
||||
self.assertEquals(result,
|
||||
[('cn=Foo4,ou=Container,'+base, {'cn': ['Foo4']})]
|
||||
)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
Loading…
Reference in New Issue