test ordering
[matthijs/upstream/django-ldapdb.git] / 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     def test_filter(self):
66         qs = LdapGroup.objects.none()
67         self.assertEquals(len(qs), 0)
68
69         qs = LdapGroup.objects.all()
70         self.assertEquals(len(qs), 2)
71
72         qs = LdapGroup.objects.filter(name='foogroup')
73         self.assertEquals(len(qs), 1)
74
75         g = qs[0]
76         self.assertEquals(g.dn, 'cn=foogroup,ou=groups,dc=nodomain')
77         self.assertEquals(g.name, 'foogroup')
78         self.assertEquals(g.gid, 1000)
79         self.assertEquals(g.usernames, ['foouser', 'baruser'])
80
81         # try to get non-existent entry
82         qs = LdapGroup.objects.filter(name='does_not_exist')
83         self.assertEquals(len(qs), 0)
84
85     def test_get(self):
86         g = LdapGroup.objects.get(name='foogroup')
87         self.assertEquals(g.dn, 'cn=foogroup,ou=groups,dc=nodomain')
88         self.assertEquals(g.name, 'foogroup')
89         self.assertEquals(g.gid, 1000)
90         self.assertEquals(g.usernames, ['foouser', 'baruser'])
91
92         self.assertRaises(LdapGroup.DoesNotExist, LdapGroup.objects.get, name='does_not_exist')
93
94     def test_order_by(self):
95         # ascending name 
96         qs = LdapGroup.objects.order_by('name')
97         self.assertEquals(len(qs), 2)
98         self.assertEquals(qs[0].name, 'bargroup')
99         self.assertEquals(qs[1].name, 'foogroup')
100
101         # descending name 
102         qs = LdapGroup.objects.order_by('-name')
103         self.assertEquals(len(qs), 2)
104         self.assertEquals(qs[0].name, 'foogroup')
105         self.assertEquals(qs[1].name, 'bargroup')
106
107         # ascending gid
108         qs = LdapGroup.objects.order_by('gid')
109         self.assertEquals(len(qs), 2)
110         self.assertEquals(qs[0].gid, 1000)
111         self.assertEquals(qs[1].gid, 1001)
112
113         # descending gid
114         qs = LdapGroup.objects.order_by('-gid')
115         self.assertEquals(len(qs), 2)
116         self.assertEquals(qs[0].gid, 1001)
117         self.assertEquals(qs[1].gid, 1000)
118
119     def test_update(self):
120         g = LdapGroup.objects.get(name='foogroup')
121
122         g.gid = 1002
123         g.usernames = ['foouser2', 'baruser2']
124         g.save()
125
126         # make sure DN gets updated if we change the pk
127         g.name = 'foogroup2'
128         g.save()
129         self.assertEquals(g.dn, 'cn=foogroup2,ou=groups,dc=nodomain')
130
131     def test_delete(self):
132         g = LdapGroup.objects.get(name='foogroup')
133         g.delete()
134
135         qs = LdapGroup.objects.all()
136         self.assertEquals(len(qs), 1)
137
138 class UserTestCase(BaseTestCase):
139     def setUp(self):
140         super(UserTestCase, self).setUp()
141
142         u = LdapUser()
143         u.first_name = u"Fôo"
144         u.last_name = u"Usér"
145         u.full_name = u"Fôo Usér"
146
147         u.group = 1000
148         u.home_directory = "/home/foouser"
149         u.uid = 2000
150         u.username = "foouser"
151         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'
152         u.save()
153
154     def test_get(self):
155         u = LdapUser.objects.get(username='foouser')
156         self.assertEquals(u.first_name, u'Fôo')
157         self.assertEquals(u.last_name, u'Usér')
158         self.assertEquals(u.full_name, u'Fôo Usér')
159
160         self.assertEquals(u.group, 1000)
161         self.assertEquals(u.home_directory, '/home/foouser')
162         self.assertEquals(u.uid, 2000)
163         self.assertEquals(u.username, 'foouser')
164         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')
165
166         self.assertRaises(LdapUser.DoesNotExist, LdapUser.objects.get, username='does_not_exist')
167
168     def test_update(self):
169         u = LdapUser.objects.get(username='foouser')
170         u.first_name = u'Fôo2'
171         u.save()
172         
173         # make sure DN gets updated if we change the pk
174         u.username = 'foouser2'
175         u.save()
176         self.assertEquals(u.dn, 'uid=foouser2,ou=people,dc=nodomain')
177
178 class AdminTestCase(BaseTestCase):
179     fixtures = ['test_users.json']
180
181     def setUp(self):
182         super(AdminTestCase, self).setUp()
183
184         g = LdapGroup()
185         g.name = "foogroup"
186         g.gid = 1000
187         g.usernames = ['foouser', 'baruser']
188         g.save()
189
190         g = LdapGroup()
191         g.name = "bargroup"
192         g.gid = 1001
193         g.usernames = ['zoouser', 'baruser']
194         g.save()
195
196         u = LdapUser()
197         u.first_name = "Foo"
198         u.last_name = "User"
199         u.full_name = "Foo User"
200         u.group = 1000
201         u.home_directory = "/home/foouser"
202         u.uid = 2000
203         u.username = "foouser"
204         u.save()
205
206         u = LdapUser()
207         u.first_name = "Bar"
208         u.last_name = "User"
209         u.full_name = "Bar User"
210         u.group = 1001
211         u.home_directory = "/home/baruser"
212         u.uid = 2001
213         u.username = "baruser"
214         u.save()
215
216         self.client.login(username="test_user", password="password")
217
218     def test_index(self):
219         response = self.client.get('/admin/examples/')
220         self.assertContains(response, "Ldap groups")
221         self.assertContains(response, "Ldap users")
222
223     def test_group_list(self):
224         response = self.client.get('/admin/examples/ldapgroup/')
225         self.assertContains(response, "Ldap groups")
226         self.assertContains(response, "foogroup")
227         self.assertContains(response, "1000")
228
229         # order by name
230         response = self.client.get('/admin/examples/ldapgroup/?o=1')
231         self.assertContains(response, "Ldap groups")
232         self.assertContains(response, "foogroup")
233         self.assertContains(response, "1000")
234
235         # order by gid
236         response = self.client.get('/admin/examples/ldapgroup/?o=2')
237         self.assertContains(response, "Ldap groups")
238         self.assertContains(response, "foogroup")
239         self.assertContains(response, "1000")
240
241     def test_group_detail(self):
242         response = self.client.get('/admin/examples/ldapgroup/foogroup/')
243         self.assertContains(response, "foogroup")
244         self.assertContains(response, "1000")
245
246     def test_group_search(self):
247         response = self.client.get('/admin/examples/ldapgroup/?q=foo')
248         self.assertContains(response, "Ldap groups")
249         self.assertContains(response, "foogroup")
250         self.assertContains(response, "1000")
251
252     def test_user_list(self):
253         response = self.client.get('/admin/examples/ldapuser/')
254         self.assertContains(response, "Ldap users")
255         self.assertContains(response, "foouser")
256         self.assertContains(response, "2000")
257
258         # order by username
259         response = self.client.get('/admin/examples/ldapuser/?o=1')
260         self.assertContains(response, "Ldap users")
261         self.assertContains(response, "foouser")
262         self.assertContains(response, "2000")
263
264         # order by uid
265         response = self.client.get('/admin/examples/ldapuser/?o=2')
266         self.assertContains(response, "Ldap users")
267         self.assertContains(response, "foouser")
268         self.assertContains(response, "2000")
269
270     def test_user_detail(self):
271         response = self.client.get('/admin/examples/ldapuser/foouser/')
272         self.assertContains(response, "foouser")
273         self.assertContains(response, "2000")