author | Sylvain Thénault <sylvain.thenault@logilab.fr> |
Tue, 21 Jul 2009 23:09:27 +0200 | |
branch | stable |
changeset 2413 | b1bea4cab594 |
parent 1977 | 606923dff11b |
child 2608 | 21856eda34f6 |
permissions | -rw-r--r-- |
1977
606923dff11b
big bunch of copyright / docstring update
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
1787
diff
changeset
|
1 |
""" |
606923dff11b
big bunch of copyright / docstring update
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
1787
diff
changeset
|
2 |
|
606923dff11b
big bunch of copyright / docstring update
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
1787
diff
changeset
|
3 |
:organization: Logilab |
606923dff11b
big bunch of copyright / docstring update
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
1787
diff
changeset
|
4 |
:copyright: 2001-2009 LOGILAB S.A. (Paris, FRANCE), license is LGPL v2. |
606923dff11b
big bunch of copyright / docstring update
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
1787
diff
changeset
|
5 |
:contact: http://www.logilab.fr/ -- mailto:contact@logilab.fr |
606923dff11b
big bunch of copyright / docstring update
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
1787
diff
changeset
|
6 |
:license: GNU Lesser General Public License, v2.1 - http://www.gnu.org/licenses |
606923dff11b
big bunch of copyright / docstring update
Adrien Di Mascio <Adrien.DiMascio@logilab.fr>
parents:
1787
diff
changeset
|
7 |
""" |
0 | 8 |
from logilab.common.testlib import unittest_main, TestCase |
9 |
from logilab.common.testlib import mock_object |
|
10 |
||
11 |
from rql import parse, nodes, RQLHelper |
|
12 |
||
13 |
from cubicweb import Unauthorized |
|
14 |
from cubicweb.server.rqlrewrite import RQLRewriter |
|
15 |
from cubicweb.devtools import repotest, TestServerConfiguration |
|
16 |
||
17 |
config = TestServerConfiguration('data') |
|
18 |
config.bootstrap_cubes() |
|
19 |
schema = config.load_schema() |
|
20 |
schema.add_relation_def(mock_object(subject='Card', name='in_state', object='State', cardinality='1*')) |
|
1787 | 21 |
|
0 | 22 |
rqlhelper = RQLHelper(schema, special_relations={'eid': 'uid', |
23 |
'has_text': 'fti'}) |
|
24 |
||
25 |
def setup_module(*args): |
|
26 |
repotest.do_monkey_patch() |
|
27 |
||
28 |
def teardown_module(*args): |
|
29 |
repotest.undo_monkey_patch() |
|
1787 | 30 |
|
0 | 31 |
def eid_func_map(eid): |
1398
5fe84a5f7035
rename internal entity types to have CW prefix instead of E
sylvain.thenault@logilab.fr
parents:
47
diff
changeset
|
32 |
return {1: 'CWUser', |
0 | 33 |
2: 'Card'}[eid] |
34 |
||
35 |
def rewrite(rqlst, snippets_map, kwargs): |
|
36 |
class FakeQuerier: |
|
37 |
schema = schema |
|
38 |
@staticmethod |
|
39 |
def solutions(sqlcursor, mainrqlst, kwargs): |
|
40 |
rqlhelper.compute_solutions(rqlst, {'eid': eid_func_map}, kwargs=kwargs) |
|
41 |
class _rqlhelper: |
|
42 |
@staticmethod |
|
43 |
def annotate(rqlst): |
|
44 |
rqlhelper.annotate(rqlst) |
|
45 |
@staticmethod |
|
46 |
def simplify(mainrqlst, needcopy=False): |
|
47 |
rqlhelper.simplify(rqlst, needcopy) |
|
48 |
rewriter = RQLRewriter(FakeQuerier, mock_object(user=(mock_object(eid=1)))) |
|
49 |
for v, snippets in snippets_map.items(): |
|
50 |
snippets_map[v] = [mock_object(snippet_rqlst=parse('Any X WHERE '+snippet).children[0], |
|
1787 | 51 |
expression='Any X WHERE '+snippet) |
0 | 52 |
for snippet in snippets] |
53 |
rqlhelper.compute_solutions(rqlst.children[0], {'eid': eid_func_map}, kwargs=kwargs) |
|
54 |
solutions = rqlst.children[0].solutions |
|
55 |
rewriter.rewrite(rqlst.children[0], snippets_map.items(), solutions, kwargs) |
|
56 |
test_vrefs(rqlst.children[0]) |
|
57 |
return rewriter.rewritten |
|
58 |
||
59 |
def test_vrefs(node): |
|
60 |
vrefmap = {} |
|
61 |
for vref in node.iget_nodes(nodes.VariableRef): |
|
62 |
vrefmap.setdefault(vref.name, set()).add(vref) |
|
63 |
for var in node.defined_vars.itervalues(): |
|
64 |
assert not (var.stinfo['references'] ^ vrefmap[var.name]) |
|
65 |
assert (var.stinfo['references']) |
|
66 |
||
67 |
class RQLRewriteTC(TestCase): |
|
68 |
"""a faire: |
|
69 |
||
70 |
* optimisation: detecter les relations utilisees dans les rqlexpressions qui |
|
71 |
sont presentes dans la requete de depart pour les reutiliser si possible |
|
1787 | 72 |
|
0 | 73 |
* "has_<ACTION>_permission" ? |
74 |
""" |
|
1787 | 75 |
|
0 | 76 |
def test_base_var(self): |
77 |
card_constraint = ('X in_state S, U in_group G, P require_state S,' |
|
78 |
'P name "read", P require_group G') |
|
79 |
rqlst = parse('Card C') |
|
80 |
rewrite(rqlst, {'C': (card_constraint,)}, {}) |
|
81 |
self.failUnlessEqual(rqlst.as_string(), |
|
82 |
u"Any C WHERE C is Card, B eid %(D)s, " |
|
83 |
"EXISTS(C in_state A, B in_group E, F require_state A, " |
|
1398
5fe84a5f7035
rename internal entity types to have CW prefix instead of E
sylvain.thenault@logilab.fr
parents:
47
diff
changeset
|
84 |
"F name 'read', F require_group E, A is State, E is CWGroup, F is CWPermission)") |
1787 | 85 |
|
0 | 86 |
def test_multiple_var(self): |
87 |
card_constraint = ('X in_state S, U in_group G, P require_state S,' |
|
88 |
'P name "read", P require_group G') |
|
89 |
affaire_constraints = ('X ref LIKE "PUBLIC%"', 'U in_group G, G name "public"') |
|
90 |
kwargs = {'u':2} |
|
91 |
rqlst = parse('Any S WHERE S documented_by C, C eid %(u)s') |
|
92 |
rewrite(rqlst, {'C': (card_constraint,), 'S': affaire_constraints}, |
|
93 |
kwargs) |
|
94 |
self.assertTextEquals(rqlst.as_string(), |
|
95 |
"Any S WHERE S documented_by C, C eid %(u)s, B eid %(D)s, " |
|
96 |
"EXISTS(C in_state A, B in_group E, F require_state A, " |
|
1398
5fe84a5f7035
rename internal entity types to have CW prefix instead of E
sylvain.thenault@logilab.fr
parents:
47
diff
changeset
|
97 |
"F name 'read', F require_group E, A is State, E is CWGroup, F is CWPermission), " |
5fe84a5f7035
rename internal entity types to have CW prefix instead of E
sylvain.thenault@logilab.fr
parents:
47
diff
changeset
|
98 |
"(EXISTS(S ref LIKE 'PUBLIC%')) OR (EXISTS(B in_group G, G name 'public', G is CWGroup)), " |
0 | 99 |
"S is Affaire") |
100 |
self.failUnless('D' in kwargs) |
|
1787 | 101 |
|
0 | 102 |
def test_or(self): |
103 |
constraint = '(X identity U) OR (X in_state ST, CL identity U, CL in_state ST, ST name "subscribed")' |
|
104 |
rqlst = parse('Any S WHERE S owned_by C, C eid %(u)s') |
|
105 |
rewrite(rqlst, {'C': (constraint,)}, {'u':1}) |
|
106 |
self.failUnlessEqual(rqlst.as_string(), |
|
107 |
"Any S WHERE S owned_by C, C eid %(u)s, A eid %(B)s, " |
|
108 |
"EXISTS((C identity A) OR (C in_state D, E identity A, " |
|
1398
5fe84a5f7035
rename internal entity types to have CW prefix instead of E
sylvain.thenault@logilab.fr
parents:
47
diff
changeset
|
109 |
"E in_state D, D name 'subscribed'), D is State, E is CWUser), " |
1787 | 110 |
"S is IN(Affaire, Basket, Bookmark, CWAttribute, CWCache, CWConstraint, CWConstraintType, CWEType, CWGroup, CWPermission, CWProperty, CWRType, CWRelation, CWUser, Card, Comment, Division, Email, EmailAddress, EmailPart, EmailThread, File, Folder, Image, Note, Personne, RQLExpression, Societe, State, SubDivision, Tag, TrInfo, Transition)") |
0 | 111 |
|
112 |
def test_simplified_rqlst(self): |
|
113 |
card_constraint = ('X in_state S, U in_group G, P require_state S,' |
|
114 |
'P name "read", P require_group G') |
|
115 |
rqlst = parse('Any 2') # this is the simplified rql st for Any X WHERE X eid 12 |
|
116 |
rewrite(rqlst, {'2': (card_constraint,)}, {}) |
|
117 |
self.failUnlessEqual(rqlst.as_string(), |
|
118 |
u"Any 2 WHERE B eid %(C)s, " |
|
119 |
"EXISTS(2 in_state A, B in_group D, E require_state A, " |
|
1398
5fe84a5f7035
rename internal entity types to have CW prefix instead of E
sylvain.thenault@logilab.fr
parents:
47
diff
changeset
|
120 |
"E name 'read', E require_group D, A is State, D is CWGroup, E is CWPermission)") |
0 | 121 |
|
122 |
def test_optional_var(self): |
|
123 |
card_constraint = ('X in_state S, U in_group G, P require_state S,' |
|
124 |
'P name "read", P require_group G') |
|
125 |
rqlst = parse('Any A,C WHERE A documented_by C?') |
|
126 |
rewrite(rqlst, {'C': (card_constraint,)}, {}) |
|
127 |
self.failUnlessEqual(rqlst.as_string(), |
|
128 |
"Any A,C WHERE A documented_by C?, A is Affaire " |
|
129 |
"WITH C BEING " |
|
130 |
"(Any C WHERE C in_state B, D in_group F, G require_state B, G name 'read', " |
|
131 |
"G require_group F, D eid %(A)s, C is Card)") |
|
132 |
rqlst = parse('Any A,C,T WHERE A documented_by C?, C title T') |
|
133 |
rewrite(rqlst, {'C': (card_constraint,)}, {}) |
|
134 |
self.failUnlessEqual(rqlst.as_string(), |
|
135 |
"Any A,C,T WHERE A documented_by C?, A is Affaire " |
|
136 |
"WITH C,T BEING " |
|
137 |
"(Any C,T WHERE C in_state B, D in_group F, G require_state B, G name 'read', " |
|
138 |
"G require_group F, C title T, D eid %(A)s, C is Card)") |
|
1787 | 139 |
|
0 | 140 |
def test_relation_optimization(self): |
141 |
# since Card in_state State as monovalued cardinality, the in_state |
|
142 |
# relation used in the rql expression can be ignored and S replaced by |
|
143 |
# the variable from the incoming query |
|
144 |
card_constraint = ('X in_state S, U in_group G, P require_state S,' |
|
145 |
'P name "read", P require_group G') |
|
146 |
rqlst = parse('Card C WHERE C in_state STATE') |
|
147 |
rewrite(rqlst, {'C': (card_constraint,)}, {}) |
|
148 |
self.failUnlessEqual(rqlst.as_string(), |
|
149 |
u"Any C WHERE C in_state STATE, C is Card, A eid %(B)s, " |
|
150 |
"EXISTS(A in_group D, E require_state STATE, " |
|
1398
5fe84a5f7035
rename internal entity types to have CW prefix instead of E
sylvain.thenault@logilab.fr
parents:
47
diff
changeset
|
151 |
"E name 'read', E require_group D, D is CWGroup, E is CWPermission), " |
0 | 152 |
"STATE is State") |
153 |
||
154 |
def test_unsupported_constraint_1(self): |
|
1398
5fe84a5f7035
rename internal entity types to have CW prefix instead of E
sylvain.thenault@logilab.fr
parents:
47
diff
changeset
|
155 |
# CWUser doesn't have require_permission |
0 | 156 |
trinfo_constraint = ('X wf_info_for Y, Y require_permission P, P name "read"') |
1398
5fe84a5f7035
rename internal entity types to have CW prefix instead of E
sylvain.thenault@logilab.fr
parents:
47
diff
changeset
|
157 |
rqlst = parse('Any U,T WHERE U is CWUser, T wf_info_for U') |
0 | 158 |
self.assertRaises(Unauthorized, rewrite, rqlst, {'T': (trinfo_constraint,)}, {}) |
1787 | 159 |
|
0 | 160 |
def test_unsupported_constraint_2(self): |
161 |
trinfo_constraint = ('X wf_info_for Y, Y require_permission P, P name "read"') |
|
1398
5fe84a5f7035
rename internal entity types to have CW prefix instead of E
sylvain.thenault@logilab.fr
parents:
47
diff
changeset
|
162 |
rqlst = parse('Any U,T WHERE U is CWUser, T wf_info_for U') |
0 | 163 |
rewrite(rqlst, {'T': (trinfo_constraint, 'X wf_info_for Y, Y in_group G, G name "managers"')}, {}) |
164 |
self.failUnlessEqual(rqlst.as_string(), |
|
1398
5fe84a5f7035
rename internal entity types to have CW prefix instead of E
sylvain.thenault@logilab.fr
parents:
47
diff
changeset
|
165 |
u"Any U,T WHERE U is CWUser, T wf_info_for U, " |
5fe84a5f7035
rename internal entity types to have CW prefix instead of E
sylvain.thenault@logilab.fr
parents:
47
diff
changeset
|
166 |
"EXISTS(U in_group B, B name 'managers', B is CWGroup), T is TrInfo") |
0 | 167 |
|
168 |
def test_unsupported_constraint_3(self): |
|
169 |
self.skip('raise unauthorized for now') |
|
170 |
trinfo_constraint = ('X wf_info_for Y, Y require_permission P, P name "read"') |
|
171 |
rqlst = parse('Any T WHERE T wf_info_for X') |
|
172 |
rewrite(rqlst, {'T': (trinfo_constraint, 'X in_group G, G name "managers"')}, {}) |
|
173 |
self.failUnlessEqual(rqlst.as_string(), |
|
174 |
u'XXX dunno what should be generated') |
|
1787 | 175 |
|
0 | 176 |
def test_add_ambiguity_exists(self): |
177 |
constraint = ('X concerne Y') |
|
178 |
rqlst = parse('Affaire X') |
|
179 |
rewrite(rqlst, {'X': (constraint,)}, {}) |
|
180 |
self.failUnlessEqual(rqlst.as_string(), |
|
181 |
u"Any X WHERE X is Affaire, (((EXISTS(X concerne A, A is Division)) OR (EXISTS(X concerne D, D is SubDivision))) OR (EXISTS(X concerne C, C is Societe))) OR (EXISTS(X concerne B, B is Note))") |
|
1787 | 182 |
|
0 | 183 |
def test_add_ambiguity_outerjoin(self): |
184 |
constraint = ('X concerne Y') |
|
185 |
rqlst = parse('Any X,C WHERE X? documented_by C') |
|
186 |
rewrite(rqlst, {'X': (constraint,)}, {}) |
|
187 |
# ambiguity are kept in the sub-query, no need to be resolved using OR |
|
188 |
self.failUnlessEqual(rqlst.as_string(), |
|
1787 | 189 |
u"Any X,C WHERE X? documented_by C, C is Card WITH X BEING (Any X WHERE X concerne A, X is Affaire)") |
190 |
||
191 |
||
192 |
||
0 | 193 |
if __name__ == '__main__': |
194 |
unittest_main() |