# copyright 2013-2014 LOGILAB S.A. (Paris, FRANCE), all rights reserved.# contact http://www.logilab.fr/ -- mailto:contact@logilab.fr## This file is part of CubicWeb.## CubicWeb is free software: you can redistribute it and/or modify it under the# terms of the GNU Lesser General Public License as published by the Free# Software Foundation, either version 2.1 of the License, or (at your option)# any later version.## CubicWeb is distributed in the hope that it will be useful, but WITHOUT# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS# FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more# details.## You should have received a copy of the GNU Lesser General Public License along# with CubicWeb. If not, see <http://www.gnu.org/licenses/>."""unittest for cubicweb.repoapi"""fromcubicweb.devtools.testlibimportCubicWebTCfromcubicwebimportProgrammingErrorfromcubicweb.repoapiimportConnection,connect,anonymous_cnxclassREPOAPITC(CubicWebTC):deftest_cnx_basic_usage(self):"""Test that a client connection can be used to access the database"""withself.admin_access.client_cnx()ascltcnx:# (1) some RQL requestrset=cltcnx.execute('Any X WHERE X is CWUser')self.assertTrue(rset)# (2) ORM usagerandom_user=rset.get_entity(0,0)# (3) Write operationrandom_user.cw_set(surname=u'babar')# (4) commitcltcnx.commit()rset=cltcnx.execute('''Any X WHERE X is CWUser, X surname "babar" ''')self.assertTrue(rset)# prepare test for implicit rollbackrandom_user=rset.get_entity(0,0)random_user.cw_set(surname=u'celestine')# implicit rollback on exitwithself.admin_access.client_cnx()ascltcnx:rset=cltcnx.execute('''Any X WHERE X is CWUser, X surname "babar" ''')self.assertTrue(rset)deftest_cnx_life_cycle(self):"""Check that ClientConnection requires explicit open and close """access=self.admin_accesscltcnx=Connection(access._session)# connection not open yetwithself.assertRaises(ProgrammingError):cltcnx.execute('Any X WHERE X is CWUser')# connection open and workingwithcltcnx:cltcnx.execute('Any X WHERE X is CWUser')# connection closedwithself.assertRaises(ProgrammingError):cltcnx.execute('Any X WHERE X is CWUser')deftest_connect(self):"""check that repoapi.connect works and returns a usable connection"""cnx=connect(self.repo,login='admin',password='gingkow')self.assertEqual('admin',cnx.user.login)withcnx:rset=cnx.execute('Any X WHERE X is CWUser')self.assertTrue(rset)deftest_anonymous_connect(self):"""check that you can get anonymous connection when the data exist"""cnx=anonymous_cnx(self.repo)self.assertEqual('anon',cnx.user.login)withcnx:rset=cnx.execute('Any X WHERE X is CWUser')self.assertTrue(rset)if__name__=='__main__':fromlogilab.common.testlibimportunittest_mainunittest_main()