mirror of
https://github.com/wassname/flask-security.git
synced 2026-09-12 12:21:37 +08:00
Stricter tests for signals and a small docs update. Fixes #308
This commit is contained in:
@@ -8,6 +8,8 @@
|
||||
|
||||
import pytest
|
||||
|
||||
from flask import Flask
|
||||
from flask_security.core import UserMixin
|
||||
from flask_security.signals import password_changed
|
||||
|
||||
from utils import authenticate
|
||||
@@ -20,6 +22,8 @@ def test_recoverable_flag(app, client, get_message):
|
||||
|
||||
@password_changed.connect_via(app)
|
||||
def on_password_changed(app, user):
|
||||
assert isinstance(app, Flask)
|
||||
assert isinstance(user, UserMixin)
|
||||
recorded.append(user)
|
||||
|
||||
authenticate(client)
|
||||
|
||||
@@ -10,6 +10,8 @@ import time
|
||||
|
||||
import pytest
|
||||
|
||||
from flask import Flask
|
||||
from flask_security.core import UserMixin
|
||||
from flask_security.signals import user_confirmed, confirm_instructions_sent
|
||||
from flask_security.utils import capture_registrations
|
||||
|
||||
@@ -25,10 +27,14 @@ def test_confirmable_flag(app, client, sqlalchemy_datastore, get_message):
|
||||
|
||||
@user_confirmed.connect_via(app)
|
||||
def on_confirmed(app, user):
|
||||
assert isinstance(app, Flask)
|
||||
assert isinstance(user, UserMixin)
|
||||
recorded_confirms.append(user)
|
||||
|
||||
@confirm_instructions_sent.connect_via(app)
|
||||
def on_instructions_sent(app, user):
|
||||
assert isinstance(app, Flask)
|
||||
assert isinstance(user, UserMixin)
|
||||
recorded_instructions_sent.append(user)
|
||||
|
||||
# Test login before confirmation
|
||||
|
||||
@@ -10,8 +10,10 @@ import time
|
||||
|
||||
import pytest
|
||||
|
||||
from flask import Flask
|
||||
from flask_security.core import UserMixin
|
||||
from flask_security.signals import login_instructions_sent
|
||||
from flask_security.utils import capture_passwordless_login_requests
|
||||
from flask_security.utils import capture_passwordless_login_requests, string_types
|
||||
|
||||
from utils import logout
|
||||
|
||||
@@ -23,6 +25,9 @@ def test_trackable_flag(app, client, get_message):
|
||||
|
||||
@login_instructions_sent.connect_via(app)
|
||||
def on_instructions_sent(app, user, login_token):
|
||||
assert isinstance(app, Flask)
|
||||
assert isinstance(user, UserMixin)
|
||||
assert isinstance(login_token, string_types)
|
||||
recorded.append(user)
|
||||
|
||||
# Test disabled account
|
||||
|
||||
@@ -10,8 +10,10 @@ import time
|
||||
|
||||
import pytest
|
||||
|
||||
from flask import Flask
|
||||
from flask_security.core import UserMixin
|
||||
from flask_security.signals import reset_password_instructions_sent, password_reset
|
||||
from flask_security.utils import capture_reset_password_requests
|
||||
from flask_security.utils import capture_reset_password_requests, string_types
|
||||
|
||||
from utils import authenticate, logout
|
||||
|
||||
@@ -28,6 +30,9 @@ def test_recoverable_flag(app, client, get_message):
|
||||
|
||||
@reset_password_instructions_sent.connect_via(app)
|
||||
def on_instructions_sent(app, user, token):
|
||||
assert isinstance(app, Flask)
|
||||
assert isinstance(user, UserMixin)
|
||||
assert isinstance(token, string_types)
|
||||
recorded_instructions_sent.append(user)
|
||||
|
||||
# Test the reset view
|
||||
|
||||
@@ -8,6 +8,8 @@
|
||||
|
||||
import pytest
|
||||
|
||||
from flask import Flask
|
||||
from flask_security.core import UserMixin
|
||||
from flask_security.signals import user_registered
|
||||
|
||||
from utils import authenticate, logout
|
||||
@@ -26,6 +28,9 @@ def test_registerable_flag(client, app, get_message):
|
||||
# Test registering is successful, sends email, and fires signal
|
||||
@user_registered.connect_via(app)
|
||||
def on_user_registerd(app, user, confirm_token):
|
||||
assert isinstance(app, Flask)
|
||||
assert isinstance(user, UserMixin)
|
||||
assert confirm_token is None
|
||||
recorded.append(user)
|
||||
|
||||
data = dict(
|
||||
|
||||
Reference in New Issue
Block a user