def initdb():
"""Init/reset database."""
try:
db.drop_all()
except sqlalchemy.exc.OperationalError:
URI = app.config[
'SQLALCHEMY_DATABASE_URI'][:app.config['SQLALCHEMY_DATABASE_URI'].
rfind('/')]
engine = sqlalchemy.create_engine(URI)
engine.execute("CREATE DATABASE fbone")
db.create_all()
admin = User(name=u'admin',
fullname=u'Agador Spartacus',
email=u'*****@*****.**',
password=u'123456',
role_code=ADMIN,
status_code=ACTIVE,
gender_code=MALE,
bio=u'FSU Grad. Go Noles!')
db.session.add(admin)
db.session.commit()
def signup(self):
user = User()
self.populate_obj(user)
db.session.add(user)
db.session.commit()
return user
def create_profile(self):
user = User()
self.populate_obj(user)
db.session.add(user)
db.session.commit()
def test_roles(self):
u = User(name='qux', email='*****@*****.**', fullname='qux')
User().save(u)
assert u.role_code == USER
def test_password_is_nullable(self):
user = User(name='zap', email='*****@*****.**')
assert user.password is None
def test_get_by_id(self):
user = User(name='bar', email='*****@*****.**', fullname='bar')
User().save(user)
retrieved = User().get_by_id(user.id)
assert retrieved == user
