def tearDown(self):
ids = []
for taxi in Taxi.query.all():
redis_store.delete('taxi:{}'.format(taxi.id))
ids.append(taxi.id)
for i in ids:
redis_store.zrem('geoindex', i)
db.session.remove()
db.drop_all()
db.get_engine(self.app).dispose()
for r in regions.values():
r.invalidate()
index_zupc.index_zupc = None
def setUp(self):
db.drop_all()
db.create_all()
regions['taxis'].invalidate()
regions['hails'].invalidate()
regions['users'].invalidate()
for role in ['admin', 'operateur', 'moteur']:
r = user_datastore.create_role(name=role)
u = user_datastore.create_user(email='user_'+role,
password=role)
user_datastore.add_role_to_user(u, r)
u = user_datastore.create_user(email='user_'+role+'_2',
password=role)
user_datastore.add_role_to_user(u, r)
db.session.commit()
r = user_datastore.find_role('operateur')
u = user_datastore.create_user(email='user_apikey',
password='******')
u.operator_header_name = 'X-API-KEY'
u.operator_api_key = 'xxx'
user_datastore.add_role_to_user(u, r)
db.session.commit()
