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)
