cubicweb/test/unittest_req.py
changeset 11057 0b59724cb3f2
parent 9654 f788e5e7e6ea
child 11794 d8830e2bd2e0
equal deleted inserted replaced
11052:058bb3dc685f 11057:0b59724cb3f2
       
     1 # copyright 2003-2014 LOGILAB S.A. (Paris, FRANCE), all rights reserved.
       
     2 # contact http://www.logilab.fr/ -- mailto:contact@logilab.fr
       
     3 #
       
     4 # This file is part of CubicWeb.
       
     5 #
       
     6 # CubicWeb is free software: you can redistribute it and/or modify it under the
       
     7 # terms of the GNU Lesser General Public License as published by the Free
       
     8 # Software Foundation, either version 2.1 of the License, or (at your option)
       
     9 # any later version.
       
    10 #
       
    11 # CubicWeb is distributed in the hope that it will be useful, but WITHOUT
       
    12 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
       
    13 # FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
       
    14 # details.
       
    15 #
       
    16 # You should have received a copy of the GNU Lesser General Public License along
       
    17 # with CubicWeb.  If not, see <http://www.gnu.org/licenses/>.
       
    18 
       
    19 from logilab.common.testlib import TestCase, unittest_main
       
    20 from cubicweb import ObjectNotFound
       
    21 from cubicweb.req import RequestSessionBase, FindEntityError
       
    22 from cubicweb.devtools.testlib import CubicWebTC
       
    23 from cubicweb import Unauthorized
       
    24 
       
    25 class RequestTC(TestCase):
       
    26     def test_rebuild_url(self):
       
    27         rebuild_url = RequestSessionBase(None).rebuild_url
       
    28         self.assertEqual(rebuild_url('http://logilab.fr?__message=pouet', __message='hop'),
       
    29                           'http://logilab.fr?__message=hop')
       
    30         self.assertEqual(rebuild_url('http://logilab.fr', __message='hop'),
       
    31                           'http://logilab.fr?__message=hop')
       
    32         self.assertEqual(rebuild_url('http://logilab.fr?vid=index', __message='hop'),
       
    33                           'http://logilab.fr?__message=hop&vid=index')
       
    34 
       
    35     def test_build_url(self):
       
    36         req = RequestSessionBase(None)
       
    37         req.from_controller = lambda : 'view'
       
    38         req.relative_path = lambda includeparams=True: None
       
    39         req.base_url = lambda secure=None: 'http://testing.fr/cubicweb/'
       
    40         self.assertEqual(req.build_url(), u'http://testing.fr/cubicweb/view')
       
    41         self.assertEqual(req.build_url(None), u'http://testing.fr/cubicweb/view')
       
    42         self.assertEqual(req.build_url('one'), u'http://testing.fr/cubicweb/one')
       
    43         self.assertEqual(req.build_url(param='ok'), u'http://testing.fr/cubicweb/view?param=ok')
       
    44         self.assertRaises(AssertionError, req.build_url, 'one', 'two not allowed')
       
    45         self.assertRaises(AssertionError, req.build_url, 'view', test=None)
       
    46 
       
    47     def test_ensure_no_rql(self):
       
    48         req = RequestSessionBase(None)
       
    49         self.assertEqual(req.ensure_ro_rql('Any X WHERE X is CWUser'), None)
       
    50         self.assertEqual(req.ensure_ro_rql('  Any X WHERE X is CWUser  '), None)
       
    51         self.assertRaises(Unauthorized, req.ensure_ro_rql, 'SET X login "toto" WHERE X is CWUser')
       
    52         self.assertRaises(Unauthorized, req.ensure_ro_rql, '   SET X login "toto" WHERE X is CWUser   ')
       
    53 
       
    54 
       
    55 class RequestCWTC(CubicWebTC):
       
    56 
       
    57     def test_base_url(self):
       
    58         base_url = self.config['base-url']
       
    59         with self.admin_access.repo_cnx() as session:
       
    60             self.assertEqual(session.base_url(), base_url)
       
    61             assert 'https-url' not in self.config
       
    62             self.assertEqual(session.base_url(secure=True), base_url)
       
    63             secure_base_url = base_url.replace('http', 'https')
       
    64             self.config.global_set_option('https-url', secure_base_url)
       
    65             self.assertEqual(session.base_url(secure=True), secure_base_url)
       
    66 
       
    67     def test_view_catch_ex(self):
       
    68         with self.admin_access.web_request() as req:
       
    69             rset = req.execute('CWUser X WHERE X login "hop"')
       
    70             self.assertEqual(req.view('oneline', rset, 'null'), '')
       
    71             self.assertRaises(ObjectNotFound, req.view, 'onelinee', rset, 'null')
       
    72 
       
    73     def test_find_one_entity(self):
       
    74         with self.admin_access.web_request() as req:
       
    75             req.create_entity(
       
    76                 'CWUser', login=u'cdevienne', upassword=u'cdevienne',
       
    77                 surname=u'de Vienne', firstname=u'Christophe',
       
    78                 in_group=req.find('CWGroup', name=u'users').one())
       
    79 
       
    80             req.create_entity(
       
    81                 'CWUser', login=u'adim', upassword='adim', surname=u'di mascio',
       
    82                 firstname=u'adrien',
       
    83                 in_group=req.find('CWGroup', name=u'users').one())
       
    84 
       
    85             u = req.find_one_entity('CWUser', login=u'cdevienne')
       
    86             self.assertEqual(u.firstname, u"Christophe")
       
    87 
       
    88             with self.assertRaises(FindEntityError):
       
    89                 req.find_one_entity('CWUser', login=u'patanok')
       
    90 
       
    91             with self.assertRaises(FindEntityError):
       
    92                 req.find_one_entity('CWUser')
       
    93 
       
    94     def test_find_entities(self):
       
    95         with self.admin_access.web_request() as req:
       
    96             req.create_entity(
       
    97                 'CWUser', login=u'cdevienne', upassword=u'cdevienne',
       
    98                 surname=u'de Vienne', firstname=u'Christophe',
       
    99                 in_group=req.find('CWGroup', name=u'users').one())
       
   100 
       
   101             req.create_entity(
       
   102                 'CWUser', login=u'adim', upassword='adim', surname=u'di mascio',
       
   103                 firstname=u'adrien',
       
   104                 in_group=req.find('CWGroup', name=u'users').one())
       
   105 
       
   106             l = list(req.find_entities('CWUser', login=u'cdevienne'))
       
   107             self.assertEqual(1, len(l))
       
   108             self.assertEqual(l[0].firstname, u"Christophe")
       
   109 
       
   110             l = list(req.find_entities('CWUser', login=u'patanok'))
       
   111             self.assertEqual(0, len(l))
       
   112 
       
   113             l = list(req.find_entities('CWUser'))
       
   114             self.assertEqual(4, len(l))
       
   115 
       
   116     def test_find(self):
       
   117         with self.admin_access.web_request() as req:
       
   118             req.create_entity(
       
   119                 'CWUser', login=u'cdevienne', upassword=u'cdevienne',
       
   120                 surname=u'de Vienne', firstname=u'Christophe',
       
   121                 in_group=req.find('CWGroup', name=u'users').one())
       
   122 
       
   123             req.create_entity(
       
   124                 'CWUser', login=u'adim', upassword='adim', surname=u'di mascio',
       
   125                 firstname=u'adrien',
       
   126                 in_group=req.find('CWGroup', name=u'users').one())
       
   127 
       
   128             u = req.find('CWUser', login=u'cdevienne').one()
       
   129             self.assertEqual(u.firstname, u"Christophe")
       
   130 
       
   131             users = list(req.find('CWUser').entities())
       
   132             self.assertEqual(len(users), 4)
       
   133 
       
   134             groups = list(
       
   135                 req.find('CWGroup', reverse_in_group=u).entities())
       
   136             self.assertEqual(len(groups), 1)
       
   137             self.assertEqual(groups[0].name, u'users')
       
   138 
       
   139             users = req.find('CWUser', in_group=groups[0]).entities()
       
   140             users = list(users)
       
   141             self.assertEqual(len(users), 2)
       
   142 
       
   143             with self.assertRaises(AssertionError):
       
   144                 req.find('CWUser', chapeau=u"melon")
       
   145 
       
   146             with self.assertRaises(AssertionError):
       
   147                 req.find('CWUser', reverse_buddy=users[0])
       
   148 
       
   149             with self.assertRaises(NotImplementedError):
       
   150                 req.find('CWUser', in_group=[1, 2])
       
   151 
       
   152 if __name__ == '__main__':
       
   153     unittest_main()