X-Git-Url: https://git.stderr.nl/gitweb?p=matthijs%2Fupstream%2Fdjango-ldapdb.git;a=blobdiff_plain;f=examples%2Ftests.py;h=bca180294e5a8f038a176561aaac6886df88e2d5;hp=dba0f5ee086f9f704b7d140c35d3e3163bfb8787;hb=5940d76f9249693c27b6665f6687a172d374dcbb;hpb=eac1a9e03d8e7dc89abf60d1e026c67164edbcb0 diff --git a/examples/tests.py b/examples/tests.py index dba0f5e..bca1802 100644 --- a/examples/tests.py +++ b/examples/tests.py @@ -18,27 +18,173 @@ # along with this program. If not, see . # +import ldap + from django.test import TestCase +from ldapdb import connection from examples.models import LdapUser, LdapGroup -class GroupTestCase(TestCase): - def test_create(self): +class BaseTestCase(TestCase): + def setUp(self): + cursor = connection._cursor() + for base in [LdapGroup.base_dn, LdapUser.base_dn]: + ou = base.split(',')[0].split('=')[1] + attrs = [('objectClass', ['top', 'organizationalUnit']), ('ou', [ou])] + try: + cursor.connection.add_s(base, attrs) + except ldap.ALREADY_EXISTS: + pass + + def tearDown(self): + cursor = connection._cursor() + for base in [LdapGroup.base_dn, LdapUser.base_dn]: + results = cursor.connection.search_s(base, ldap.SCOPE_SUBTREE) + for dn, attrs in reversed(results): + cursor.connection.delete_s(dn) + +class GroupTestCase(BaseTestCase): + def setUp(self): + super(GroupTestCase, self).setUp() + + g = LdapGroup() + g.name = "foogroup" + g.gid = 1000 + g.usernames = ['foouser', 'baruser'] + g.save() + + g = LdapGroup() + g.name = "bargroup" + g.gid = 1001 + g.usernames = ['zoouser', 'baruser'] + g.save() + + def test_filter(self): + qs = LdapGroup.objects.none() + self.assertEquals(len(qs), 0) + + qs = LdapGroup.objects.all() + self.assertEquals(len(qs), 2) + + qs = LdapGroup.objects.filter(name='foogroup') + self.assertEquals(len(qs), 1) + + qs = LdapGroup.objects.filter(name='does_not_exist') + self.assertEquals(len(qs), 0) + + def test_get(self): + g = LdapGroup.objects.get(name='foogroup') + self.assertEquals(g.dn, 'cn=foogroup,ou=groups,dc=nodomain') + self.assertEquals(g.name, 'foogroup') + self.assertEquals(g.gid, 1000) + self.assertEquals(g.usernames, ['foouser', 'baruser']) + + self.assertRaises(LdapGroup.DoesNotExist, LdapGroup.objects.get, name='does_not_exist') + + def test_update(self): + g = LdapGroup.objects.get(name='foogroup') + + g.gid = 1002 + g.usernames = ['foouser2', 'baruser2'] + g.save() + + # make sure DN gets updated if we change the pk + g.name = 'foogroup2' + g.save() + self.assertEquals(g.dn, 'cn=foogroup2,ou=groups,dc=nodomain') + + def test_delete(self): + g = LdapGroup.objects.get(name='foogroup') + g.delete() + + qs = LdapGroup.objects.all() + self.assertEquals(len(qs), 1) + +class UserTestCase(BaseTestCase): + def setUp(self): + super(UserTestCase, self).setUp() + + u = LdapUser() + u.first_name = u"Fôo" + u.last_name = u"Usér" + u.full_name = u"Fôo Usér" + + u.group = 1000 + u.home_directory = "/home/foouser" + u.uid = 1000 + u.username = "foouser" + u.photo = '\xff\xd8\xff\xe0\x00\x10JFIF\x00\x01\x01\x01\x00H\x00H\x00\x00\xff\xfe\x00\x1cCreated with GIMP on a Mac\xff\xdb\x00C\x00\x05\x03\x04\x04\x04\x03\x05\x04\x04\x04\x05\x05\x05\x06\x07\x0c\x08\x07\x07\x07\x07\x0f\x0b\x0b\t\x0c\x11\x0f\x12\x12\x11\x0f\x11\x11\x13\x16\x1c\x17\x13\x14\x1a\x15\x11\x11\x18!\x18\x1a\x1d\x1d\x1f\x1f\x1f\x13\x17"$"\x1e$\x1c\x1e\x1f\x1e\xff\xdb\x00C\x01\x05\x05\x05\x07\x06\x07\x0e\x08\x08\x0e\x1e\x14\x11\x14\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\xff\xc0\x00\x11\x08\x00\x08\x00\x08\x03\x01"\x00\x02\x11\x01\x03\x11\x01\xff\xc4\x00\x15\x00\x01\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x08\xff\xc4\x00\x19\x10\x00\x03\x01\x01\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01\x02\x06\x11A\xff\xc4\x00\x14\x01\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xff\xc4\x00\x14\x11\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xff\xda\x00\x0c\x03\x01\x00\x02\x11\x03\x11\x00?\x00\x9d\xf29wU5Q\xd6\xfd\x00\x01\xff\xd9' + u.save() + + def test_get(self): + u = LdapUser.objects.get(username='foouser') + self.assertEquals(u.first_name, u'Fôo') + self.assertEquals(u.last_name, u'Usér') + self.assertEquals(u.full_name, u'Fôo Usér') + + self.assertEquals(u.group, 1000) + self.assertEquals(u.home_directory, '/home/foouser') + self.assertEquals(u.uid, 1000) + self.assertEquals(u.username, 'foouser') + self.assertEquals(u.photo, '\xff\xd8\xff\xe0\x00\x10JFIF\x00\x01\x01\x01\x00H\x00H\x00\x00\xff\xfe\x00\x1cCreated with GIMP on a Mac\xff\xdb\x00C\x00\x05\x03\x04\x04\x04\x03\x05\x04\x04\x04\x05\x05\x05\x06\x07\x0c\x08\x07\x07\x07\x07\x0f\x0b\x0b\t\x0c\x11\x0f\x12\x12\x11\x0f\x11\x11\x13\x16\x1c\x17\x13\x14\x1a\x15\x11\x11\x18!\x18\x1a\x1d\x1d\x1f\x1f\x1f\x13\x17"$"\x1e$\x1c\x1e\x1f\x1e\xff\xdb\x00C\x01\x05\x05\x05\x07\x06\x07\x0e\x08\x08\x0e\x1e\x14\x11\x14\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\x1e\xff\xc0\x00\x11\x08\x00\x08\x00\x08\x03\x01"\x00\x02\x11\x01\x03\x11\x01\xff\xc4\x00\x15\x00\x01\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x08\xff\xc4\x00\x19\x10\x00\x03\x01\x01\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01\x02\x06\x11A\xff\xc4\x00\x14\x01\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xff\xc4\x00\x14\x11\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xff\xda\x00\x0c\x03\x01\x00\x02\x11\x03\x11\x00?\x00\x9d\xf29wU5Q\xd6\xfd\x00\x01\xff\xd9') + + self.assertRaises(LdapUser.DoesNotExist, LdapUser.objects.get, username='does_not_exist') + + def test_update(self): + u = LdapUser.objects.get(username='foouser') + u.first_name = u'Fôo2' + u.save() + + # make sure DN gets updated if we change the pk + u.username = 'foouser2' + u.save() + self.assertEquals(u.dn, 'uid=foouser2,ou=people,dc=nodomain') + +class AdminTestCase(BaseTestCase): + fixtures = ['test_users.json'] + + def setUp(self): + super(AdminTestCase, self).setUp() + g = LdapGroup() g.name = "foogroup" g.gid = 1000 + g.usernames = ['foouser', 'baruser'] g.save() - -class UserTestCase(TestCase): - def test_create(self): + u = LdapUser() u.first_name = "Foo" u.last_name = "User" u.full_name = "Foo User" - u.group = 1000 u.home_directory = "/home/foouser" u.uid = 1000 u.username = "foouser" u.save() - + + self.client.login(username="test_user", password="password") + + def test_index(self): + response = self.client.get('/admin/examples/') + self.assertContains(response, "Ldap groups") + self.assertContains(response, "Ldap users") + + def test_group_list(self): + response = self.client.get('/admin/examples/ldapgroup/') + self.assertContains(response, "Ldap groups") + self.assertContains(response, "foogroup") + + def test_group_detail(self): + response = self.client.get('/admin/examples/ldapgroup/foogroup/') + self.assertContains(response, "foogroup") + self.assertContains(response, "1000") + + def test_user_list(self): + response = self.client.get('/admin/examples/ldapuser/') + self.assertContains(response, "Ldap users") + self.assertContains(response, "foouser") + + def test_user_detail(self): + response = self.client.get('/admin/examples/ldapuser/foouser/') + self.assertContains(response, "foouser") + self.assertContains(response, "1000")