ldapdb/tests.py

   1 # -*- coding: utf-8 -*-
   2 #
   3 # django-ldapdb
   4 # Copyright (C) 2009 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 from django.db.models.sql.where import Constraint, AND, OR
  23
  24 from ldapdb.models.query import WhereNode
  25
  26 class WhereTestCase(TestCase):
  27     def test_single(self):
  28         where = WhereNode()
  29         where.add((Constraint("cn", "cn", None), 'exact', "test"), AND)
  30         self.assertEquals(where.as_sql(), "(cn=test)")
  31
  32         where = WhereNode()
  33         where.add((Constraint("cn", "cn", None), 'startswith', "test"), AND)
  34         self.assertEquals(where.as_sql(), "(cn=test*)")
  35
  36         where = WhereNode()
  37         where.add((Constraint("cn", "cn", None), 'endswith', "test"), AND)
  38         self.assertEquals(where.as_sql(), "(cn=*test)")
  39
  40         where = WhereNode()
  41         where.add((Constraint("cn", "cn", None), 'in', ["foo", "bar"]), AND)
  42         self.assertEquals(where.as_sql(), "(|(cn=foo)(cn=bar))")
  43
  44     def test_and(self):
  45         where = WhereNode()
  46         where.add((Constraint("cn", "cn", None), 'exact', "foo"), AND)
  47         where.add((Constraint("givenName", "givenName", None), 'exact', "bar"), AND)
  48         self.assertEquals(where.as_sql(), "(&(cn=foo)(givenName=bar))")
  49
  50     def test_or(self):
  51         where = WhereNode()
  52         where.add((Constraint("cn", "cn", None), 'exact', "foo"), AND)
  53         where.add((Constraint("givenName", "givenName", None), 'exact', "bar"), OR)
  54         self.assertEquals(where.as_sql(), "(|(cn=foo)(givenName=bar))")
  55