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             ou = base.split(',')[0].split('=')[1]
  33             attrs = [('objectClass', ['top', 'organizationalUnit']), ('ou', [ou])]
  34             try:
  35                 cursor.connection.add_s(base, attrs)
  36             except ldap.ALREADY_EXISTS:
  37                 pass
  38
  39     def tearDown(self):
  40         cursor = connection._cursor()
  41         for base in [LdapGroup.base_dn, LdapUser.base_dn]:
  42             results = cursor.connection.search_s(base, ldap.SCOPE_SUBTREE)
  43             for dn, attrs in reversed(results):
  44                 cursor.connection.delete_s(dn)
  45
  46 class GroupTestCase(BaseTestCase):
  47     def test_create(self):
  48         g = LdapGroup()
  49         g.name = "foogroup"
  50         g.gid = 1000
  51         g.usernames = ['foouser', 'baruser']
  52         g.save()
  53
  54     def test_get(self):
  55         self.test_create()
  56
  57         g = LdapGroup.objects.get(name='foogroup')
  58         self.assertEquals(g.name, 'foogroup')
  59         self.assertEquals(g.gid, 1000)
  60         self.assertEquals(g.usernames, ['foouser', 'baruser'])
  61
  62 class UserTestCase(BaseTestCase):
  63     def test_create(self):
  64         u = LdapUser()
  65         u.first_name = "Foo"
  66         u.last_name = "User"
  67         u.full_name = "Foo User"
  68
  69         u.group = 1000
  70         u.home_directory = "/home/foouser"
  71         u.uid = 1000
  72         u.username = "foouser"
  73         u.save()
  74
  75     def test_get(self):
  76         self.test_create()
  77
  78         u = LdapUser.objects.get(username='foouser')
  79         self.assertEquals(u.first_name, 'Foo')
  80         self.assertEquals(u.last_name, 'User')
  81         self.assertEquals(u.full_name, 'Foo User')
  82
  83         self.assertEquals(u.group, 1000)
  84         self.assertEquals(u.home_directory, '/home/foouser')
  85         self.assertEquals(u.uid, 1000)
  86         self.assertEquals(u.username, 'foouser')
  87         self.assertEquals(u.photo, '')
  88