mirror of
https://github.com/wassname/flask-security.git
synced 2026-06-27 16:10:11 +08:00
Matt Wright
47 lines
948 B
Python
47 lines
948 B
Python
# -*- coding: utf-8 -*-
|
|
"""
|
|
test_entities
|
|
~~~~~~~~~~~~~
|
|
|
|
Entity tests
|
|
"""
|
|
|
|
from flask_security import RoleMixin, UserMixin, AnonymousUser
|
|
|
|
|
|
class Role(RoleMixin):
|
|
def __init__(self, name):
|
|
self.name = name
|
|
|
|
|
|
class User(UserMixin):
|
|
def __init__(self, roles):
|
|
self.roles = roles
|
|
|
|
|
|
def test_role_mixin_equal():
|
|
admin1 = Role('admin')
|
|
admin2 = Role('admin')
|
|
assert admin1 == admin2
|
|
|
|
|
|
def test_role_mixin_not_equal():
|
|
admin = Role('admin')
|
|
editor = Role('editor')
|
|
assert admin != editor
|
|
|
|
|
|
def test_user_mixin_has_role_with_string():
|
|
admin = Role('admin')
|
|
editor = Role('editor')
|
|
user = User([admin, editor])
|
|
assert user.has_role('admin') is True
|
|
assert user.has_role('editor') is True
|
|
assert user.has_role(admin) is True
|
|
assert user.has_role(editor) is True
|
|
|
|
|
|
def test_anonymous_user_has_no_roles():
|
|
user = AnonymousUser()
|
|
assert not user.has_role('admin')
|