def test_user_follow_duplicate(self):
bob = User(username="******", email="*****@*****.**")
bob.set_password("hunter2")
bob.save()
bob.follow(self.user)
with self.assertRaises(Exception):
bob.follow(self.user)
def test_user_is_following(self):
bob = User(username="******", email="*****@*****.**")
bob.set_password("hunter2")
bob.save()
self.assertFalse(bob.is_following(self.user))
bob.follow(self.user)
self.assertTrue(bob.is_following(self.user))
def test_user_unfollow_error(self):
bob = User(username="******", email="*****@*****.**")
bob.set_password("hunter2")
bob.save()
with self.assertRaises(Exception):
bob.unfollow(self.user)
def test_redirect_logged_in_post(self):
user = User(username='******', email='*****@*****.**')
user.set_password('password')
db.session.add(user)
db.session.commit()
self.loginForm2 = dict(username='******',
email='*****@*****.**',
password='******')
with self.client:
self.client.post('/login', data=self.loginForm)
result = self.client.post('/login', data=self.loginForm2)
self.assertRedirects(result, url_for('index'))
self.assertEquals(current_user.username,
self.loginForm['username'])
def test_user_follow(self):
bob = User(username="******", email="*****@*****.**")
bob.set_password("hunter2")
bob.save()
bob.follow(self.user)
self.assertTrue(
bob.followed.filter(
followers.c.followed_id == self.user.id).count() > 0)
self.assertTrue(bob.followed.all()[0].username == "user")
class BaseCase(TestCase):
def create_app(self):
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite://'
app.config['TESTING'] = True
app.config['WTF_CSRF_ENABLED'] = False
app.config['UPLOAD_FOLDER'] = "./tests/uploads/"
return app
def setUp(self):
db.create_all()
self.client = app.test_client()
# Register data
self.registerForm = dict(username='******',
email='*****@*****.**',
password='******',
password2='pass')
# Existing user
self.user = User(username='******', email='*****@*****.**')
self.user.set_password('password')
db.session.add(self.user)
db.session.commit()
self.loginForm = dict(username='******',
email='*****@*****.**',
password='******')
# Create upload folder
if not os.path.exists('./tests/uploads/'): #pragma: no cover
os.makedirs('./tests/uploads/')
def tearDown(self):
db.session.remove()
db.drop_all()
# Delete files in upload folder
folder = './tests/uploads/'
for the_file in os.listdir(folder):
file_path = os.path.join(folder, the_file)
try:
if os.path.isfile(file_path): #pragma: no cover
os.unlink(file_path)
#elif os.path.isdir(file_path): shutil.rmtree(file_path)
except Exception as e: #pragma: no cover
print(e)
def setUp(self):
db.create_all()
self.client = app.test_client()
# Register data
self.registerForm = dict(username='******',
email='*****@*****.**',
password='******',
password2='pass')
# Existing user
self.user = User(username='******', email='*****@*****.**')
self.user.set_password('password')
db.session.add(self.user)
db.session.commit()
self.loginForm = dict(username='******',
email='*****@*****.**',
password='******')
# Create upload folder
if not os.path.exists('./tests/uploads/'): #pragma: no cover
os.makedirs('./tests/uploads/')
def test_user_save(self):
user = User(username="******", email="*****@*****.**")
user.set_password("hunter2")
self.assertFalse(user_exists("bob"))
user.save()
self.assertTrue(user_exists("bob"))
def register():
if current_user.is_authenticated:
return redirect(url_for('index'))
form = RegistrationForm()
if form.validate_on_submit():
user = User(username=form.username.data, email=form.email.data)
user.set_password(form.password.data)
user.save()
flash('Congratulations, you are now a registered user!')
return redirect(url_for('login'))
return render_template('register.html', form=form)
def test_user_get_following(self):
bob = User(username="******", email="*****@*****.**")
bob.set_password("hunter2")
bob.save()
user2 = User(username="******", email="*****@*****.**")
user2.set_password("hunter2")
user2.save()
bob.follow(self.user)
bob.follow(user2)
self.assertTrue(bob.get_following() == [self.user, user2])
def test_password_hash(self):
user = User(username="******", email="*****@*****.**")
user.set_password("password")
self.assertFalse(user.check_password("hunter12"))
self.assertTrue(user.check_password("password"))