def setUp(self):
db.create_all()
self.init_data = init_api_data(self.app)
register_restaurants_blueprints(self.app)
data = json.loads(
self.client.post('/auth',
data=json.dumps(dict(
username=self.init_data['USER_ADMIN_EMAIL'],
password=self.init_data['USER_ADMIN_PASSWORD'])
),
content_type='application/json').data.decode('utf-8'))
self.john_token = data['access_token']
data = json.loads(
self.client.post('/auth',
data=json.dumps(dict(
username=self.init_data['USER_COMMON_EMAIL'],
password=self.init_data['USER_COMMON_PASSWORD'])
),
content_type='application/json').data.decode('utf-8'))
self.susan_token = data['access_token']
data = json.loads(
self.client.post('/auth',
data=json.dumps(dict(
username=self.init_data['USER_MANAGER_EMAIL'],
password=self.init_data['USER_MANAGER_PASSWORD'])
),
content_type='application/json').data.decode('utf-8'))
self.karl_token = data['access_token']
def db(app):
_db.app = app
_db.create_all()
yield _db
# Code after yield executes on teardown
_db.session.close()
_db.drop_all()
def db(app):
url = app.config['SQLALCHEMY_DATABASE_URI']
if not sqlalchemy_utils.database_exists(url):
sqlalchemy_utils.create_database(url)
db_drop_everything(_db)
_db.create_all()
yield _db
_db.session.close()
db_drop_everything(_db)
def create_app(config='config.py', db=db):
app = Flask(__name__, instance_relative_config=True)
CORS(app)
if config:
app.config.from_pyfile(config)
else:
app.config['SQLALCHEMY_DATABASE_URI'] = os.environ.get('DATABASE_URL')
app.config['SQLALCHEMY_TRACK_MODIFICATIONS'] = False
app.secret_key = os.environ.get('SECRET_KEY')
from api import models
db.init_app(app)
db.app = app
db.create_all()
from api.conversation_controller import conversation_api
app.register_blueprint(conversation_api)
return app
def inituser():
db.create_all()
try:
group = Group(name='admin',
app_add=True,
app_edit_all=True,
app_edit_my=True,
app_drop=True,
user_add=True,
user_drop=False)
db.session.add(group)
except:
pass
try:
hash = sha256_crypt.encrypt('admin')
user = User(email='admin@intra', password_hash=hash, group_id=1)
db.session.add(user)
except:
pass
db.session.commit()
def setUp(self):
self.db_fd, self.db_filename = tempfile.mkstemp()
app.config['SQLALCHEMY_DATABASE_URI'] = 'postgresql://localhost/monkeystest'
app.config['TESTING'] = True
app.testing = True
app.config['WTF_CSRF_ENABLED'] = False
self.app = app.test_client()
connect_db(app)
if self.logged_in:
self.fake_login()
with app.app_context():
db.drop_all()
db.create_all()
load_student()
load_coach()
load_class_schedule()
load_class_instance()
load_student_class_instance()
load_student_class_schedule()
def create_app(pconfig=None, debug=False):
app = Flask(__name__)
app.debug = debug
configobj = pconfig or \
os.environ.get('PYSTRO_SETTINGS') or \
defaultconfig
app.config.from_object(configobj)
app.app_context().push()
with app.app_context():
db.init_app(app)
db.drop_all()
try:
db.create_all()
if not app.testing:
init_api_data(app)
except (OperationalError, ObjectDeletedError):
db.session.rollback()
db.session.remove()
setup_admin_api_routes(app)
setup_public_api_routes(app)
JWT(app, authenticate, identity)
register_restaurants_blueprints(app)
return app
def setup_database(app):
with app.app_context():
db.create_all()
# db_mock_data.mock_data()
db.session.commit()
new_item = ClassSchedule(name=item['name'],
day_of_week=item['day_of_week'],
time=item['time'])
db.session.add(new_item)
db.session.commit()
def load_class_instance():
"""Load class instance from seed_data into database."""
tablename = 'class_instance'
data = jsonify_seed_data(tablename)
for item in data[tablename]:
new_item = ClassInstance(class_schedule_id=item['class_schedule_id'],
date=item['date'])
db.session.add(new_item)
db.session.commit()
if __name__ == "__main__":
connect_db(app)
db.drop_all()
db.create_all()
load_student()
load_coach()
load_class_schedule()
load_class_instance()
load_student_class_instance()
load_student_class_schedule()
def setUp(self):
self.client = test_app.test_client()
db.create_all()
# email = "*****@*****.**"
# id = "some_shibboleth_id_of_the_user"
# return jsonify({
# "token": create_jwt(identity={ "id": id, "name": name, "email": email })
# })
# current user identity
api.add_resource(IdentityResource, "/users/me")
# deployments
api.add_resource(Deployments, "/deployments")
api.add_resource(Deployment, "/deployments/<int:deployment_id>")
api.add_resource(DeploymentLog, "/deployments/<int:deployment_id>/log")
api.add_resource(DeploymentUndeploy,
"/deployments/<int:deployment_id>/undeploy")
# templates
api.add_resource(Templates, "/templates")
api.add_resource(Template, "/templates/<int:template_id>")
db.init_app(app)
return app
if __name__ == "__main__":
app = get_app()
if not os.path.exists(const.DATABASE_URI):
db.create_all(app=app)
app.run(debug=True, threaded=True)
def setUp(self):
self.app_context = self.app.app_context()
self.app_context.push()
db.create_all()
def init_db(app):
db.drop_all()
db.create_all()
yield db
db.session.close()
db.drop_all()
