add db prep
[matthijs/upstream/django-ldapdb.git] / 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.fields import CharField
25 from ldapdb.models.query import WhereNode
26
27 class FieldTestCase(TestCase):
28     def test_db_prep(self):
29         field = CharField()
30
31 class WhereTestCase(TestCase):
32     def test_single(self):
33         where = WhereNode()
34         where.add((Constraint("cn", "cn", None), 'exact', "test"), AND)
35         self.assertEquals(where.as_sql(), "(cn=test)")
36
37         where = WhereNode()
38         where.add((Constraint("cn", "cn", None), 'startswith', "test"), AND)
39         self.assertEquals(where.as_sql(), "(cn=test*)")
40
41         where = WhereNode()
42         where.add((Constraint("cn", "cn", None), 'endswith', "test"), AND)
43         self.assertEquals(where.as_sql(), "(cn=*test)")
44
45         where = WhereNode()
46         where.add((Constraint("cn", "cn", None), 'in', ["foo", "bar"]), AND)
47         self.assertEquals(where.as_sql(), "(|(cn=foo)(cn=bar))")
48
49     def test_and(self):
50         where = WhereNode()
51         where.add((Constraint("cn", "cn", None), 'exact', "foo"), AND)
52         where.add((Constraint("givenName", "givenName", None), 'exact', "bar"), AND)
53         self.assertEquals(where.as_sql(), "(&(cn=foo)(givenName=bar))")
54
55     def test_or(self):
56         where = WhereNode()
57         where.add((Constraint("cn", "cn", None), 'exact', "foo"), AND)
58         where.add((Constraint("givenName", "givenName", None), 'exact', "bar"), OR)
59         self.assertEquals(where.as_sql(), "(|(cn=foo)(givenName=bar))")
60