1 # -*- coding: utf-8 -*-
4 # Copyright (C) 2009-2010 Bolloré telecom
5 # See AUTHORS file for a full list of contributors.
7 # This program is free software: you can redistribute it and/or modify
8 # it under the terms of the GNU General Public License as published by
9 # the Free Software Foundation, either version 3 of the License, or
10 # (at your option) any later version.
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
15 # GNU General Public License for more details.
17 # You should have received a copy of the GNU General Public License
18 # along with this program. If not, see <http://www.gnu.org/licenses/>.
23 from django.test import TestCase
25 from ldapdb import connection
26 from examples.models import LdapUser, LdapGroup
28 class BaseTestCase(TestCase):
30 cursor = connection._cursor()
31 for base in [LdapGroup.base_dn, LdapUser.base_dn]:
32 rdn = base.split(',')[0]
33 key, val = rdn.split('=')
34 attrs = [('objectClass', ['top', 'organizationalUnit']), (key, [val])]
36 cursor.connection.add_s(base, attrs)
37 except ldap.ALREADY_EXISTS:
41 cursor = connection._cursor()
42 for base in [LdapGroup.base_dn, LdapUser.base_dn]:
43 results = cursor.connection.search_s(base, ldap.SCOPE_SUBTREE)
44 for dn, attrs in reversed(results):
45 cursor.connection.delete_s(dn)
47 class GroupTestCase(BaseTestCase):
49 super(GroupTestCase, self).setUp()
54 g.usernames = ['foouser', 'baruser']
60 g.usernames = ['zoouser', 'baruser']
63 def test_filter(self):
64 qs = LdapGroup.objects.none()
65 self.assertEquals(len(qs), 0)
67 qs = LdapGroup.objects.all()
68 self.assertEquals(len(qs), 2)
70 qs = LdapGroup.objects.filter(name='foogroup')
71 self.assertEquals(len(qs), 1)
74 self.assertEquals(g.dn, 'cn=foogroup,ou=groups,dc=nodomain')
75 self.assertEquals(g.name, 'foogroup')
76 self.assertEquals(g.gid, 1000)
77 self.assertEquals(g.usernames, ['foouser', 'baruser'])
79 qs = LdapGroup.objects.filter(name='does_not_exist')
80 self.assertEquals(len(qs), 0)
83 g = LdapGroup.objects.get(name='foogroup')
84 self.assertEquals(g.dn, 'cn=foogroup,ou=groups,dc=nodomain')
85 self.assertEquals(g.name, 'foogroup')
86 self.assertEquals(g.gid, 1000)
87 self.assertEquals(g.usernames, ['foouser', 'baruser'])
89 self.assertRaises(LdapGroup.DoesNotExist, LdapGroup.objects.get, name='does_not_exist')
91 def test_update(self):
92 g = LdapGroup.objects.get(name='foogroup')
95 g.usernames = ['foouser2', 'baruser2']
98 # make sure DN gets updated if we change the pk
101 self.assertEquals(g.dn, 'cn=foogroup2,ou=groups,dc=nodomain')
103 def test_delete(self):
104 g = LdapGroup.objects.get(name='foogroup')
107 qs = LdapGroup.objects.all()
108 self.assertEquals(len(qs), 1)
110 class UserTestCase(BaseTestCase):
112 super(UserTestCase, self).setUp()
115 u.first_name = u"Fôo"
116 u.last_name = u"Usér"
117 u.full_name = u"Fôo Usér"
120 u.home_directory = "/home/foouser"
122 u.username = "foouser"
123 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'
127 u = LdapUser.objects.get(username='foouser')
128 self.assertEquals(u.first_name, u'Fôo')
129 self.assertEquals(u.last_name, u'Usér')
130 self.assertEquals(u.full_name, u'Fôo Usér')
132 self.assertEquals(u.group, 1000)
133 self.assertEquals(u.home_directory, '/home/foouser')
134 self.assertEquals(u.uid, 1000)
135 self.assertEquals(u.username, 'foouser')
136 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')
138 self.assertRaises(LdapUser.DoesNotExist, LdapUser.objects.get, username='does_not_exist')
140 def test_update(self):
141 u = LdapUser.objects.get(username='foouser')
142 u.first_name = u'Fôo2'
145 # make sure DN gets updated if we change the pk
146 u.username = 'foouser2'
148 self.assertEquals(u.dn, 'uid=foouser2,ou=people,dc=nodomain')
150 class AdminTestCase(BaseTestCase):
151 fixtures = ['test_users.json']
154 super(AdminTestCase, self).setUp()
159 g.usernames = ['foouser', 'baruser']
165 u.full_name = "Foo User"
167 u.home_directory = "/home/foouser"
169 u.username = "foouser"
172 self.client.login(username="test_user", password="password")
174 def test_index(self):
175 response = self.client.get('/admin/examples/')
176 self.assertContains(response, "Ldap groups")
177 self.assertContains(response, "Ldap users")
179 def test_group_list(self):
180 response = self.client.get('/admin/examples/ldapgroup/')
181 self.assertContains(response, "Ldap groups")
182 self.assertContains(response, "foogroup")
184 def test_group_detail(self):
185 response = self.client.get('/admin/examples/ldapgroup/foogroup/')
186 self.assertContains(response, "foogroup")
187 self.assertContains(response, "1000")
189 def test_user_list(self):
190 response = self.client.get('/admin/examples/ldapuser/')
191 self.assertContains(response, "Ldap users")
192 self.assertContains(response, "foouser")
194 def test_user_detail(self):
195 response = self.client.get('/admin/examples/ldapuser/foouser/')
196 self.assertContains(response, "foouser")
197 self.assertContains(response, "1000")