def setup(self):
super(TestUser, self).setup()
self.cli = self.app.test_client()
self.user = User('test', '*****@*****.**', 'password')
db.session.add(self.user)
db.session.commit()
def setup(self):
super(TestMain, self).setup()
self.cli = self.app.test_client()
self.user = User('test', '*****@*****.**', 'password')
self.consumer = Consumer('annotateit')
self.consumer.secret = 'secret'
db.session.add(self.user)
db.session.add(self.consumer)
db.session.commit()
def _add_user(form):
user = User(username=form.username.data,
email=form.email.data,
password=form.password.data)
db.session.add(user)
try:
db.session.commit()
except sqlalchemy.exc.IntegrityError as e:
if 'email is not unique' in e.message:
form.email.errors.append("This email address is already registered: please use another.")
if 'username is not unique' in e.message:
form.username.errors.append("This username is taken: please use another.")
return False
# Fallthrough: all's gone well.
return True
def test_constructor(self):
u = User('joe', '*****@*****.**')
h.assert_equal(u.username, 'joe')
h.assert_equal(u.email, '*****@*****.**')
def test_gravatar_url(self):
u = User('joe', '*****@*****.**')
h.assert_equal(u.gravatar_url, 'http://www.gravatar.com/avatar/011b2f33289a5f9941e457bdd1e549ff?d=mm')
def test_null_password(self):
u = User('joe', '*****@*****.**')
h.assert_false(u.check_password('foo'))
def test_password(self):
u = User('joe', '*****@*****.**')
u.password = '******'
h.assert_is_not_none(u.password_hash)
h.assert_true(u.check_password('foo'))
print("Migrating database... ")
migrate.upgrade(**migrate_args)
print("done.")
ckey = os.environ.get('CONSUMER_KEY', 'annotateit')
csecret = os.environ.get('CONSUMER_SECRET', 'annotate.it.secret')
with app.test_request_context():
users_count = User.query.count()
print("Users in DB: " + str(users_count))
if users_count == 0:
print("Creating admin user... ")
u = User(username, email, password)
u.is_admin = True
db.session.add(u)
db.session.commit()
print("done.\n")
print("Creating primary consumer... ")
c = Consumer(ckey)
c.user_id = u.id
c.secret = csecret
db.session.add(c)
db.session.commit()
