def setup(self): try: new_attrs = {} new_attrs.update(self.attrs) new_attrs.update(self.do_get_dependencies()) self.obj = self.klass(**new_attrs) DBSession.add(self.obj) DBSession.flush() return self.obj except: DBSession.rollback() raise
def add_test_data(options): """ Load a few test points. @@This ought to have a 'from file' flag and data/config driven. """ from loafapp.model import DBSession, Spot from geoalchemy import WKTSpatialElement from loafapp.config.environment import load_environment import transaction eng = sqla.create_engine(get_app_dbstr()) DBSession.configure(bind=eng) wkt = "POINT(-80.3 38.2, -81.03 38.04)" spot1 = Spot(name="My Bike Route", width=6, geom=WKTSpatialElement(wkt)) wkt = "POINT(-79.8 38.5, -80.03 38.2)" spot2 = Spot(name="George Ave", width=8, geom=WKTSpatialElement(wkt)) DBSession.add_all([spot1, spot2]) DBSession.flush() transaction.commit()
def by_user_name(cls, username): """Return the user object whose user name is ``username``.""" return DBSession.query(cls).filter(cls.user_name==username).first()
def by_email_address(cls, email): """Return the user object whose email address is ``email``.""" return DBSession.query(cls).filter(cls.email_address==email).first()
def test_query_obj(self): obj = DBSession.query(self.klass).one() for key, value in self.attrs.iteritems(): assert_equals(getattr(obj, key), value)
def tearDown(self): DBSession.rollback()