# copyright 2003-2012 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/>."""Fake objects to ease testing of cubicweb without a fully working environment"""__docformat__="restructuredtext en"fromlogilab.databaseimportget_db_helperfromcubicweb.reqimportRequestSessionBasefromcubicweb.cwvregimportCWRegistryStorefromcubicweb.web.requestimportCubicWebRequestBasefromcubicweb.devtoolsimportBASE_URL,BaseApptestConfigurationclassFakeConfig(dict,BaseApptestConfiguration):translations={}uiprops={}https_uiprops={}apphome=Nonedebugmode=Falsedef__init__(self,appid='data',apphome=None,cubes=()):self.appid=appidself.apphome=apphomeself._cubes=cubesself['auth-mode']='cookie'self['uid']=Noneself['base-url']=BASE_URLself['rql-cache-size']=3000self.datadir_url=BASE_URL+'data/'self.https_datadir_url=(BASE_URL+'data/').replace('http://','https://')defcubes(self,expand=False):returnself._cubesdefsources(self):return{'system':{'db-driver':'sqlite'}}classFakeRequest(CubicWebRequestBase):"""test implementation of an cubicweb request object"""def__init__(self,*args,**kwargs):ifnot(argsor'vreg'inkwargs):kwargs['vreg']=CWRegistryStore(FakeConfig(),initlog=False)kwargs['https']=Falseself._http_method=kwargs.pop('method','GET')self._url=kwargs.pop('url',None)or'view?rql=Blop&vid=blop'super(FakeRequest,self).__init__(*args,**kwargs)self._session_data={}defset_cookie(self,name,value,maxage=300,expires=None,secure=False):super(FakeRequest,self).set_cookie(name,value,maxage,expires,secure)cookie=self.get_response_header('Set-Cookie')self._headers_in.setHeader('Cookie',cookie)## Implement request abstract APIdefheader_accept_language(self):"""returns an ordered list of preferred languages"""return('en',)defhttp_method(self):returnself._http_methoddefrelative_path(self,includeparams=True):"""return the normalized path of the request (ie at least relative to the instance's root, but some other normalization may be needed so that the returned path may be used to compare to generated urls """ifself._url.startswith(BASE_URL):url=self._url[len(BASE_URL):]else:url=self._urlifincludeparams:returnurlreturnurl.split('?',1)[0]defset_request_header(self,header,value,raw=False):"""set an incoming HTTP header (For test purpose only)"""ifisinstance(value,basestring):value=[value]ifraw:## adding encoded header is important, else page content# will be reconverted back to unicode and apart unefficiency, this# may cause decoding problem (e.g. when downloading a file)self._headers_in.setRawHeaders(header,value)else:#self._headers_in.setHeader(header,value)#defget_response_header(self,header,default=None,raw=False):"""return output header (For test purpose only"""ifraw:#returnself.headers_out.getRawHeaders(header,[default])[0]returnself.headers_out.getHeader(header,default)defbuild_url_params(self,**kwargs):# overriden to get predictable resulttsargs=[]forparam,valuesinsorted(kwargs.iteritems()):ifnotisinstance(values,(list,tuple)):values=(values,)forvalueinvalues:assertvalueisnotNoneargs.append(u'%s=%s'%(param,self.url_quote(value)))return'&'.join(args)classFakeUser(object):login='toto'eid=0defin_groups(self,groups):returnTrueclassFakeSession(RequestSessionBase):def__init__(self,repo=None,user=None,vreg=None):self.repo=repoifvregisNone:vreg=getattr(self.repo,'vreg',None)ifvregisNone:vreg=CWRegistryStore(FakeConfig(),initlog=False)self.vreg=vregself.cnxset=FakeConnectionsSet()self.user=userorFakeUser()self.is_internal_session=Falseself.transaction_data={}defexecute(self,*args,**kwargs):passdefcommit(self,*args):self.transaction_data.clear()defclose(self,*args):passdefsystem_sql(self,sql,args=None):passdefset_entity_cache(self,entity):passdefsecurity_enabled(self,read=False,write=False):classFakeCM(object):def__enter__(self):passdef__exit__(self,exctype,exc,traceback):passreturnFakeCM()# for use with enabled_security context managerread_security=write_security=TrueclassFakeRepo(object):querier=Nonedef__init__(self,schema,vreg=None,config=None):self.extids={}self.eids={}self._count=0self.schema=schemaself.config=configorFakeConfig()self.vreg=vregorCWRegistryStore(self.config,initlog=False)self.vreg.schema=schemaself.sources=[]definternal_session(self):returnFakeSession(self)defextid2eid(self,source,extid,etype,session,insert=True):try:returnself.extids[extid]exceptKeyError:ifnotinsert:returnNoneself._count+=1eid=self._countentity=source.before_entity_insertion(session,extid,etype,eid)self.extids[extid]=eidself.eids[eid]=extidsource.after_entity_insertion(session,extid,entity)returneiddefeid2extid(self,source,eid,session=None):returnself.eids[eid]classFakeSource(object):dbhelper=get_db_helper('sqlite')def__init__(self,uri):self.uri=uriclassFakeConnectionsSet(object):defsource(self,uri):returnFakeSource(uri)