Views get forms from _security

This commit is contained in:
Eskil Heyn Olsen
2013-01-06 20:20:06 -08:00
parent 09fe5a2cb7
commit 81040a57a6
3 changed files with 37 additions and 51 deletions
+12
View File
@@ -6,6 +6,10 @@ import simplejson as json
from flask.ext.security.utils import capture_registrations, \
capture_reset_password_requests, capture_passwordless_login_requests
from flask.ext.security.forms import LoginForm, ConfirmRegisterForm, RegisterForm, \
ForgotPasswordForm, ResetPasswordForm, SendConfirmationForm, \
PasswordlessLoginForm
from tests import SecurityTest
@@ -488,13 +492,21 @@ class NoBlueprintTests(SecurityTest):
class ExtendFormsTest(SecurityTest):
class MyLoginForm(LoginForm):
hidden_tag = 'not-so-secret'
APP_KWARGS = {
'login_form': MyLoginForm,
}
AUTH_CONFIG = {
'SECURITY_REGISTERABLE': True,
}
def test_login_view(self):
r = self._get('/login')
self.assertIn("<h1>Login</h1>", r.data)
def test_register(self):
r = self._get('/register')
self.assertIn('<h1>Register</h1>', r.data)