from rhodecode.tests import * from rhodecode.model.db import User from rhodecode.lib.auth import check_password class TestLoginController(TestController): def test_index(self): response = self.app.get(url(controller='login', action='index')) assert response.status == '200 OK', 'Wrong response from login page got %s' % response.status # Test response... def test_login_admin_ok(self): response = self.app.post(url(controller='login', action='index'), {'username':'test_admin', 'password':'test12'}) assert response.status == '302 Found', 'Wrong response code from login got %s' % response.status assert response.session['rhodecode_user'].username == 'test_admin', 'wrong logged in user' response = response.follow() assert 'vcs_test repository' in response.body def test_login_regular_ok(self): response = self.app.post(url(controller='login', action='index'), {'username':'test_regular', 'password':'test12'}) print response assert response.status == '302 Found', 'Wrong response code from login got %s' % response.status assert response.session['rhodecode_user'].username == 'test_regular', 'wrong logged in user' response = response.follow() assert 'vcs_test repository' in response.body assert '' not in response.body def test_login_ok_came_from(self): test_came_from = '/_admin/users' response = self.app.post(url(controller='login', action='index', came_from=test_came_from), {'username':'test_admin', 'password':'test12'}) assert response.status == '302 Found', 'Wrong response code from came from redirection' response = response.follow() assert response.status == '200 OK', 'Wrong response from login page got %s' % response.status assert 'Users administration' in response.body, 'No proper title in response' def test_login_short_password(self): response = self.app.post(url(controller='login', action='index'), {'username':'error', 'password':'test'}) assert response.status == '200 OK', 'Wrong response from login page' print response.body assert 'Enter 6 characters or more' in response.body, 'No error password message in response' def test_login_wrong_username_password(self): response = self.app.post(url(controller='login', action='index'), {'username':'error', 'password':'test12'}) assert response.status == '200 OK', 'Wrong response from login page' assert 'invalid user name' in response.body, 'No error username message in response' assert 'invalid password' in response.body, 'No error password message in response' def test_register(self): response = self.app.get(url(controller='login', action='register')) assert 'Sign Up to rhodecode' in response.body, 'wrong page for user registration' def test_register_err_same_username(self): response = self.app.post(url(controller='login', action='register'), {'username':'test_admin', 'password':'test', 'email':'goodmail@domain.com', 'name':'test', 'lastname':'test'}) assert response.status == '200 OK', 'Wrong response from register page got %s' % response.status assert 'This username already exists' in response.body def test_register_err_wrong_data(self): response = self.app.post(url(controller='login', action='register'), {'username':'xs', 'password':'', 'email':'goodmailm', 'name':'test', 'lastname':'test'}) assert response.status == '200 OK', 'Wrong response from register page got %s' % response.status assert 'An email address must contain a single @' in response.body assert 'Please enter a value' in response.body def test_register_ok(self): username = 'test_regular4' password = 'qweqwe' email = 'marcin@test.com' name = 'testname' lastname = 'testlastname' response = self.app.post(url(controller='login', action='register'), {'username':username, 'password':password, 'email':email, 'name':name, 'lastname':lastname}) print response.body assert response.status == '302 Found', 'Wrong response from register page got %s' % response.status assert 'You have successfully registered into rhodecode' in response.session['flash'][0], 'No flash message about user registration' ret = self.sa.query(User).filter(User.username == 'test_regular4').one() assert ret.username == username , 'field mismatch %s %s' % (ret.username, username) assert check_password(password, ret.password) == True , 'password mismatch' assert ret.email == email , 'field mismatch %s %s' % (ret.email, email) assert ret.name == name , 'field mismatch %s %s' % (ret.name, name) assert ret.lastname == lastname , 'field mismatch %s %s' % (ret.lastname, lastname) def test_forgot_password_wrong_mail(self): response = self.app.post(url(controller='login', action='password_reset'), {'email':'marcin@wrongmail.org', }) assert "That e-mail address doesn't exist" in response.body, 'Missing error message about wrong email' def test_forgot_password(self): response = self.app.get(url(controller='login', action='password_reset')) assert response.status == '200 OK', 'Wrong response from login page got %s' % response.status username = 'test_password_reset_1' password = 'qweqwe' email = 'marcin@python-works.com' name = 'passwd' lastname = 'reset' response = self.app.post(url(controller='login', action='register'), {'username':username, 'password':password, 'email':email, 'name':name, 'lastname':lastname}) #register new user for email test response = self.app.post(url(controller='login', action='password_reset'), {'email':email, }) print response.session['flash'] assert 'You have successfully registered into rhodecode' in response.session['flash'][0], 'No flash message about user registration' assert 'Your new password was sent' in response.session['flash'][1], 'No flash message about password reset'