def client():
test_db, app.config['DATABASE'] = tempfile.mkstemp()
app.config['TESTING'] = True
client = app.test_client()
with app.app_context():
db.init_app(app)
db.create_all()
yield client
with app.app_context():
db.drop_all()
os.close(test_db)
os.unlink(app.config['DATABASE'])
def setUp(self):
self.app = self.create_app().test_client()
db.drop_all()
db.create_all()
res = self.app.post("/api/create_user",
data=json.dumps(self.default_user),
content_type='application/json')
self.token = json.loads(res.data.decode("utf-8"))["token"]
test_program = ProgramTemplate(name="test_program",
program=json.dumps({
"weights": {
"pre":
"0.9*X",
"session": ["X+0"],
"set":
[["0.65*X", "0.75*X", "0.85*X"],
["0.7*X", "0.8*X", "0.9*X"],
["0.75*X", "0.85*X", "0.95*X"],
["0.4*X", "0.5*X", "0.6*X"]],
"rep": [["X+0"]]
},
"times": {
"session": [],
"set": [[180, 150]],
"rep": [[[10, 6, 2], [5, 3, 1]],
[[6, 2], [3, 1]],
[[2], [1]]]
},
"freqs": {
"session": [2, 2, 2, 1],
"set": [3],
"rep": [3, 2, 2]
}
}))
db.session.add(test_program)
def tearDown(self):
db.session.remove()
db.drop_all()
from application.app import app, db
from application.models import User
if __name__ == '__main__':
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('-c',
'--create',
action='store_true',
help='Create all tables')
parser.add_argument('-d',
'--drop_all',
action='store_true',
help='Drop all tables')
args = parser.parse_args()
with app.app_context():
if args.drop_all:
db.drop_all()
if args.create:
db.create_all()
# add dummy users
User.create_user('user1', '*****@*****.**')
User.create_user('user2', '*****@*****.**')
def delete_db():
"""Deletes the db tables."""
db.drop_all()
def tearDown(self):
db.session.remove()
db.drop_all()
def drop_db():
"""Use when rebuilding DB."""
db.drop_all()
def drop_tables():
"Drop all project relational database tables. THIS DELETES ALL DATA."
if prompt_bool("Are you sure you want to lose all your data:"):
db.drop_all()
print("Drop database tables successfully.")
print("Drop database tables cancelled.")
