examples/tests.py

   1 # -*- coding: utf-8 -*-
   2 #
   3 # django-ldapdb
   4 # Copyright (C) 2009-2010 Bolloré telecom
   5 # See AUTHORS file for a full list of contributors.
   6 #
   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.
  11 #
  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.
  16 #
  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/>.
  19 #
  20
  21 from django.test import TestCase
  22
  23 import ldap
  24 import ldapdb
  25 from examples.models import LdapUser, LdapGroup
  26
  27 class BaseTestCase(TestCase):
  28     def setUp(self):
  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])]
  34             try:
  35                 cursor.connection.add_s(base, attrs)
  36             except ldap.ALREADY_EXISTS:
  37                 pass
  38
  39     def tearDown(self):
  40         cursor = ldapdb.connection._cursor()
  41         for base in [LdapGroup.base_dn, LdapUser.base_dn]:
  42             try:
  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:
  47                 pass
  48
  49 class GroupTestCase(BaseTestCase):
  50     def setUp(self):
  51         super(GroupTestCase, self).setUp()
  52
  53         g = LdapGroup()
  54         g.name = "foogroup"
  55         g.gid = 1000
  56         g.usernames = ['foouser', 'baruser']
  57         g.save()
  58
  59         g = LdapGroup()
  60         g.name = "bargroup"
  61         g.gid = 1001
  62         g.usernames = ['zoouser', 'baruser']
  63         g.save()
  64
  65         g = LdapGroup()
  66         g.name = "wizgroup"
  67         g.gid = 1002
  68         g.usernames = ['wizuser', 'baruser']
  69         g.save()
  70
  71     def test_count(self):
  72         # empty query
  73         qs = LdapGroup.objects.none()
  74         self.assertEquals(qs.count(), 0)
  75
  76         qs = LdapGroup.objects.none()
  77         self.assertEquals(len(qs), 0)
  78
  79         # all query
  80         qs = LdapGroup.objects.all()
  81         self.assertEquals(qs.count(), 3)
  82
  83         qs = LdapGroup.objects.all()
  84         self.assertEquals(len(qs), 3)
  85
  86     def test_filter(self):
  87         qs = LdapGroup.objects.filter(name='foogroup')
  88         self.assertEquals(qs.count(), 1)
  89
  90         qs = LdapGroup.objects.filter(name='foogroup')
  91         self.assertEquals(len(qs), 1)
  92
  93         g = qs[0]
  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'])
  98
  99         # try to filter non-existent entries
 100         qs = LdapGroup.objects.filter(name='does_not_exist')
 101         self.assertEquals(qs.count(), 0)
 102
 103         qs = LdapGroup.objects.filter(name='does_not_exist')
 104         self.assertEquals(len(qs), 0)
 105
 106     def test_get(self):
 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'])
 112
 113         # try to get a non-existent entry
 114         self.assertRaises(LdapGroup.DoesNotExist, LdapGroup.objects.get, name='does_not_exist')
 115
 116     def test_order_by(self):
 117         # ascending name
 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')
 123
 124         # descending name
 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')
 130
 131         # ascending gid
 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)
 137
 138         # descending gid
 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)
 144
 145     def test_bulk_delete(self):
 146         LdapGroup.objects.all().delete()
 147
 148         qs = LdapGroup.objects.all()
 149         self.assertEquals(len(qs), 0)
 150
 151     def test_slice(self):
 152         qs = LdapGroup.objects.all()
 153         objs = list(qs)
 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)
 158
 159         # limit only
 160         qs = LdapGroup.objects.all()
 161         objs = qs[:2]
 162         self.assertEquals(objs.count(), 2)
 163
 164         objs = qs[:2]
 165         self.assertEquals(len(objs), 2)
 166         self.assertEquals(objs[0].gid, 1000)
 167         self.assertEquals(objs[1].gid, 1001)
 168
 169         # offset only
 170         qs = LdapGroup.objects.all()
 171         objs = qs[1:]
 172         self.assertEquals(objs.count(), 2)
 173
 174         objs = qs[1:]
 175         self.assertEquals(len(objs), 2)
 176         self.assertEquals(objs[0].gid, 1001)
 177         self.assertEquals(objs[1].gid, 1002)
 178
 179         # offset and limit
 180         qs = LdapGroup.objects.all()
 181         objs = qs[1:2]
 182         self.assertEquals(objs.count(), 1)
 183
 184         objs = qs[1:2]
 185         self.assertEquals(len(objs), 1)
 186         self.assertEquals(objs[0].gid, 1001)
 187
 188     def test_update(self):
 189         g = LdapGroup.objects.get(name='foogroup')
 190
 191         g.gid = 1002
 192         g.usernames = ['foouser2', 'baruser2']
 193         g.save()
 194
 195         # make sure DN gets updated if we change the pk
 196         g.name = 'foogroup2'
 197         g.save()
 198         self.assertEquals(g.dn, 'cn=foogroup2,%s' % LdapGroup.base_dn)
 199
 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'})
 206
 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',))
 213
 214     def test_delete(self):
 215         g = LdapGroup.objects.get(name='foogroup')
 216         g.delete()
 217
 218         qs = LdapGroup.objects.all()
 219         self.assertEquals(len(qs), 2)
 220
 221 class UserTestCase(BaseTestCase):
 222     def setUp(self):
 223         super(UserTestCase, self).setUp()
 224
 225         u = LdapUser()
 226         u.first_name = u"Fôo"
 227         u.last_name = u"Usér"
 228         u.full_name = u"Fôo Usér"
 229
 230         u.group = 1000
 231         u.home_directory = "/home/foouser"
 232         u.uid = 2000
 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'
 235         u.save()
 236
 237     def test_get(self):
 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')
 242
 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')
 248
 249         self.assertRaises(LdapUser.DoesNotExist, LdapUser.objects.get, username='does_not_exist')
 250
 251     def test_update(self):
 252         u = LdapUser.objects.get(username='foouser')
 253         u.first_name = u'Fôo2'
 254         u.save()
 255
 256         # make sure DN gets updated if we change the pk
 257         u.username = 'foouser2'
 258         u.save()
 259         self.assertEquals(u.dn, 'uid=foouser2,%s' % LdapUser.base_dn)
 260
 261 class AdminTestCase(BaseTestCase):
 262     fixtures = ['test_users.json']
 263
 264     def setUp(self):
 265         super(AdminTestCase, self).setUp()
 266
 267         g = LdapGroup()
 268         g.name = "foogroup"
 269         g.gid = 1000
 270         g.usernames = ['foouser', 'baruser']
 271         g.save()
 272
 273         g = LdapGroup()
 274         g.name = "bargroup"
 275         g.gid = 1001
 276         g.usernames = ['zoouser', 'baruser']
 277         g.save()
 278
 279         u = LdapUser()
 280         u.first_name = "Foo"
 281         u.last_name = "User"
 282         u.full_name = "Foo User"
 283         u.group = 1000
 284         u.home_directory = "/home/foouser"
 285         u.uid = 2000
 286         u.username = "foouser"
 287         u.save()
 288
 289         u = LdapUser()
 290         u.first_name = "Bar"
 291         u.last_name = "User"
 292         u.full_name = "Bar User"
 293         u.group = 1001
 294         u.home_directory = "/home/baruser"
 295         u.uid = 2001
 296         u.username = "baruser"
 297         u.save()
 298
 299         self.client.login(username="test_user", password="password")
 300
 301     def test_index(self):
 302         response = self.client.get('/admin/examples/')
 303         self.assertContains(response, "Ldap groups")
 304         self.assertContains(response, "Ldap users")
 305
 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")
 311
 312         # order by name
 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")
 317
 318         # order by gid
 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")
 323
 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")
 328
 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/')
 332
 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")
 338
 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")
 344
 345         # order by username
 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")
 350
 351         # order by uid
 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")
 356
 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")
 361
 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/')
 365