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 import ldap
  22
  23 from django.test import TestCase
  24
  25 from ldapdb import connection
  26 from examples.models import LdapUser, LdapGroup
  27
  28 class BaseTestCase(TestCase):
  29     def setUp(self):
  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])]
  35             try:
  36                 cursor.connection.add_s(base, attrs)
  37             except ldap.ALREADY_EXISTS:
  38                 pass
  39
  40     def tearDown(self):
  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)
  46
  47 class GroupTestCase(BaseTestCase):
  48     def setUp(self):
  49         super(GroupTestCase, self).setUp()
  50
  51         g = LdapGroup()
  52         g.name = "foogroup"
  53         g.gid = 1000
  54         g.usernames = ['foouser', 'baruser']
  55         g.save()
  56
  57         g = LdapGroup()
  58         g.name = "bargroup"
  59         g.gid = 1001
  60         g.usernames = ['zoouser', 'baruser']
  61         g.save()
  62
  63     def test_filter(self):
  64         qs = LdapGroup.objects.none()
  65         self.assertEquals(len(qs), 0)
  66
  67         qs = LdapGroup.objects.all()
  68         self.assertEquals(len(qs), 2)
  69
  70         qs = LdapGroup.objects.filter(name='foogroup')
  71         self.assertEquals(len(qs), 1)
  72
  73         g = qs[0]
  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'])
  78
  79         qs = LdapGroup.objects.filter(name='does_not_exist')
  80         self.assertEquals(len(qs), 0)
  81
  82     def test_get(self):
  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'])
  88
  89         self.assertRaises(LdapGroup.DoesNotExist, LdapGroup.objects.get, name='does_not_exist')
  90
  91     def test_update(self):
  92         g = LdapGroup.objects.get(name='foogroup')
  93
  94         g.gid = 1002
  95         g.usernames = ['foouser2', 'baruser2']
  96         g.save()
  97
  98         # make sure DN gets updated if we change the pk
  99         g.name = 'foogroup2'
 100         g.save()
 101         self.assertEquals(g.dn, 'cn=foogroup2,ou=groups,dc=nodomain')
 102
 103     def test_delete(self):
 104         g = LdapGroup.objects.get(name='foogroup')
 105         g.delete()
 106
 107         qs = LdapGroup.objects.all()
 108         self.assertEquals(len(qs), 1)
 109
 110 class UserTestCase(BaseTestCase):
 111     def setUp(self):
 112         super(UserTestCase, self).setUp()
 113
 114         u = LdapUser()
 115         u.first_name = u"Fôo"
 116         u.last_name = u"Usér"
 117         u.full_name = u"Fôo Usér"
 118
 119         u.group = 1000
 120         u.home_directory = "/home/foouser"
 121         u.uid = 1000
 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'
 124         u.save()
 125
 126     def test_get(self):
 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')
 131
 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')
 137
 138         self.assertRaises(LdapUser.DoesNotExist, LdapUser.objects.get, username='does_not_exist')
 139
 140     def test_update(self):
 141         u = LdapUser.objects.get(username='foouser')
 142         u.first_name = u'Fôo2'
 143         u.save()
 144
 145         # make sure DN gets updated if we change the pk
 146         u.username = 'foouser2'
 147         u.save()
 148         self.assertEquals(u.dn, 'uid=foouser2,ou=people,dc=nodomain')
 149
 150 class AdminTestCase(BaseTestCase):
 151     fixtures = ['test_users.json']
 152
 153     def setUp(self):
 154         super(AdminTestCase, self).setUp()
 155
 156         g = LdapGroup()
 157         g.name = "foogroup"
 158         g.gid = 1000
 159         g.usernames = ['foouser', 'baruser']
 160         g.save()
 161
 162         u = LdapUser()
 163         u.first_name = "Foo"
 164         u.last_name = "User"
 165         u.full_name = "Foo User"
 166         u.group = 1000
 167         u.home_directory = "/home/foouser"
 168         u.uid = 1000
 169         u.username = "foouser"
 170         u.save()
 171
 172         self.client.login(username="test_user", password="password")
 173
 174     def test_index(self):
 175         response = self.client.get('/admin/examples/')
 176         self.assertContains(response, "Ldap groups")
 177         self.assertContains(response, "Ldap users")
 178
 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")
 183
 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")
 188
 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")
 193
 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")