1 # -*- coding: utf-8 -*-
4 # Copyright (c) 2009-2010, Bolloré telecom
7 # See AUTHORS file for a full list of contributors.
9 # Redistribution and use in source and binary forms, with or without modification,
10 # are permitted provided that the following conditions are met:
12 # 1. Redistributions of source code must retain the above copyright notice,
13 # this list of conditions and the following disclaimer.
15 # 2. Redistributions in binary form must reproduce the above copyright
16 # notice, this list of conditions and the following disclaimer in the
17 # documentation and/or other materials provided with the distribution.
19 # 3. Neither the name of Bolloré telecom nor the names of its contributors
20 # may be used to endorse or promote products derived from this software
21 # without specific prior written permission.
23 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
24 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
25 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
26 # DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
27 # ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
28 # (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
29 # LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
30 # ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
31 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
32 # SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
35 from django.test import TestCase
39 from examples.models import LdapUser, LdapGroup
41 class BaseTestCase(TestCase):
43 cursor = ldapdb.connection._cursor()
44 for dn in [LdapGroup.base_dn, LdapUser.base_dn]:
45 rdn = dn.split(',')[0]
46 key, val = rdn.split('=')
47 attrs = [('objectClass', ['top', 'organizationalUnit']), (key, [val])]
49 cursor.connection.add_s(dn, attrs)
50 except ldap.ALREADY_EXISTS:
54 cursor = ldapdb.connection._cursor()
55 for base in [LdapGroup.base_dn, LdapUser.base_dn]:
57 results = cursor.connection.search_s(base, ldap.SCOPE_SUBTREE)
58 for dn, attrs in reversed(results):
59 cursor.connection.delete_s(dn)
60 except ldap.NO_SUCH_OBJECT:
63 class GroupTestCase(BaseTestCase):
65 super(GroupTestCase, self).setUp()
70 g.usernames = ['foouser', 'baruser']
76 g.usernames = ['zoouser', 'baruser']
82 g.usernames = ['wizuser', 'baruser']
87 qs = LdapGroup.objects.none()
88 self.assertEquals(qs.count(), 0)
90 qs = LdapGroup.objects.none()
91 self.assertEquals(len(qs), 0)
94 qs = LdapGroup.objects.all()
95 self.assertEquals(qs.count(), 3)
97 qs = LdapGroup.objects.all()
98 self.assertEquals(len(qs), 3)
100 def test_filter(self):
101 qs = LdapGroup.objects.filter(name='foogroup')
102 self.assertEquals(qs.count(), 1)
104 qs = LdapGroup.objects.filter(name='foogroup')
105 self.assertEquals(len(qs), 1)
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 filter non-existent entries
114 qs = LdapGroup.objects.filter(name='does_not_exist')
115 self.assertEquals(qs.count(), 0)
117 qs = LdapGroup.objects.filter(name='does_not_exist')
118 self.assertEquals(len(qs), 0)
121 g = LdapGroup.objects.get(name='foogroup')
122 self.assertEquals(g.dn, 'cn=foogroup,%s' % LdapGroup.base_dn)
123 self.assertEquals(g.name, 'foogroup')
124 self.assertEquals(g.gid, 1000)
125 self.assertEquals(g.usernames, ['foouser', 'baruser'])
127 # try to get a non-existent entry
128 self.assertRaises(LdapGroup.DoesNotExist, LdapGroup.objects.get, name='does_not_exist')
130 def test_order_by(self):
132 qs = LdapGroup.objects.order_by('name')
133 self.assertEquals(len(qs), 3)
134 self.assertEquals(qs[0].name, 'bargroup')
135 self.assertEquals(qs[1].name, 'foogroup')
136 self.assertEquals(qs[2].name, 'wizgroup')
139 qs = LdapGroup.objects.order_by('-name')
140 self.assertEquals(len(qs), 3)
141 self.assertEquals(qs[0].name, 'wizgroup')
142 self.assertEquals(qs[1].name, 'foogroup')
143 self.assertEquals(qs[2].name, 'bargroup')
146 qs = LdapGroup.objects.order_by('gid')
147 self.assertEquals(len(qs), 3)
148 self.assertEquals(qs[0].gid, 1000)
149 self.assertEquals(qs[1].gid, 1001)
150 self.assertEquals(qs[2].gid, 1002)
153 qs = LdapGroup.objects.order_by('-gid')
154 self.assertEquals(len(qs), 3)
155 self.assertEquals(qs[0].gid, 1002)
156 self.assertEquals(qs[1].gid, 1001)
157 self.assertEquals(qs[2].gid, 1000)
159 def test_bulk_delete(self):
160 LdapGroup.objects.all().delete()
162 qs = LdapGroup.objects.all()
163 self.assertEquals(len(qs), 0)
165 def test_slice(self):
166 qs = LdapGroup.objects.all()
168 self.assertEquals(len(objs), 3)
169 self.assertEquals(objs[0].gid, 1000)
170 self.assertEquals(objs[1].gid, 1001)
171 self.assertEquals(objs[2].gid, 1002)
174 qs = LdapGroup.objects.all()
176 self.assertEquals(objs.count(), 2)
179 self.assertEquals(len(objs), 2)
180 self.assertEquals(objs[0].gid, 1000)
181 self.assertEquals(objs[1].gid, 1001)
184 qs = LdapGroup.objects.all()
186 self.assertEquals(objs.count(), 2)
189 self.assertEquals(len(objs), 2)
190 self.assertEquals(objs[0].gid, 1001)
191 self.assertEquals(objs[1].gid, 1002)
194 qs = LdapGroup.objects.all()
196 self.assertEquals(objs.count(), 1)
199 self.assertEquals(len(objs), 1)
200 self.assertEquals(objs[0].gid, 1001)
202 def test_update(self):
203 g = LdapGroup.objects.get(name='foogroup')
206 g.usernames = ['foouser2', 'baruser2']
209 # make sure DN gets updated if we change the pk
212 self.assertEquals(g.dn, 'cn=foogroup2,%s' % LdapGroup.base_dn)
214 def test_values(self):
215 qs = LdapGroup.objects.values('name')
216 self.assertEquals(len(qs), 3)
217 self.assertEquals(qs[0], {'name': 'foogroup'})
218 self.assertEquals(qs[1], {'name': 'bargroup'})
219 self.assertEquals(qs[2], {'name': 'wizgroup'})
221 def test_values_list(self):
222 qs = LdapGroup.objects.values_list('name')
223 self.assertEquals(len(qs), 3)
224 self.assertEquals(qs[0], ('foogroup',))
225 self.assertEquals(qs[1], ('bargroup',))
226 self.assertEquals(qs[2], ('wizgroup',))
228 def test_delete(self):
229 g = LdapGroup.objects.get(name='foogroup')
232 qs = LdapGroup.objects.all()
233 self.assertEquals(len(qs), 2)
235 class UserTestCase(BaseTestCase):
237 super(UserTestCase, self).setUp()
240 u.first_name = u"Fôo"
241 u.last_name = u"Usér"
242 u.full_name = u"Fôo Usér"
245 u.home_directory = "/home/foouser"
247 u.username = "foouser"
248 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'
252 u = LdapUser.objects.get(username='foouser')
253 self.assertEquals(u.first_name, u'Fôo')
254 self.assertEquals(u.last_name, u'Usér')
255 self.assertEquals(u.full_name, u'Fôo Usér')
257 self.assertEquals(u.group, 1000)
258 self.assertEquals(u.home_directory, '/home/foouser')
259 self.assertEquals(u.uid, 2000)
260 self.assertEquals(u.username, 'foouser')
261 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')
263 self.assertRaises(LdapUser.DoesNotExist, LdapUser.objects.get, username='does_not_exist')
265 def test_update(self):
266 u = LdapUser.objects.get(username='foouser')
267 u.first_name = u'Fôo2'
270 # make sure DN gets updated if we change the pk
271 u.username = 'foouser2'
273 self.assertEquals(u.dn, 'uid=foouser2,%s' % LdapUser.base_dn)
275 class ScopedTestCase(BaseTestCase):
277 super(ScopedTestCase, self).setUp()
279 cursor = ldapdb.connection._cursor()
280 self.scoped_dn = "ou=contacts,%s" % LdapGroup.base_dn
281 attrs = [('objectClass', ['top', 'organizationalUnit']), ("ou", ["contacts"])]
282 cursor.connection.add_s(self.scoped_dn, attrs)
284 def test_scope(self):
285 ScopedGroup = LdapGroup.scoped(self.scoped_dn)
293 qs = LdapGroup.objects.all()
294 self.assertEquals(qs.count(), 1)
296 qs = ScopedGroup.objects.all()
297 self.assertEquals(qs.count(), 0)
299 # create scoped group
301 g2.name = "scopedgroup"
305 qs = LdapGroup.objects.all()
306 self.assertEquals(qs.count(), 2)
308 qs = ScopedGroup.objects.all()
309 self.assertEquals(qs.count(), 1)
311 class AdminTestCase(BaseTestCase):
312 fixtures = ['test_users.json']
315 super(AdminTestCase, self).setUp()
320 g.usernames = ['foouser', 'baruser']
326 g.usernames = ['zoouser', 'baruser']
332 u.full_name = "Foo User"
334 u.home_directory = "/home/foouser"
336 u.username = "foouser"
342 u.full_name = "Bar User"
344 u.home_directory = "/home/baruser"
346 u.username = "baruser"
349 self.client.login(username="test_user", password="password")
351 def test_index(self):
352 response = self.client.get('/admin/examples/')
353 self.assertContains(response, "Ldap groups")
354 self.assertContains(response, "Ldap users")
356 def test_group_list(self):
357 response = self.client.get('/admin/examples/ldapgroup/')
358 self.assertContains(response, "Ldap groups")
359 self.assertContains(response, "foogroup")
360 self.assertContains(response, "1000")
363 response = self.client.get('/admin/examples/ldapgroup/?o=1')
364 self.assertContains(response, "Ldap groups")
365 self.assertContains(response, "foogroup")
366 self.assertContains(response, "1000")
369 response = self.client.get('/admin/examples/ldapgroup/?o=2')
370 self.assertContains(response, "Ldap groups")
371 self.assertContains(response, "foogroup")
372 self.assertContains(response, "1000")
374 def test_group_detail(self):
375 response = self.client.get('/admin/examples/ldapgroup/foogroup/')
376 self.assertContains(response, "foogroup")
377 self.assertContains(response, "1000")
379 def test_group_delete(self):
380 response = self.client.post('/admin/examples/ldapgroup/foogroup/delete/', {'yes': 'post'})
381 self.assertRedirects(response, '/admin/examples/ldapgroup/')
383 def test_group_search(self):
384 response = self.client.get('/admin/examples/ldapgroup/?q=foo')
385 self.assertContains(response, "Ldap groups")
386 self.assertContains(response, "foogroup")
387 self.assertContains(response, "1000")
389 def test_user_list(self):
390 response = self.client.get('/admin/examples/ldapuser/')
391 self.assertContains(response, "Ldap users")
392 self.assertContains(response, "foouser")
393 self.assertContains(response, "2000")
396 response = self.client.get('/admin/examples/ldapuser/?o=1')
397 self.assertContains(response, "Ldap users")
398 self.assertContains(response, "foouser")
399 self.assertContains(response, "2000")
402 response = self.client.get('/admin/examples/ldapuser/?o=2')
403 self.assertContains(response, "Ldap users")
404 self.assertContains(response, "foouser")
405 self.assertContains(response, "2000")
407 def test_user_detail(self):
408 response = self.client.get('/admin/examples/ldapuser/foouser/')
409 self.assertContains(response, "foouser")
410 self.assertContains(response, "2000")
412 def test_user_delete(self):
413 response = self.client.post('/admin/examples/ldapuser/foouser/delete/', {'yes': 'post'})
414 self.assertRedirects(response, '/admin/examples/ldapuser/')