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()