X-Git-Url: https://git.stderr.nl/gitweb?p=matthijs%2Fupstream%2Fdjango-ldapdb.git;a=blobdiff_plain;f=examples%2Ftests.py;h=60f2abc8f26cb2141742df5ccc4a426eccaca506;hp=abdc64809c19d592aed3575c5b34964ab74c4b44;hb=1cbff9629c9af335cc485b0c2a5ef60ec20e87e9;hpb=ddbd5c9e9d0d3343f38906906793bb84a672ad5b diff --git a/examples/tests.py b/examples/tests.py index abdc648..60f2abc 100644 --- a/examples/tests.py +++ b/examples/tests.py @@ -1,23 +1,38 @@ # -*- coding: utf-8 -*- # # django-ldapdb -# Copyright (C) 2009-2010 Bolloré telecom +# Copyright (c) 2009-2010, Bolloré telecom +# All rights reserved. +# # See AUTHORS file for a full list of contributors. # -# This program is free software: you can redistribute it and/or modify -# it under the terms of the GNU General Public License as published by -# the Free Software Foundation, either version 3 of the License, or -# (at your option) any later version. +# Redistribution and use in source and binary forms, with or without modification, +# are permitted provided that the following conditions are met: +# +# 1. Redistributions of source code must retain the above copyright notice, +# this list of conditions and the following disclaimer. +# +# 2. Redistributions in binary form must reproduce the above copyright +# notice, this list of conditions and the following disclaimer in the +# documentation and/or other materials provided with the distribution. # -# This program is distributed in the hope that it will be useful, -# but WITHOUT ANY WARRANTY; without even the implied warranty of -# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the -# GNU General Public License for more details. +# 3. Neither the name of Bolloré telecom nor the names of its contributors +# may be used to endorse or promote products derived from this software +# without specific prior written permission. # -# You should have received a copy of the GNU General Public License -# along with this program. If not, see . +# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND +# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED +# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE +# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR +# ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES +# (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; +# LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON +# ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT +# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS +# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. # +from django.db.models import Q from django.test import TestCase import ldap @@ -27,12 +42,12 @@ from examples.models import LdapUser, LdapGroup class BaseTestCase(TestCase): def setUp(self): cursor = ldapdb.connection._cursor() - for base in [LdapGroup.base_dn, LdapUser.base_dn]: - rdn = base.split(',')[0] + for dn in [LdapGroup.base_dn, LdapUser.base_dn]: + rdn = dn.split(',')[0] key, val = rdn.split('=') attrs = [('objectClass', ['top', 'organizationalUnit']), (key, [val])] try: - cursor.connection.add_s(base, attrs) + cursor.connection.add_s(dn, attrs) except ldap.ALREADY_EXISTS: pass @@ -83,6 +98,39 @@ class GroupTestCase(BaseTestCase): qs = LdapGroup.objects.all() self.assertEquals(len(qs), 3) + def test_ldap_filter(self): + # single filter + qs = LdapGroup.objects.filter(name='foogroup') + self.assertEquals(qs.query._ldap_filter(), '(&(objectClass=posixGroup)(cn=foogroup))') + + qs = LdapGroup.objects.filter(Q(name='foogroup')) + self.assertEquals(qs.query._ldap_filter(), '(&(objectClass=posixGroup)(cn=foogroup))') + + # AND filter + qs = LdapGroup.objects.filter(gid=1000, name='foogroup') + self.assertEquals(qs.query._ldap_filter(), '(&(objectClass=posixGroup)(&(gidNumber=1000)(cn=foogroup)))') + + qs = LdapGroup.objects.filter(Q(gid=1000) & Q(name='foogroup')) + self.assertEquals(qs.query._ldap_filter(), '(&(objectClass=posixGroup)(&(gidNumber=1000)(cn=foogroup)))') + + # OR filter + qs = LdapGroup.objects.filter(Q(gid=1000) | Q(name='foogroup')) + self.assertEquals(qs.query._ldap_filter(), '(&(objectClass=posixGroup)(|(gidNumber=1000)(cn=foogroup)))') + + # single exclusion + qs = LdapGroup.objects.exclude(name='foogroup') + self.assertEquals(qs.query._ldap_filter(), '(&(objectClass=posixGroup)(!(cn=foogroup)))') + + qs = LdapGroup.objects.filter(~Q(name='foogroup')) + self.assertEquals(qs.query._ldap_filter(), '(&(objectClass=posixGroup)(!(cn=foogroup)))') + + # multiple exclusion + qs = LdapGroup.objects.exclude(name='foogroup', gid=1000) + self.assertEquals(qs.query._ldap_filter(), '(&(objectClass=posixGroup)(!(&(gidNumber=1000)(cn=foogroup))))') + + qs = LdapGroup.objects.filter(name='foogroup').exclude(gid=1000) + self.assertEquals(qs.query._ldap_filter(), '(&(objectClass=posixGroup)(&(cn=foogroup)(!(gidNumber=1000))))') + def test_filter(self): qs = LdapGroup.objects.filter(name='foogroup') self.assertEquals(qs.count(), 1) @@ -258,6 +306,42 @@ class UserTestCase(BaseTestCase): u.save() self.assertEquals(u.dn, 'uid=foouser2,%s' % LdapUser.base_dn) +class ScopedTestCase(BaseTestCase): + def setUp(self): + super(ScopedTestCase, self).setUp() + + cursor = ldapdb.connection._cursor() + self.scoped_dn = "ou=contacts,%s" % LdapGroup.base_dn + attrs = [('objectClass', ['top', 'organizationalUnit']), ("ou", ["contacts"])] + cursor.connection.add_s(self.scoped_dn, attrs) + + def test_scope(self): + ScopedGroup = LdapGroup.scoped(self.scoped_dn) + + # create group + g = LdapGroup() + g.name = "foogroup" + g.gid = 1000 + g.save() + + qs = LdapGroup.objects.all() + self.assertEquals(qs.count(), 1) + + qs = ScopedGroup.objects.all() + self.assertEquals(qs.count(), 0) + + # create scoped group + g2 = ScopedGroup() + g2.name = "scopedgroup" + g2.gid = 5000 + g2.save() + + qs = LdapGroup.objects.all() + self.assertEquals(qs.count(), 2) + + qs = ScopedGroup.objects.all() + self.assertEquals(qs.count(), 1) + class AdminTestCase(BaseTestCase): fixtures = ['test_users.json'] @@ -326,9 +410,17 @@ class AdminTestCase(BaseTestCase): self.assertContains(response, "foogroup") self.assertContains(response, "1000") + def test_group_add(self): + response = self.client.post('/admin/examples/ldapgroup/add/', {'gid': '1002', 'name': 'wizgroup'}) + self.assertRedirects(response, '/admin/examples/ldapgroup/') + qs = LdapGroup.objects.all() + self.assertEquals(qs.count(), 3) + def test_group_delete(self): response = self.client.post('/admin/examples/ldapgroup/foogroup/delete/', {'yes': 'post'}) self.assertRedirects(response, '/admin/examples/ldapgroup/') + qs = LdapGroup.objects.all() + self.assertEquals(qs.count(), 1) def test_group_search(self): response = self.client.get('/admin/examples/ldapgroup/?q=foo')