1 from cubicweb.devtools.apptest import EnvBasedTC

     2 from cubicweb.web.views.searchrestriction import extract_filter_fields, insert_attr_select_relation

     3 

     4 class ExtractFilterFieldsTC(EnvBasedTC):

     5     def test_relations_cleanup(self):

     6         # removing relation should be done in the table filter form but not

     7         # from the facets box

     8         rset = self.execute('Any X, S WHERE X in_state S')

     9         afielddefs, baserql, groupby, orderby = extract_filter_fields(rset, 0)

    10         afielddefs = [(getattr(r, 'r_type', r), role, type) for r, role, type, values in afielddefs]

    11         self.assertEquals(afielddefs, [('has_text', 'subject', 'rstring'),

    12                                        ('in_state', 'subject', 'eid')])

    13         self.assertEquals(baserql, 'DISTINCT Any X,S')

    14         self.assertEquals(groupby, '')

    15         self.assertEquals(orderby, '')

    16         # test rql st state

    17         self.assertEquals(rset.syntax_tree().as_string(), 'Any X,S WHERE X in_state S')

    18         afielddefs, baserql, groupby, orderby = extract_filter_fields(rset, 0, removerels=False)

    19         afielddefs = [(getattr(r, 'r_type', r), role, type) for r, role, type, values in afielddefs]

    20         self.assertEquals(afielddefs, [('has_text', 'subject', 'rstring'),

    21                                        ('in_state', 'subject', 'eid')])

    22         self.assertEquals(baserql, 'DISTINCT Any X,S WHERE X in_state S')

    23 

    24 

    25 class InsertAttrRelationTC(EnvBasedTC):

    26     def parse(self, query):

    27         rqlst = self.vreg.parse(self.session, query)

    28         select = rqlst.children[0]

    29         select.remove_groups()

    30         return select

    31 

    32     def _generate(self, select, rel, var, attr):

    33         return insert_attr_select_relation(select, select.defined_vars[var], rel, attr)

    34         

    35     @property

    36     def select(self):

    37         return self.parse('Any B,(NOW - CD),S,V,U,GROUP_CONCAT(TN),VN,P,CD,BMD '

    38                           'GROUPBY B,CD,S,V,U,VN,P,BMD '

    39                           'WHERE B in_state S, B creation_date CD, '

    40                           'B modification_date BMD, T? tags B, T name TN, '

    41                           'V? bookmarked_by B, V title VN, B created_by U?, '

    42                           'B in_group P, P name "managers"')

    43     

    44     def test_1(self):

    45         self.assertEquals(self._generate(self.select, 'in_state', 'S', 'name'),

    46                           "DISTINCT Any S,A ORDERBY A WHERE B in_state S, B in_group P, "

    47                           "P name 'managers', S name A, B is EUser")

    48         

    49     def test_2(self):

    50         self.assertEquals(self._generate(self.select, 'tags', 'T', 'name'),

    51                           "DISTINCT Any T,TN ORDERBY TN WHERE T tags B, T name TN, "

    52                           "B in_group P, P name 'managers', B is EUser")

    53         

    54     def test_3(self):

    55         self.assertEquals(self._generate(self.select, 'created_by', 'U', 'login'),

    56                           "DISTINCT Any U,A ORDERBY A WHERE B created_by U, B in_group P, "

    57                           "P name 'managers', U login A, B is EUser")

    58         

    59     def test_nonregr1(self):

    60         select = self.parse('Any T,V WHERE T bookmarked_by V?, '

    61                             'V in_state VS, VS name "published", T created_by U')

    62         self.assertEquals(self._generate(select, 'created_by', 'U', 'login'),

    63                           'DISTINCT Any U,A ORDERBY A WHERE T created_by U, U login A, '

    64                           'T is Bookmark')

    65 

    66     def test_nonregr2(self):

    67         #'DISTINCT Any X,TMP,N WHERE P name TMP, X version_of P, P is Project, X is Version, not X in_state S,S name "published", X num N ORDERBY TMP,N'

    68         select = self.parse('DISTINCT Any V,TN,L ORDERBY TN,L WHERE T nom TN, V connait T, T is Personne, V is EUser,'

    69                             'NOT V in_state VS, VS name "published", V login L')

    70         rschema = self.schema['connait']

    71         for s, o in rschema.iter_rdefs():

    72             rschema.set_rproperty(s, o, 'cardinality', '++')

    73         try:

    74             self.assertEquals(self._generate(select, 'in_state', 'VS', 'name'),

    75                               "DISTINCT Any VS,A ORDERBY A WHERE V is EUser, NOT V in_state VS, VS name 'published', VS name A")

    76         finally:

    77             for s, o in rschema.iter_rdefs():

    78                 rschema.set_rproperty(s, o, 'cardinality', '**')

    79 

    80     def test_nonregr3(self):

    81         #'DISTINCT Any X,TMP,N WHERE P name TMP, X version_of P, P is Project, X is Version, not X in_state S,S name "published", X num N ORDERBY TMP,N'

    82         select = self.parse('DISTINCT Any X, MAX(Y) GROUPBY X WHERE X is EUser, Y is Bookmark, X in_group A')

    83         self.assertEquals(self._generate(select, 'in_group', 'A', 'name'),

    84                           "DISTINCT Any A,B ORDERBY B WHERE X is EUser, X in_group A, A name B")

    85 

    86         

    87 if __name__ == '__main__':

    88     from logilab.common.testlib import unittest_main

    89     unittest_main()