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/>.
21 from django.test import TestCase
25 from examples.models import LdapUser, LdapGroup
27 class BaseTestCase(TestCase):
29 cursor = ldapdb.connection._cursor()
30 for base in [LdapGroup.base_dn, LdapUser.base_dn]:
31 rdn = base.split(',')[0]
32 key, val = rdn.split('=')
33 attrs = [('objectClass', ['top', 'organizationalUnit']), (key, [val])]
35 cursor.connection.add_s(base, attrs)
36 except ldap.ALREADY_EXISTS:
40 cursor = ldapdb.connection._cursor()
41 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)
46 except ldap.NO_SUCH_OBJECT:
49 class GroupTestCase(BaseTestCase):
51 super(GroupTestCase, self).setUp()
56 g.usernames = ['foouser', 'baruser']
62 g.usernames = ['zoouser', 'baruser']
68 g.usernames = ['wizuser', 'baruser']
73 qs = LdapGroup.objects.none()
74 self.assertEquals(qs.count(), 0)
76 qs = LdapGroup.objects.none()
77 self.assertEquals(len(qs), 0)
80 qs = LdapGroup.objects.all()
81 self.assertEquals(qs.count(), 3)
83 qs = LdapGroup.objects.all()
84 self.assertEquals(len(qs), 3)
86 def test_filter(self):
87 qs = LdapGroup.objects.filter(name='foogroup')
88 self.assertEquals(qs.count(), 1)
90 qs = LdapGroup.objects.filter(name='foogroup')
91 self.assertEquals(len(qs), 1)
94 self.assertEquals(g.dn, 'cn=foogroup,%s' % LdapGroup.base_dn)
95 self.assertEquals(g.name, 'foogroup')
96 self.assertEquals(g.gid, 1000)
97 self.assertEquals(g.usernames, ['foouser', 'baruser'])
99 # try to filter non-existent entries
100 qs = LdapGroup.objects.filter(name='does_not_exist')
101 self.assertEquals(qs.count(), 0)
103 qs = LdapGroup.objects.filter(name='does_not_exist')
104 self.assertEquals(len(qs), 0)
107 g = LdapGroup.objects.get(name='foogroup')
108 self.assertEquals(g.dn, 'cn=foogroup,%s' % LdapGroup.base_dn)
109 self.assertEquals(g.name, 'foogroup')
110 self.assertEquals(g.gid, 1000)
111 self.assertEquals(g.usernames, ['foouser', 'baruser'])
113 # try to get a non-existent entry
114 self.assertRaises(LdapGroup.DoesNotExist, LdapGroup.objects.get, name='does_not_exist')
116 def test_order_by(self):
118 qs = LdapGroup.objects.order_by('name')
119 self.assertEquals(len(qs), 3)
120 self.assertEquals(qs[0].name, 'bargroup')
121 self.assertEquals(qs[1].name, 'foogroup')
122 self.assertEquals(qs[2].name, 'wizgroup')
125 qs = LdapGroup.objects.order_by('-name')
126 self.assertEquals(len(qs), 3)
127 self.assertEquals(qs[0].name, 'wizgroup')
128 self.assertEquals(qs[1].name, 'foogroup')
129 self.assertEquals(qs[2].name, 'bargroup')
132 qs = LdapGroup.objects.order_by('gid')
133 self.assertEquals(len(qs), 3)
134 self.assertEquals(qs[0].gid, 1000)
135 self.assertEquals(qs[1].gid, 1001)
136 self.assertEquals(qs[2].gid, 1002)
139 qs = LdapGroup.objects.order_by('-gid')
140 self.assertEquals(len(qs), 3)
141 self.assertEquals(qs[0].gid, 1002)
142 self.assertEquals(qs[1].gid, 1001)
143 self.assertEquals(qs[2].gid, 1000)
145 def test_bulk_delete(self):
146 LdapGroup.objects.all().delete()
148 qs = LdapGroup.objects.all()
149 self.assertEquals(len(qs), 0)
151 def test_slice(self):
152 qs = LdapGroup.objects.all()
154 self.assertEquals(len(objs), 3)
155 self.assertEquals(objs[0].gid, 1000)
156 self.assertEquals(objs[1].gid, 1001)
157 self.assertEquals(objs[2].gid, 1002)
160 qs = LdapGroup.objects.all()
162 self.assertEquals(objs.count(), 2)
165 self.assertEquals(len(objs), 2)
166 self.assertEquals(objs[0].gid, 1000)
167 self.assertEquals(objs[1].gid, 1001)
170 qs = LdapGroup.objects.all()
172 self.assertEquals(objs.count(), 2)
175 self.assertEquals(len(objs), 2)
176 self.assertEquals(objs[0].gid, 1001)
177 self.assertEquals(objs[1].gid, 1002)
180 qs = LdapGroup.objects.all()
182 self.assertEquals(objs.count(), 1)
185 self.assertEquals(len(objs), 1)
186 self.assertEquals(objs[0].gid, 1001)
188 def test_update(self):
189 g = LdapGroup.objects.get(name='foogroup')
192 g.usernames = ['foouser2', 'baruser2']
195 # make sure DN gets updated if we change the pk
198 self.assertEquals(g.dn, 'cn=foogroup2,%s' % LdapGroup.base_dn)
200 def test_values(self):
201 qs = LdapGroup.objects.values('name')
202 self.assertEquals(len(qs), 3)
203 self.assertEquals(qs[0], {'name': 'foogroup'})
204 self.assertEquals(qs[1], {'name': 'bargroup'})
205 self.assertEquals(qs[2], {'name': 'wizgroup'})
207 def test_values_list(self):
208 qs = LdapGroup.objects.values_list('name')
209 self.assertEquals(len(qs), 3)
210 self.assertEquals(qs[0], ('foogroup',))
211 self.assertEquals(qs[1], ('bargroup',))
212 self.assertEquals(qs[2], ('wizgroup',))
214 def test_delete(self):
215 g = LdapGroup.objects.get(name='foogroup')
218 qs = LdapGroup.objects.all()
219 self.assertEquals(len(qs), 2)
221 class UserTestCase(BaseTestCase):
223 super(UserTestCase, self).setUp()
226 u.first_name = u"Fôo"
227 u.last_name = u"Usér"
228 u.full_name = u"Fôo Usér"
231 u.home_directory = "/home/foouser"
233 u.username = "foouser"
234 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'
238 u = LdapUser.objects.get(username='foouser')
239 self.assertEquals(u.first_name, u'Fôo')
240 self.assertEquals(u.last_name, u'Usér')
241 self.assertEquals(u.full_name, u'Fôo Usér')
243 self.assertEquals(u.group, 1000)
244 self.assertEquals(u.home_directory, '/home/foouser')
245 self.assertEquals(u.uid, 2000)
246 self.assertEquals(u.username, 'foouser')
247 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')
249 self.assertRaises(LdapUser.DoesNotExist, LdapUser.objects.get, username='does_not_exist')
251 def test_update(self):
252 u = LdapUser.objects.get(username='foouser')
253 u.first_name = u'Fôo2'
256 # make sure DN gets updated if we change the pk
257 u.username = 'foouser2'
259 self.assertEquals(u.dn, 'uid=foouser2,%s' % LdapUser.base_dn)
261 class AdminTestCase(BaseTestCase):
262 fixtures = ['test_users.json']
265 super(AdminTestCase, self).setUp()
270 g.usernames = ['foouser', 'baruser']
276 g.usernames = ['zoouser', 'baruser']
282 u.full_name = "Foo User"
284 u.home_directory = "/home/foouser"
286 u.username = "foouser"
292 u.full_name = "Bar User"
294 u.home_directory = "/home/baruser"
296 u.username = "baruser"
299 self.client.login(username="test_user", password="password")
301 def test_index(self):
302 response = self.client.get('/admin/examples/')
303 self.assertContains(response, "Ldap groups")
304 self.assertContains(response, "Ldap users")
306 def test_group_list(self):
307 response = self.client.get('/admin/examples/ldapgroup/')
308 self.assertContains(response, "Ldap groups")
309 self.assertContains(response, "foogroup")
310 self.assertContains(response, "1000")
313 response = self.client.get('/admin/examples/ldapgroup/?o=1')
314 self.assertContains(response, "Ldap groups")
315 self.assertContains(response, "foogroup")
316 self.assertContains(response, "1000")
319 response = self.client.get('/admin/examples/ldapgroup/?o=2')
320 self.assertContains(response, "Ldap groups")
321 self.assertContains(response, "foogroup")
322 self.assertContains(response, "1000")
324 def test_group_detail(self):
325 response = self.client.get('/admin/examples/ldapgroup/foogroup/')
326 self.assertContains(response, "foogroup")
327 self.assertContains(response, "1000")
329 def test_group_delete(self):
330 response = self.client.post('/admin/examples/ldapgroup/foogroup/delete/', {'yes': 'post'})
331 self.assertRedirects(response, '/admin/examples/ldapgroup/')
333 def test_group_search(self):
334 response = self.client.get('/admin/examples/ldapgroup/?q=foo')
335 self.assertContains(response, "Ldap groups")
336 self.assertContains(response, "foogroup")
337 self.assertContains(response, "1000")
339 def test_user_list(self):
340 response = self.client.get('/admin/examples/ldapuser/')
341 self.assertContains(response, "Ldap users")
342 self.assertContains(response, "foouser")
343 self.assertContains(response, "2000")
346 response = self.client.get('/admin/examples/ldapuser/?o=1')
347 self.assertContains(response, "Ldap users")
348 self.assertContains(response, "foouser")
349 self.assertContains(response, "2000")
352 response = self.client.get('/admin/examples/ldapuser/?o=2')
353 self.assertContains(response, "Ldap users")
354 self.assertContains(response, "foouser")
355 self.assertContains(response, "2000")
357 def test_user_detail(self):
358 response = self.client.get('/admin/examples/ldapuser/foouser/')
359 self.assertContains(response, "foouser")
360 self.assertContains(response, "2000")
362 def test_user_delete(self):
363 response = self.client.post('/admin/examples/ldapuser/foouser/delete/', {'yes': 'post'})
364 self.assertRedirects(response, '/admin/examples/ldapuser/')