diff -r 672acc730ce5 -r d628defebc17 goa/testlib.py --- a/goa/testlib.py Thu May 14 10:24:56 2009 +0200 +++ b/goa/testlib.py Thu May 14 11:38:40 2009 +0200 @@ -1,3 +1,10 @@ +""" +:organization: Logilab +:copyright: 2008-2009 LOGILAB S.A. (Paris, FRANCE), all rights reserved. +:contact: http://www.logilab.fr/ -- mailto:contact@logilab.fr +""" +__docformat__ = "restructuredtext en" + from logilab.common.testlib import TestCase import os, os.path as osp @@ -18,7 +25,7 @@ import_appengine_failed = None except ImportError, exc: # XXX necessary ? - class db: + class db: class Model: pass class DummyProperty: @@ -31,7 +38,7 @@ ReferenceProperty = DummyProperty SelfReferenceProperty = DummyProperty import_appengine_failed = 'cannot import appengine: %s' % exc - + from cubicweb.devtools.fake import FakeRequest from cubicweb.goa.goavreg import GAERegistry @@ -56,11 +63,11 @@ def load_schema_hook(self, loader): loader.import_yams_cube_schema('data') - + @property def DS_FILE(self): return self.DS_TEMPL_FILE.replace('-template', '') - + @property def DS_TEMPL_FILE(self): return self._DS_TEMPL_FILE + '_'.join(sorted(cls.__name__ for cls in self.MODEL_CLASSES)) @@ -72,7 +79,7 @@ stub = datastore_file_stub.DatastoreFileStub(self.APP_ID, dsfile, dsfile+'.history') apiproxy_stub_map.apiproxy.RegisterStub('datastore_v3', stub) - + def setUp(self): if import_appengine_failed: self.skip(import_appengine_failed) @@ -88,7 +95,7 @@ self._set_ds_file(self.DS_TEMPL_FILE) # from google.appengine.api import mail_stub # from google3.apphosting.api import urlfetch_stub -# from google3.apphosting.api import user_service_stub +# from google3.apphosting.api import user_service_stub # # Use a fresh stub UserService. # apiproxy_stub_map.apiproxy.RegisterStub( # 'user', user_service_stub.UserServiceStub()) @@ -149,15 +156,15 @@ req = FakeRequest(vreg=self.vreg) self.session = self.session_manager.open_session(req) self.user = self.session.user() - + def tearDown(self): self.session.close() - + def request(self): req = FakeRequest(vreg=self.vreg) req.set_connection(self.session, self.user) return req - + def add_entity(self, etype, **kwargs): cu = self.session.cursor() rql = 'INSERT %s X' % etype @@ -174,7 +181,7 @@ def rollback(self): self.session.rollback() - + def create_user(self, login, groups=('users',), req=None): assert not self.config['use-google-auth'] user = self.add_entity('CWUser', upassword=str(login), login=unicode(login)) @@ -190,4 +197,3 @@ req.form['__login'] = login req.form['__password'] = password or login return self.session_manager.open_session(req) -