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 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.")