def setUp(self):
"""Set up the test database and test user."""
self.app = create_app(config_name="testing")
self.client = self.app.test_client()
self.app_context = self.app.app_context()
self.app_context.push()
db.create_all()
user = User(username="******",
email="*****@*****.**",
password="******")
bucketlist = Bucketlist(bucketlist_title="Visit Paris", creator_id=1)
item = Item(item_name="The Eiffel Tower",
description="Wrought iron lattice tower in France",
bucketlist_id=1)
db.session.add(user)
db.session.add(bucketlist)
db.session.add(item)
db.session.commit()
# set header
self.auth_header = {'Authorization': user.generate_auth_token(user.id)}
self.token = user.generate_auth_token(user.id)
"""manage.py."""
import os
from flask_script import Manager, prompt_bool
from flask_migrate import Migrate, MigrateCommand
from bucketlist import db, create_app
# from app import models
app = create_app(config_name=os.getenv('APP_SETTINGS'))
migrate = Migrate(app, db)
manager = Manager(app)
manager.add_command('db', MigrateCommand)
# manually create_db
@manager.command
def create_db():
"""Create db from models."""
db.create_all()
print ('Database created!')
# manually drop db
@manager.command
def drop_db():
"""Drop db tables."""
if prompt_bool("Are you sure you want to drop the database?"):
db.drop_all()
print ('Database has been dropped!')
from flask_script import Manager
from bucketlist import create_app
from flask_migrate import MigrateCommand
app = create_app(os.getenv('APP_CONFIG'))
manager = Manager(app)
manager.add_command('db', MigrateCommand)
if __name__ == '__main__':
manager.run()
import os
from bucketlist import create_app
config_name = os.getenv('APP_SETTINGS') # config_name = "development"
app = create_app(config_name)
if __name__ == '__main__':
app.run()
def setUpClass(self):
self.app = create_app("test_config.py")
init_db()
self.client = bucketlist.app.test_client()
import unittest, datetime
from bucketlist import create_app
import json
app = create_app(True)
from bucketlist.views import *
class TestUserModel(unittest.TestCase):
def setUp(self):
first_name = "John"
last_name = 'Doe'
username = '******'
email = '*****@*****.**'
password = '******'
with app.app_context():
db.create_all()
user = User.query.filter_by(email = email).first()
if user == None:
user = User(first_name, last_name, username, email, password)
db.session.add(user)
db.session.commit()
self.user = User.query.filter_by(email=user.email).first()
else:
self.user = user
import unittest, datetime
from bucketlist import create_app
import json, os
app = create_app(testing = True)
from bucketlist.views import *
with app.app_context():
db.drop_all()
class TestEndpoints(unittest.TestCase):
def setUp(self):
self.client = app.test_client
self.user_data = {
"first_name": "John",
"last_name": "Doe",
"username": "******",
"email": "*****@*****.**",
"password": "******"
}
with app.app_context():
db.create_all()
@classmethod
def tearDownClass(cls):
with app.app_context():
db.session.expunge_all()
db.session.close()
db.drop_all()
from flask_script import Manager
from bucketlist import create_app
app = create_app()
manager = Manager(app)
manager.add_command('db', MigrateCommand)
if __name__ == '__main__':
manager.run()