def setUp(self): engine = create_engine('sqlite:///:memory:', echo=True) Model.metadata.create_all(engine) Session.configure(bind=engine) import app # Must be after Session is configured app.app.config['TESTING'] = True self.app = app.app.test_client()
if 'MEMCACHEDCLOUD_SERVERS' in os.environ and 'OPENID_AUTH_IDENTITY' in os.environ: use_openid = True app.cache = bmemcached.Client( os.environ['MEMCACHEDCLOUD_SERVERS'].split(','), os.environ.get('MEMCACHEDCLOUD_USERNAME'), os.environ.get('MEMCACHEDCLOUD_PASSWORD') ) app.session_interface = memcache_session.Session() openid_store = MemoryStore() openid_identity = os.environ["OPENID_AUTH_IDENTITY"] TOKEN = os.environ.get('AUTH_TOKEN') if not is_database_bound(): # Don't mess with the database connection if someone else set it up already Session.configure(bind=create_engine(os.environ['DATABASE_URL'])) def check_auth_token(fn): @wraps(fn) def _fn(*args, **kwargs): token = request.headers.get('X-Auth-Token', request.args.get('auth')) if TOKEN == token: return fn(*args, **kwargs) else: abort(403) return _fn def check_openid_credentials(fn): if not use_openid:
def setUp(self): engine = create_engine("sqlite:///:memory:", echo=True) Model.metadata.create_all(engine) Session.configure(bind=engine) event.Event.for_name("Drink glass of water", create=True)