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/>.
23 from django.test import TestCase
25 from ldapdb import connection
26 from examples.models import LdapUser, LdapGroup
28 class BaseTestCase(TestCase):
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])]
35 cursor.connection.add_s(base, attrs)
36 except ldap.ALREADY_EXISTS:
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)
46 class GroupTestCase(BaseTestCase):
47 def test_create(self):
51 g.usernames = ['foouser', 'baruser']
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'])
62 class UserTestCase(BaseTestCase):
63 def test_create(self):
67 u.full_name = "Foo User"
70 u.home_directory = "/home/foouser"
72 u.username = "foouser"
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')
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, '')