server/test/unittest_rqlannotation.py
changeset 9850 5ef9dd383ae2
parent 9675 8aabfefc8a81
child 10600 180aa08cad48
equal deleted inserted replaced
9849:b18ef631e72c 9850:5ef9dd383ae2
     1 # -*- coding: iso-8859-1 -*-
     1 # -*- coding: iso-8859-1 -*-
     2 # copyright 2003-2012 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
     2 # copyright 2003-2014 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
     3 # contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
     3 # contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
     4 #
     4 #
     5 # This file is part of CubicWeb.
     5 # This file is part of CubicWeb.
     6 #
     6 #
     7 # CubicWeb is free software: you can redistribute it and/or modify it under the
     7 # CubicWeb is free software: you can redistribute it and/or modify it under the
    19 """unit tests for modules cubicweb.server.rqlannotation"""
    19 """unit tests for modules cubicweb.server.rqlannotation"""
    20 
    20 
    21 from cubicweb.devtools import TestServerConfiguration, get_test_db_handler
    21 from cubicweb.devtools import TestServerConfiguration, get_test_db_handler
    22 from cubicweb.devtools.repotest import BaseQuerierTC
    22 from cubicweb.devtools.repotest import BaseQuerierTC
    23 
    23 
    24 
       
    25 def setUpModule(*args):
       
    26     handler = get_test_db_handler(TestServerConfiguration(
       
    27         'data2', apphome=SQLGenAnnotatorTC.datadir))
       
    28     handler.build_db_cache()
       
    29     global repo, cnx
       
    30     repo, cnx = handler.get_repo_and_cnx()
       
    31 
       
    32 def tearDownModule(*args):
       
    33     global repo, cnx
       
    34     del repo, cnx
       
    35 
       
    36 
       
    37 class SQLGenAnnotatorTC(BaseQuerierTC):
    24 class SQLGenAnnotatorTC(BaseQuerierTC):
    38 
    25 
    39     def setUp(self):
    26     def setUp(self):
       
    27         handler = get_test_db_handler(TestServerConfiguration(
       
    28             'data2', apphome=SQLGenAnnotatorTC.datadir))
       
    29         handler.build_db_cache()
       
    30         repo, _cnx = handler.get_repo_and_cnx()
    40         self.__class__.repo = repo
    31         self.__class__.repo = repo
    41         super(SQLGenAnnotatorTC, self).setUp()
    32         super(SQLGenAnnotatorTC, self).setUp()
    42 
    33 
    43     def get_max_eid(self):
    34     def get_max_eid(self):
    44         # no need for cleanup here
    35         # no need for cleanup here
    45         return None
    36         return None
       
    37 
    46     def cleanup(self):
    38     def cleanup(self):
    47         # no need for cleanup here
    39         # no need for cleanup here
    48         pass
    40         pass
    49 
    41 
    50     def test_0_1(self):
    42     def test_0_1(self):
    51         rqlst = self._prepare('Any SEN,RN,OEN WHERE X from_entity SE, SE eid 44, X relation_type R, R eid 139, X to_entity OE, OE eid 42, R name RN, SE name SEN, OE name OEN')
    43         with self.session.new_cnx() as cnx:
    52         self.assertEqual(rqlst.defined_vars['SE']._q_invariant, False)
    44             rqlst = self._prepare(cnx, 'Any SEN,RN,OEN WHERE X from_entity SE, '
    53         self.assertEqual(rqlst.defined_vars['OE']._q_invariant, False)
    45                                   'SE eid 44, X relation_type R, R eid 139, '
    54         self.assertEqual(rqlst.defined_vars['R']._q_invariant, False)
    46                                   'X to_entity OE, OE eid 42, R name RN, SE name SEN, '
    55         self.assertEqual(rqlst.defined_vars['SE'].stinfo['attrvar'], None)
    47                                   'OE name OEN')
    56         self.assertEqual(rqlst.defined_vars['OE'].stinfo['attrvar'], None)
    48             self.assertEqual(rqlst.defined_vars['SE']._q_invariant, False)
    57         self.assertEqual(rqlst.defined_vars['R'].stinfo['attrvar'], None)
    49             self.assertEqual(rqlst.defined_vars['OE']._q_invariant, False)
       
    50             self.assertEqual(rqlst.defined_vars['R']._q_invariant, False)
       
    51             self.assertEqual(rqlst.defined_vars['SE'].stinfo['attrvar'], None)
       
    52             self.assertEqual(rqlst.defined_vars['OE'].stinfo['attrvar'], None)
       
    53             self.assertEqual(rqlst.defined_vars['R'].stinfo['attrvar'], None)
    58 
    54 
    59     def test_0_2(self):
    55     def test_0_2(self):
    60         rqlst = self._prepare('Any O WHERE NOT S ecrit_par O, S eid 1, S inline1 P, O inline2 P')
    56         with self.session.new_cnx() as cnx:
    61         self.assertEqual(rqlst.defined_vars['P']._q_invariant, True)
    57             rqlst = self._prepare(cnx, 'Any O WHERE NOT S ecrit_par O, S eid 1, '
    62         self.assertEqual(rqlst.defined_vars['O'].stinfo['attrvar'], None)
    58                                   'S inline1 P, O inline2 P')
       
    59             self.assertEqual(rqlst.defined_vars['P']._q_invariant, True)
       
    60             self.assertEqual(rqlst.defined_vars['O'].stinfo['attrvar'], None)
    63 
    61 
    64     def test_0_4(self):
    62     def test_0_4(self):
    65         rqlst = self._prepare('Any A,B,C WHERE A eid 12,A comment B, A ?wf_info_for C')
    63         with self.session.new_cnx() as cnx:
    66         self.assertEqual(rqlst.defined_vars['A']._q_invariant, False)
    64             rqlst = self._prepare(cnx, 'Any A,B,C WHERE A eid 12,A comment B, '
    67         self.assert_(rqlst.defined_vars['B'].stinfo['attrvar'])
    65                                   'A ?wf_info_for C')
    68         self.assertEqual(rqlst.defined_vars['C']._q_invariant, False)
    66             self.assertEqual(rqlst.defined_vars['A']._q_invariant, False)
    69         self.assertEqual(rqlst.solutions, [{'A': 'TrInfo', 'B': 'String', 'C': 'Affaire'},
    67             self.assert_(rqlst.defined_vars['B'].stinfo['attrvar'])
    70                                       {'A': 'TrInfo', 'B': 'String', 'C': 'CWUser'},
    68             self.assertEqual(rqlst.defined_vars['C']._q_invariant, False)
    71                                       {'A': 'TrInfo', 'B': 'String', 'C': 'Note'}])
    69             self.assertEqual(rqlst.solutions, [{'A': 'TrInfo', 'B': 'String', 'C': 'Affaire'},
       
    70                                           {'A': 'TrInfo', 'B': 'String', 'C': 'CWUser'},
       
    71                                           {'A': 'TrInfo', 'B': 'String', 'C': 'Note'}])
    72 
    72 
    73     def test_0_5(self):
    73     def test_0_5(self):
    74         rqlst = self._prepare('Any P WHERE N ecrit_par P, N eid 0')
    74         with self.session.new_cnx() as cnx:
    75         self.assertEqual(rqlst.defined_vars['N']._q_invariant, False)
    75             rqlst = self._prepare(cnx, 'Any P WHERE N ecrit_par P, N eid 0')
    76         self.assertEqual(rqlst.defined_vars['P']._q_invariant, True)
    76             self.assertEqual(rqlst.defined_vars['N']._q_invariant, False)
       
    77             self.assertEqual(rqlst.defined_vars['P']._q_invariant, True)
    77 
    78 
    78     def test_0_6(self):
    79     def test_0_6(self):
    79         rqlst = self._prepare('Any P WHERE NOT N ecrit_par P, N eid 512')
    80         with self.session.new_cnx() as cnx:
    80         self.assertEqual(rqlst.defined_vars['P']._q_invariant, False)
    81             rqlst = self._prepare(cnx, 'Any P WHERE NOT N ecrit_par P, N eid 512')
       
    82             self.assertEqual(rqlst.defined_vars['P']._q_invariant, False)
    81 
    83 
    82     def test_0_7(self):
    84     def test_0_7(self):
    83         rqlst = self._prepare('Personne X,Y where X nom NX, Y nom NX, X eid XE, not Y eid XE')
    85         with self.session.new_cnx() as cnx:
    84         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
    86             rqlst = self._prepare(cnx, 'Personne X,Y where X nom NX, '
    85         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
    87                                   'Y nom NX, X eid XE, not Y eid XE')
    86         self.assert_(rqlst.defined_vars['XE'].stinfo['attrvar'])
    88             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
       
    89             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
       
    90             self.assert_(rqlst.defined_vars['XE'].stinfo['attrvar'])
    87 
    91 
    88     def test_0_8(self):
    92     def test_0_8(self):
    89         rqlst = self._prepare('Any P WHERE X eid 0, NOT X connait P')
    93         with self.session.new_cnx() as cnx:
    90         self.assertEqual(rqlst.defined_vars['P']._q_invariant, False)
    94             rqlst = self._prepare(cnx, 'Any P WHERE X eid 0, NOT X connait P')
    91         #self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
    95             self.assertEqual(rqlst.defined_vars['P']._q_invariant, False)
    92         self.assertEqual(len(rqlst.solutions), 1, rqlst.solutions)
    96             #self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
       
    97             self.assertEqual(len(rqlst.solutions), 1, rqlst.solutions)
    93 
    98 
    94     def test_0_10(self):
    99     def test_0_10(self):
    95         rqlst = self._prepare('Any X WHERE X concerne Y, Y is Note')
   100         with self.session.new_cnx() as cnx:
    96         self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   101             rqlst = self._prepare(cnx, 'Any X WHERE X concerne Y, Y is Note')
    97         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   102             self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
       
   103             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
    98 
   104 
    99     def test_0_11(self):
   105     def test_0_11(self):
   100         rqlst = self._prepare('Any X WHERE X todo_by Y, X is Affaire')
   106         with self.session.new_cnx() as cnx:
   101         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   107             rqlst = self._prepare(cnx, 'Any X WHERE X todo_by Y, X is Affaire')
   102         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   108             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
       
   109             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   103 
   110 
   104     def test_0_12(self):
   111     def test_0_12(self):
   105         rqlst = self._prepare('Personne P WHERE P concerne A, A concerne S, S nom "Logilab"')
   112         with self.session.new_cnx() as cnx:
   106         self.assertEqual(rqlst.defined_vars['P']._q_invariant, True)
   113             rqlst = self._prepare(cnx, 'Personne P WHERE P concerne A, '
   107         self.assertEqual(rqlst.defined_vars['A']._q_invariant, True)
   114                                   'A concerne S, S nom "Logilab"')
   108         self.assertEqual(rqlst.defined_vars['S']._q_invariant, False)
   115             self.assertEqual(rqlst.defined_vars['P']._q_invariant, True)
       
   116             self.assertEqual(rqlst.defined_vars['A']._q_invariant, True)
       
   117             self.assertEqual(rqlst.defined_vars['S']._q_invariant, False)
   109 
   118 
   110     def test_1_0(self):
   119     def test_1_0(self):
   111         rqlst = self._prepare('Any X,Y WHERE X created_by Y, X eid 5, NOT Y eid 6')
   120         with self.session.new_cnx() as cnx:
   112         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   121             rqlst = self._prepare(cnx, 'Any X,Y WHERE X created_by Y, '
       
   122                                   'X eid 5, NOT Y eid 6')
       
   123             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   113 
   124 
   114     def test_1_1(self):
   125     def test_1_1(self):
   115         rqlst = self._prepare('Any X,Y WHERE X created_by Y, X eid 5, NOT Y eid IN (6,7)')
   126         with self.session.new_cnx() as cnx:
   116         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   127             rqlst = self._prepare(cnx, 'Any X,Y WHERE X created_by Y, X eid 5, '
       
   128                                   'NOT Y eid IN (6,7)')
       
   129             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   117 
   130 
   118     def test_2(self):
   131     def test_2(self):
   119         rqlst = self._prepare('Any X WHERE X identity Y, Y eid 1')
   132         with self.session.new_cnx() as cnx:
   120         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   133             rqlst = self._prepare(cnx, 'Any X WHERE X identity Y, Y eid 1')
       
   134             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   121 
   135 
   122     def test_7(self):
   136     def test_7(self):
   123         rqlst = self._prepare('Personne X,Y where X nom NX, Y nom NX, X eid XE, not Y eid XE')
   137         with self.session.new_cnx() as cnx:
   124         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   138             rqlst = self._prepare(cnx, 'Personne X,Y where X nom NX, Y nom NX, '
   125         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   139                                   'X eid XE, not Y eid XE')
       
   140             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
       
   141             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   126 
   142 
   127     def test_8(self):
   143     def test_8(self):
   128         # DISTINCT Any P WHERE P require_group %(g)s, NOT %(u)s has_group_permission P, P is CWPermission
   144         with self.session.new_cnx() as cnx:
   129         rqlst = self._prepare('DISTINCT Any X WHERE A concerne X, NOT N migrated_from X, '
   145             # DISTINCT Any P WHERE P require_group %(g)s,
   130                               'X is Note, N eid 1')
   146             # NOT %(u)s has_group_permission P, P is CWPermission
   131         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   147             rqlst = self._prepare(cnx, 'DISTINCT Any X WHERE A concerne X, '
       
   148                                   'NOT N migrated_from X, '
       
   149                                   'X is Note, N eid 1')
       
   150             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   132 
   151 
   133     def test_diff_scope_identity_deamb(self):
   152     def test_diff_scope_identity_deamb(self):
   134         rqlst = self._prepare('Any X WHERE X concerne Y, Y is Note, EXISTS(Y identity Z, Z migrated_from N)')
   153         with self.session.new_cnx() as cnx:
   135         self.assertEqual(rqlst.defined_vars['Z']._q_invariant, True)
   154             rqlst = self._prepare(cnx, 'Any X WHERE X concerne Y, Y is Note, '
   136         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   155                                   'EXISTS(Y identity Z, Z migrated_from N)')
       
   156             self.assertEqual(rqlst.defined_vars['Z']._q_invariant, True)
       
   157             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   137 
   158 
   138     def test_optional_inlined(self):
   159     def test_optional_inlined(self):
   139         rqlst = self._prepare('Any X,S where X from_state S?')
   160         with self.session.new_cnx() as cnx:
   140         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   161             rqlst = self._prepare(cnx, 'Any X,S where X from_state S?')
   141         self.assertEqual(rqlst.defined_vars['S']._q_invariant, True)
   162             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
       
   163             self.assertEqual(rqlst.defined_vars['S']._q_invariant, True)
   142 
   164 
   143     def test_optional_inlined_2(self):
   165     def test_optional_inlined_2(self):
   144         rqlst = self._prepare('Any N,A WHERE N? inline1 A')
   166         with self.session.new_cnx() as cnx:
   145         self.assertEqual(rqlst.defined_vars['N']._q_invariant, False)
   167             rqlst = self._prepare(cnx, 'Any N,A WHERE N? inline1 A')
   146         self.assertEqual(rqlst.defined_vars['A']._q_invariant, False)
   168             self.assertEqual(rqlst.defined_vars['N']._q_invariant, False)
       
   169             self.assertEqual(rqlst.defined_vars['A']._q_invariant, False)
   147 
   170 
   148     def test_optional_1(self):
   171     def test_optional_1(self):
   149         rqlst = self._prepare('Any X,S WHERE X travaille S?')
   172         with self.session.new_cnx() as cnx:
   150         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   173             rqlst = self._prepare(cnx, 'Any X,S WHERE X travaille S?')
   151         self.assertEqual(rqlst.defined_vars['S']._q_invariant, True)
   174             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
       
   175             self.assertEqual(rqlst.defined_vars['S']._q_invariant, True)
   152 
   176 
   153     def test_greater_eid(self):
   177     def test_greater_eid(self):
   154         rqlst = self._prepare('Any X WHERE X eid > 5')
   178         with self.session.new_cnx() as cnx:
   155         self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   179             rqlst = self._prepare(cnx, 'Any X WHERE X eid > 5')
       
   180             self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   156 
   181 
   157     def test_greater_eid_typed(self):
   182     def test_greater_eid_typed(self):
   158         rqlst = self._prepare('Any X WHERE X eid > 5, X is Note')
   183         with self.session.new_cnx() as cnx:
   159         self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   184             rqlst = self._prepare(cnx, 'Any X WHERE X eid > 5, X is Note')
       
   185             self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   160 
   186 
   161     def test_max_eid(self):
   187     def test_max_eid(self):
   162         rqlst = self._prepare('Any MAX(X)')
   188         with self.session.new_cnx() as cnx:
   163         self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   189             rqlst = self._prepare(cnx, 'Any MAX(X)')
       
   190             self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   164 
   191 
   165     def test_max_eid_typed(self):
   192     def test_max_eid_typed(self):
   166         rqlst = self._prepare('Any MAX(X) WHERE X is Note')
   193         with self.session.new_cnx() as cnx:
   167         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   194             rqlst = self._prepare(cnx, 'Any MAX(X) WHERE X is Note')
       
   195             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   168 
   196 
   169     def test_all_entities(self):
   197     def test_all_entities(self):
   170         rqlst = self._prepare('Any X')
   198         with self.session.new_cnx() as cnx:
   171         self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   199             rqlst = self._prepare(cnx, 'Any X')
       
   200             self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   172 
   201 
   173     def test_all_typed_entity(self):
   202     def test_all_typed_entity(self):
   174         rqlst = self._prepare('Any X WHERE X is Note')
   203         with self.session.new_cnx() as cnx:
   175         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   204             rqlst = self._prepare(cnx, 'Any X WHERE X is Note')
       
   205             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   176 
   206 
   177     def test_has_text_1(self):
   207     def test_has_text_1(self):
   178         rqlst = self._prepare('Any X WHERE X has_text "toto tata"')
   208         with self.session.new_cnx() as cnx:
   179         self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   209             rqlst = self._prepare(cnx, 'Any X WHERE X has_text "toto tata"')
   180         self.assertEqual(rqlst.defined_vars['X'].stinfo['principal'].r_type, 'has_text')
   210             self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
       
   211             self.assertEqual(rqlst.defined_vars['X'].stinfo['principal'].r_type,
       
   212                              'has_text')
   181 
   213 
   182     def test_has_text_2(self):
   214     def test_has_text_2(self):
   183         rqlst = self._prepare('Any X WHERE X is Personne, X has_text "coucou"')
   215         with self.session.new_cnx() as cnx:
   184         self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   216             rqlst = self._prepare(cnx, 'Any X WHERE X is Personne, '
   185         self.assertEqual(rqlst.defined_vars['X'].stinfo['principal'].r_type, 'has_text')
   217                                   'X has_text "coucou"')
       
   218             self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
       
   219             self.assertEqual(rqlst.defined_vars['X'].stinfo['principal'].r_type,
       
   220                              'has_text')
   186 
   221 
   187     def test_not_relation_1(self):
   222     def test_not_relation_1(self):
   188         # P can't be invariant since deambiguification caused by "NOT X require_permission P"
   223         with self.session.new_cnx() as cnx:
   189         # is not considered by generated sql (NOT EXISTS(...))
   224             # P can't be invariant since deambiguification caused by "NOT X require_permission P"
   190         rqlst = self._prepare('Any P,G WHERE P require_group G, NOT X require_permission P')
   225             # is not considered by generated sql (NOT EXISTS(...))
   191         self.assertEqual(rqlst.defined_vars['P']._q_invariant, False)
   226             rqlst = self._prepare(cnx, 'Any P,G WHERE P require_group G, '
   192         self.assertEqual(rqlst.defined_vars['G']._q_invariant, True)
   227                                   'NOT X require_permission P')
   193         self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   228             self.assertEqual(rqlst.defined_vars['P']._q_invariant, False)
       
   229             self.assertEqual(rqlst.defined_vars['G']._q_invariant, True)
       
   230             self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   194 
   231 
   195     def test_not_relation_2(self):
   232     def test_not_relation_2(self):
   196         rqlst = self._prepare('TrInfo X WHERE X eid 2, NOT X from_state Y, Y is State')
   233         with self.session.new_cnx() as cnx:
   197         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   234             rqlst = self._prepare(cnx, 'TrInfo X WHERE X eid 2, '
   198         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   235                                   'NOT X from_state Y, Y is State')
       
   236             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
       
   237             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   199 
   238 
   200     def test_not_relation_3(self):
   239     def test_not_relation_3(self):
   201         rqlst = self._prepare('Any X, Y WHERE X eid 1, Y eid in (2, 3)')
   240         with self.session.new_cnx() as cnx:
   202         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   241             rqlst = self._prepare(cnx, 'Any X, Y WHERE X eid 1, Y eid in (2, 3)')
       
   242             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   203 
   243 
   204     def test_not_relation_4_1(self):
   244     def test_not_relation_4_1(self):
   205         rqlst = self._prepare('Note X WHERE NOT Y evaluee X')
   245         with self.session.new_cnx() as cnx:
   206         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   246             rqlst = self._prepare(cnx, 'Note X WHERE NOT Y evaluee X')
   207         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   247             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
       
   248             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   208 
   249 
   209     def test_not_relation_4_2(self):
   250     def test_not_relation_4_2(self):
   210         rqlst = self._prepare('Any X WHERE NOT Y evaluee X')
   251         with self.session.new_cnx() as cnx:
   211         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   252             rqlst = self._prepare(cnx, 'Any X WHERE NOT Y evaluee X')
   212         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   253             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
       
   254             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   213 
   255 
   214     def test_not_relation_4_3(self):
   256     def test_not_relation_4_3(self):
   215         rqlst = self._prepare('Any Y WHERE NOT Y evaluee X')
   257         with self.session.new_cnx() as cnx:
   216         self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   258             rqlst = self._prepare(cnx, 'Any Y WHERE NOT Y evaluee X')
   217         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   259             self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
       
   260             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   218 
   261 
   219     def test_not_relation_4_4(self):
   262     def test_not_relation_4_4(self):
   220         rqlst = self._prepare('Any X WHERE NOT Y evaluee X, Y is CWUser')
   263         with self.session.new_cnx() as cnx:
   221         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   264             rqlst = self._prepare(cnx, 'Any X WHERE NOT Y evaluee X, Y is CWUser')
   222         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   265             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
       
   266             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   223 
   267 
   224     def test_not_relation_4_5(self):
   268     def test_not_relation_4_5(self):
   225         rqlst = self._prepare('Any X WHERE NOT Y evaluee X, Y eid %s, X is Note' % self.ueid)
   269         with self.session.new_cnx() as cnx:
   226         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   270             rqlst = self._prepare(cnx, 'Any X WHERE NOT Y evaluee X, '
   227         self.assertEqual(rqlst.solutions, [{'X': 'Note'}])
   271                                   'Y eid %s, X is Note' % self.ueid)
       
   272             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
       
   273             self.assertEqual(rqlst.solutions, [{'X': 'Note'}])
   228 
   274 
   229     def test_not_relation_5_1(self):
   275     def test_not_relation_5_1(self):
   230         rqlst = self._prepare('Any X,Y WHERE X name "CWGroup", Y eid IN(1, 2, 3), NOT X read_permission Y')
   276         with self.session.new_cnx() as cnx:
   231         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   277             rqlst = self._prepare(cnx, 'Any X,Y WHERE X name "CWGroup", '
   232         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   278                                   'Y eid IN(1, 2, 3), NOT X read_permission Y')
       
   279             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
       
   280             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   233 
   281 
   234     def test_not_relation_5_2(self):
   282     def test_not_relation_5_2(self):
   235         rqlst = self._prepare('DISTINCT Any X,Y WHERE X name "CWGroup", Y eid IN(1, 2, 3), NOT X read_permission Y')
   283         with self.session.new_cnx() as cnx:
   236         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   284             rqlst = self._prepare(cnx, 'DISTINCT Any X,Y WHERE X name "CWGroup", '
   237         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   285                                   'Y eid IN(1, 2, 3), NOT X read_permission Y')
       
   286             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
       
   287             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   238 
   288 
   239     def test_not_relation_6(self):
   289     def test_not_relation_6(self):
   240         rqlst = self._prepare('Personne P where NOT P concerne A')
   290         with self.session.new_cnx() as cnx:
   241         self.assertEqual(rqlst.defined_vars['P']._q_invariant, False)
   291             rqlst = self._prepare(cnx, 'Personne P where NOT P concerne A')
   242         self.assertEqual(rqlst.defined_vars['A']._q_invariant, True)
   292             self.assertEqual(rqlst.defined_vars['P']._q_invariant, False)
       
   293             self.assertEqual(rqlst.defined_vars['A']._q_invariant, True)
   243 
   294 
   244     def test_not_relation_7(self):
   295     def test_not_relation_7(self):
   245         rqlst = self._prepare('Any K,V WHERE P is CWProperty, P pkey K, P value V, NOT P for_user U')
   296         with self.session.new_cnx() as cnx:
   246         self.assertEqual(rqlst.defined_vars['P']._q_invariant, False)
   297             rqlst = self._prepare(cnx, 'Any K,V WHERE P is CWProperty, '
   247         self.assertEqual(rqlst.defined_vars['U']._q_invariant, True)
   298                                   'P pkey K, P value V, NOT P for_user U')
       
   299             self.assertEqual(rqlst.defined_vars['P']._q_invariant, False)
       
   300             self.assertEqual(rqlst.defined_vars['U']._q_invariant, True)
   248 
   301 
   249     def test_exists_1(self):
   302     def test_exists_1(self):
   250         rqlst = self._prepare('Any U WHERE U eid IN (1,2), EXISTS(X owned_by U)')
   303         with self.session.new_cnx() as cnx:
   251         self.assertEqual(rqlst.defined_vars['U']._q_invariant, False)
   304             rqlst = self._prepare(cnx, 'Any U WHERE U eid IN (1,2), EXISTS(X owned_by U)')
   252         self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   305             self.assertEqual(rqlst.defined_vars['U']._q_invariant, False)
       
   306             self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   253 
   307 
   254     def test_exists_2(self):
   308     def test_exists_2(self):
   255         rqlst = self._prepare('Any U WHERE EXISTS(U eid IN (1,2), X owned_by U)')
   309         with self.session.new_cnx() as cnx:
   256         self.assertEqual(rqlst.defined_vars['U']._q_invariant, False)
   310             rqlst = self._prepare(cnx, 'Any U WHERE EXISTS(U eid IN (1,2), X owned_by U)')
   257         self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   311             self.assertEqual(rqlst.defined_vars['U']._q_invariant, False)
       
   312             self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   258 
   313 
   259     def test_exists_3(self):
   314     def test_exists_3(self):
   260         rqlst = self._prepare('Any U WHERE EXISTS(X owned_by U, X bookmarked_by U)')
   315         with self.session.new_cnx() as cnx:
   261         self.assertEqual(rqlst.defined_vars['U']._q_invariant, False)
   316             rqlst = self._prepare(cnx, 'Any U WHERE EXISTS(X owned_by U, X bookmarked_by U)')
   262         self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   317             self.assertEqual(rqlst.defined_vars['U']._q_invariant, False)
       
   318             self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   263 
   319 
   264     def test_exists_4(self):
   320     def test_exists_4(self):
   265         rqlst = self._prepare('Any X,Y WHERE X name "CWGroup", Y eid IN(1, 2, 3), EXISTS(X read_permission Y)')
   321         with self.session.new_cnx() as cnx:
   266         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   322             rqlst = self._prepare(cnx, 'Any X,Y WHERE X name "CWGroup", '
   267         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   323                                   'Y eid IN(1, 2, 3), EXISTS(X read_permission Y)')
       
   324             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
       
   325             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   268 
   326 
   269     def test_exists_5(self):
   327     def test_exists_5(self):
   270         rqlst = self._prepare('DISTINCT Any X,Y WHERE X name "CWGroup", Y eid IN(1, 2, 3), EXISTS(X read_permission Y)')
   328         with self.session.new_cnx() as cnx:
   271         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   329             rqlst = self._prepare(cnx, 'DISTINCT Any X,Y WHERE X name "CWGroup", '
   272         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   330                                   'Y eid IN(1, 2, 3), EXISTS(X read_permission Y)')
       
   331             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
       
   332             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   273 
   333 
   274     def test_not_exists_1(self):
   334     def test_not_exists_1(self):
   275         rqlst = self._prepare('Any U WHERE NOT EXISTS(X owned_by U, X bookmarked_by U)')
   335         with self.session.new_cnx() as cnx:
   276         self.assertEqual(rqlst.defined_vars['U']._q_invariant, False)
   336             rqlst = self._prepare(cnx, 'Any U WHERE NOT EXISTS(X owned_by U, '
   277         self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   337                                   'X bookmarked_by U)')
       
   338             self.assertEqual(rqlst.defined_vars['U']._q_invariant, False)
       
   339             self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   278 
   340 
   279     def test_not_exists_2(self):
   341     def test_not_exists_2(self):
   280         rqlst = self._prepare('Any X,Y WHERE X name "CWGroup", Y eid IN(1, 2, 3), NOT EXISTS(X read_permission Y)')
   342         with self.session.new_cnx() as cnx:
   281         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   343             rqlst = self._prepare(cnx, 'Any X,Y WHERE X name "CWGroup", '
       
   344                                   'Y eid IN(1, 2, 3), NOT EXISTS(X read_permission Y)')
       
   345             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   282 
   346 
   283     def test_not_exists_distinct_1(self):
   347     def test_not_exists_distinct_1(self):
   284         rqlst = self._prepare('DISTINCT Any X,Y WHERE X name "CWGroup", Y eid IN(1, 2, 3), NOT EXISTS(X read_permission Y)')
   348         with self.session.new_cnx() as cnx:
   285         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   349             rqlst = self._prepare(cnx, 'DISTINCT Any X,Y WHERE X name "CWGroup", '
       
   350                                   'Y eid IN(1, 2, 3), NOT EXISTS(X read_permission Y)')
       
   351             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, False)
   286 
   352 
   287     def test_or_1(self):
   353     def test_or_1(self):
   288         rqlst = self._prepare('Any X WHERE X concerne B OR C concerne X, B eid 12, C eid 13')
   354         with self.session.new_cnx() as cnx:
   289         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   355             rqlst = self._prepare(cnx, 'Any X WHERE X concerne B OR '
       
   356                                   'C concerne X, B eid 12, C eid 13')
       
   357             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   290 
   358 
   291     def test_or_2(self):
   359     def test_or_2(self):
   292         rqlst = self._prepare('Any X WHERE X created_by U, X concerne B OR C concerne X, B eid 12, C eid 13')
   360         with self.session.new_cnx() as cnx:
   293         self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   361             rqlst = self._prepare(cnx, 'Any X WHERE X created_by U, X concerne B OR '
   294         self.assertEqual(rqlst.defined_vars['U']._q_invariant, True)
   362                                   'C concerne X, B eid 12, C eid 13')
   295         self.assertEqual(rqlst.defined_vars['X'].stinfo['principal'].r_type, 'created_by')
   363             self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
       
   364             self.assertEqual(rqlst.defined_vars['U']._q_invariant, True)
       
   365             self.assertEqual(rqlst.defined_vars['X'].stinfo['principal'].r_type, 'created_by')
   296 
   366 
   297     def test_or_3(self):
   367     def test_or_3(self):
   298         rqlst = self._prepare('Any N WHERE A evaluee N or EXISTS(N todo_by U)')
   368         with self.session.new_cnx() as cnx:
   299         self.assertEqual(rqlst.defined_vars['N']._q_invariant, False)
   369             rqlst = self._prepare(cnx, 'Any N WHERE A evaluee N or EXISTS(N todo_by U)')
   300         self.assertEqual(rqlst.defined_vars['A']._q_invariant, True)
   370             self.assertEqual(rqlst.defined_vars['N']._q_invariant, False)
   301         self.assertEqual(rqlst.defined_vars['U']._q_invariant, True)
   371             self.assertEqual(rqlst.defined_vars['A']._q_invariant, True)
       
   372             self.assertEqual(rqlst.defined_vars['U']._q_invariant, True)
   302 
   373 
   303     def test_or_exists_1(self):
   374     def test_or_exists_1(self):
   304         # query generated by security rewriting
   375         with self.session.new_cnx() as cnx:
   305         rqlst = self._prepare('DISTINCT Any A,S WHERE A is Affaire, S nom "chouette", S is IN(Division, Societe, SubDivision),'
   376             # query generated by security rewriting
   306                               '(EXISTS(A owned_by D)) '
   377             rqlst = self._prepare(cnx, 'DISTINCT Any A,S WHERE A is Affaire, S nom "chouette", '
   307                               'OR ((((EXISTS(E concerne C?, C owned_by D, A identity E, C is Note, E is Affaire)) '
   378                                   'S is IN(Division, Societe, SubDivision),'
   308                               'OR (EXISTS(I concerne H?, H owned_by D, H is Societe, A identity I, I is Affaire))) '
   379                                   '(EXISTS(A owned_by D)) '
   309                               'OR (EXISTS(J concerne G?, G owned_by D, G is SubDivision, A identity J, J is Affaire))) '
   380                                   'OR ((((EXISTS(E concerne C?, C owned_by D, A identity E, '
   310                               'OR (EXISTS(K concerne F?, F owned_by D, F is Division, A identity K, K is Affaire)))')
   381                                   '              C is Note, E is Affaire)) '
   311         self.assertEqual(rqlst.defined_vars['A']._q_invariant, False)
   382                                   'OR (EXISTS(I concerne H?, H owned_by D, H is Societe, '
   312         self.assertEqual(rqlst.defined_vars['S']._q_invariant, False)
   383                                   '           A identity I, I is Affaire))) '
       
   384                                   'OR (EXISTS(J concerne G?, G owned_by D, G is SubDivision, '
       
   385                                   '           A identity J, J is Affaire))) '
       
   386                                   'OR (EXISTS(K concerne F?, F owned_by D, F is Division, '
       
   387                                   '           A identity K, K is Affaire)))')
       
   388             self.assertEqual(rqlst.defined_vars['A']._q_invariant, False)
       
   389             self.assertEqual(rqlst.defined_vars['S']._q_invariant, False)
   313 
   390 
   314     def test_or_exists_2(self):
   391     def test_or_exists_2(self):
   315         rqlst = self._prepare('Any U WHERE EXISTS(U in_group G, G name "managers") OR EXISTS(X owned_by U, X bookmarked_by U)')
   392         with self.session.new_cnx() as cnx:
   316         self.assertEqual(rqlst.defined_vars['U']._q_invariant, False)
   393             rqlst = self._prepare(cnx, 'Any U WHERE EXISTS(U in_group G, G name "managers") OR '
   317         self.assertEqual(rqlst.defined_vars['G']._q_invariant, False)
   394                                   'EXISTS(X owned_by U, X bookmarked_by U)')
   318         self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   395             self.assertEqual(rqlst.defined_vars['U']._q_invariant, False)
       
   396             self.assertEqual(rqlst.defined_vars['G']._q_invariant, False)
       
   397             self.assertEqual(rqlst.defined_vars['X']._q_invariant, True)
   319 
   398 
   320     def test_or_exists_3(self):
   399     def test_or_exists_3(self):
   321         rqlst = self._prepare('Any COUNT(S),CS GROUPBY CS ORDERBY 1 DESC LIMIT 10 '
   400         with self.session.new_cnx() as cnx:
   322                               'WHERE C is Societe, S concerne C, C nom CS, '
   401             rqlst = self._prepare(cnx, 'Any COUNT(S),CS GROUPBY CS ORDERBY 1 DESC LIMIT 10 '
   323                               '(EXISTS(S owned_by D)) OR (EXISTS(S documented_by N, N title "published"))')
   402                                   'WHERE C is Societe, S concerne C, C nom CS, '
   324         self.assertEqual(rqlst.defined_vars['S']._q_invariant, True)
   403                                   '(EXISTS(S owned_by D)) OR (EXISTS(S documented_by N, N title "published"))')
   325         rqlst = self._prepare('Any COUNT(S),CS GROUPBY CS ORDERBY 1 DESC LIMIT 10 '
   404             self.assertEqual(rqlst.defined_vars['S']._q_invariant, True)
   326                               'WHERE S is Affaire, C is Societe, S concerne C, C nom CS, '
   405             rqlst = self._prepare(cnx, 'Any COUNT(S),CS GROUPBY CS ORDERBY 1 DESC LIMIT 10 '
   327                               '(EXISTS(S owned_by D)) OR (EXISTS(S documented_by N, N title "published"))')
   406                                   'WHERE S is Affaire, C is Societe, S concerne C, C nom CS, '
   328         self.assertEqual(rqlst.defined_vars['S']._q_invariant, True)
   407                                   '(EXISTS(S owned_by D)) OR (EXISTS(S documented_by N, N title "published"))')
       
   408             self.assertEqual(rqlst.defined_vars['S']._q_invariant, True)
   329 
   409 
   330     def test_nonregr_ambiguity(self):
   410     def test_nonregr_ambiguity(self):
   331         rqlst = self._prepare('Note N WHERE N attachment F')
   411         with self.session.new_cnx() as cnx:
   332         # N may be an image as well, not invariant
   412             rqlst = self._prepare(cnx, 'Note N WHERE N attachment F')
   333         self.assertEqual(rqlst.defined_vars['N']._q_invariant, False)
   413             # N may be an image as well, not invariant
   334         self.assertEqual(rqlst.defined_vars['F']._q_invariant, True)
   414             self.assertEqual(rqlst.defined_vars['N']._q_invariant, False)
       
   415             self.assertEqual(rqlst.defined_vars['F']._q_invariant, True)
   335 
   416 
   336     def test_nonregr_ambiguity_2(self):
   417     def test_nonregr_ambiguity_2(self):
   337         rqlst = self._prepare('Any S,SN WHERE X has_text "tot", X in_state S, S name SN, X is CWUser')
   418         with self.session.new_cnx() as cnx:
   338         # X use has_text but should not be invariant as ambiguous, and has_text
   419             rqlst = self._prepare(cnx, 'Any S,SN WHERE X has_text "tot", X in_state S, S name SN, X is CWUser')
   339         # may not be its principal
   420             # X use has_text but should not be invariant as ambiguous, and has_text
   340         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   421             # may not be its principal
   341         self.assertEqual(rqlst.defined_vars['S']._q_invariant, False)
   422             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
       
   423             self.assertEqual(rqlst.defined_vars['S']._q_invariant, False)
   342 
   424 
   343     def test_remove_from_deleted_source_1(self):
   425     def test_remove_from_deleted_source_1(self):
   344         rqlst = self._prepare('Note X WHERE X eid 999998, NOT X cw_source Y')
   426         with self.session.new_cnx() as cnx:
   345         self.assertNotIn('X', rqlst.defined_vars) # simplified
   427             rqlst = self._prepare(cnx, 'Note X WHERE X eid 999998, NOT X cw_source Y')
   346         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   428             self.assertNotIn('X', rqlst.defined_vars) # simplified
       
   429             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   347 
   430 
   348     def test_remove_from_deleted_source_2(self):
   431     def test_remove_from_deleted_source_2(self):
   349         rqlst = self._prepare('Note X WHERE X eid IN (999998, 999999), NOT X cw_source Y')
   432         with self.session.new_cnx() as cnx:
   350         self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   433             rqlst = self._prepare(cnx, 'Note X WHERE X eid IN (999998, 999999), NOT X cw_source Y')
   351         self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   434             self.assertEqual(rqlst.defined_vars['X']._q_invariant, False)
   352 
   435             self.assertEqual(rqlst.defined_vars['Y']._q_invariant, True)
   353 
   436 
   354     def test_has_text_security_cache_bug(self):
   437     def test_has_text_security_cache_bug(self):
   355         rqlst = self._prepare('Any X WHERE X has_text "toto" WITH X BEING '
   438         with self.session.new_cnx() as cnx:
   356                               '(Any C WHERE C is Societe, C nom CS)')
   439             rqlst = self._prepare(cnx, 'Any X WHERE X has_text "toto" WITH X BEING '
   357         self.assertTrue(rqlst.parent.has_text_query)
   440                                   '(Any C WHERE C is Societe, C nom CS)')
       
   441             self.assertTrue(rqlst.parent.has_text_query)
   358 
   442 
   359 if __name__ == '__main__':
   443 if __name__ == '__main__':
   360     from logilab.common.testlib import unittest_main
   444     from logilab.common.testlib import unittest_main
   361     unittest_main()
   445     unittest_main()