def test_nav_group_auto_assigned(self):
node = NavItem(
NavItem(
'This Beard Stays',
NavItem('Foo2', NavURL('public.home')),
), NavItem(
'Bar',
NavItem('Bar2', NavURL('private.secret1')),
))
assert node.sub_nodes[0].nav_group == 'this-beard-stays'
def test_nav_group_manual_preserved(self):
node = NavItem(
NavItem('Foo',
NavItem('Foo2', NavURL('public.home')),
nav_group='this-beard-stays'),
NavItem(
'Bar',
NavItem('Bar2', NavURL('private.secret1')),
),
)
assert node.sub_nodes[0].nav_group == 'this-beard-stays'
def test_current_route_matched_nested(self):
node = NavItem(
NavItem(
'Foo',
NavItem('Foo2', NavURL('public.home')),
), NavItem(
'Bar',
NavItem('Bar2', NavURL('private.secret1')),
))
with flask.current_app.test_request_context('/'):
assert node.has_current_route
assert node.sub_nodes[0].has_current_route
assert not node.sub_nodes[1].has_current_route
def test_leaf_method_and_class_both_require(self, endpoint):
node = NavItem('Foo', NavURL(endpoint))
with flask.current_app.test_request_context('/'):
flask_login.logout_user()
assert not node.is_permitted
user = flask.current_app.auth_manager.entity_registry.user_cls.testing_create(
)
flask_login.login_user(user)
node.clear_authorization(user.get_id())
assert not node.is_permitted
perm1 = self.Permission.testing_create(token='permission1')
perm2 = self.Permission.testing_create(token='permission2')
user = flask.current_app.auth_manager.entity_registry.user_cls.testing_create(
permissions=[perm1])
flask_login.login_user(user)
node.clear_authorization(user.get_id())
assert not node.is_permitted
user = flask.current_app.auth_manager.entity_registry.user_cls.testing_create(
permissions=[perm2])
flask_login.login_user(user)
node.clear_authorization(user.get_id())
assert not node.is_permitted
user = flask.current_app.auth_manager.entity_registry.user_cls.testing_create(
permissions=[perm1, perm2])
flask_login.login_user(user)
node.clear_authorization(user.get_id())
assert node.is_permitted
def test_permitted_subnodes(self):
perm1 = self.Permission.testing_create(token='permission1')
node = NavItem(
'Menu',
NavItem('Index',
NavURL('public.home', requires_permissions='permission2')),
NavItem(
'Submenu',
NavItem(
'Profile',
NavURL('private.secret1',
requires_permissions='permission1')),
NavItem(
'Control Panel',
NavURL('private.secret2',
requires_permissions='permission2')), # noqa
NavItem(
'Accounts',
NavURL('private.secret3',
requires_permissions='permission1')),
),
NavItem(
'History',
NavURL('private.secret4', requires_permissions='permission2')),
)
with flask.current_app.test_request_context('/'):
flask_login.logout_user()
assert not node.permitted_sub_nodes
user = flask.current_app.auth_manager.entity_registry.user_cls.testing_create(
permissions=[perm1])
flask_login.login_user(user)
node.clear_authorization(user.get_id())
assert len(node.permitted_sub_nodes) == 1
assert node.permitted_sub_nodes[0].label == 'Submenu'
assert len(node.permitted_sub_nodes[0].permitted_sub_nodes) == 2
assert node.permitted_sub_nodes[0].permitted_sub_nodes[
0].label == 'Profile'
assert node.permitted_sub_nodes[0].permitted_sub_nodes[
1].label == 'Accounts'
def test_leaf_class_requires_user(self, endpoint):
node = NavItem('Foo', NavURL(endpoint))
with flask.current_app.test_request_context('/'):
flask_login.logout_user()
assert not node.is_permitted
user = flask.current_app.auth_manager.entity_registry.user_cls.testing_create(
)
flask_login.login_user(user)
node.clear_authorization(user.get_id())
assert node.is_permitted
def test_stem_requirement_from_subnode(self):
node = NavItem('Menu', NavItem('Foo', NavURL('private.secret1-class')))
with flask.current_app.test_request_context('/'):
flask_login.logout_user()
assert not node.is_permitted
user = flask.current_app.auth_manager.entity_registry.user_cls.testing_create(
)
flask_login.login_user(user)
node.clear_authorization(user.get_id())
assert node.is_permitted
def test_login_presence(self):
node = NavItem('Foo', NavURL('auth.login', requires_anonymous=True))
with flask.current_app.test_request_context('/'):
flask_login.logout_user()
assert node.is_permitted
user = flask.current_app.auth_manager.entity_registry.user_cls.testing_create(
)
flask_login.login_user(user)
node.clear_authorization(user.get_id())
assert not node.is_permitted
def test_leaf_method_requires_callable_permissions(self):
node = NavItem('Foo', NavURL('private.secret_callable'))
with flask.current_app.test_request_context('/'):
flask_login.logout_user()
assert not node.is_permitted
user = flask.current_app.auth_manager.entity_registry.user_cls.testing_create(
)
flask_login.login_user(user)
node.clear_authorization(user.get_id())
assert not node.is_permitted
flask_login.current_user.email = '*****@*****.**'
node.clear_authorization(user.get_id())
assert node.is_permitted
def test_leaf_class_requires_callable_permissions(self):
node = NavItem('Foo',
NavURL('callable_protected.callable-protected-class'))
with flask.current_app.test_request_context('/'):
flask_login.logout_user()
assert not node.is_permitted
user = flask.current_app.auth_manager.entity_registry.user_cls.testing_create(
)
flask_login.login_user(user)
node.clear_authorization(user.get_id())
assert not node.is_permitted
user.class_pass = True
user.blueprint_pass = True
node.clear_authorization(user.get_id())
assert node.is_permitted
def test_per_user_menu_items(self):
node = NavItem(
'Foo', NavURL('private.secret2',
requires_permissions='permission1'))
with flask.current_app.test_request_context('/'):
flask_login.logout_user()
assert not node.is_permitted
user = flask.current_app.auth_manager.entity_registry.user_cls.testing_create(
)
flask_login.login_user(user)
assert not node.is_permitted
perm1 = self.Permission.testing_create(token='permission1')
user = flask.current_app.auth_manager.entity_registry.user_cls.testing_create(
permissions=[perm1])
flask_login.login_user(user)
assert node.is_permitted
def test_leaf_class_blueprint_requires_permissions(self):
node = NavItem('Foo', NavURL('protected.protected-class'))
with flask.current_app.test_request_context('/'):
flask_login.logout_user()
assert not node.is_permitted
user = flask.current_app.auth_manager.entity_registry.user_cls.testing_create(
)
flask_login.login_user(user)
node.clear_authorization(user.get_id())
assert not node.is_permitted
perm1 = self.Permission.testing_create(token='permission1')
user = flask.current_app.auth_manager.entity_registry.user_cls.testing_create(
permissions=[perm1])
flask_login.login_user(user)
node.clear_authorization(user.get_id())
assert node.is_permitted
# Using unicode_literals instead of adding 'u' prefix to all stings that go to SA.
from __future__ import unicode_literals
import sys
from unittest import mock
import flask
import flask_login
import pytest
from keg_auth.libs.navigation import NavItem, NavURL
from keg_auth_ta import views
nav_menu = NavItem(
NavItem('Home', NavURL('public.home', arg1='foo')),
NavItem(
'Nesting',
NavItem('Secret1', NavURL('private.secret1')),
NavItem('Secret1 Class', NavURL('private.secret1-class')),
),
NavItem('Permissions On Stock Methods', NavURL('private.secret2')),
NavItem('Permissions On Methods', NavURL('private.someroute')),
NavItem('Permissions On Class And Method', NavURL('private.secret4')),
NavItem('Permissions On NavURL',
NavURL('private.secret3', requires_permissions='permission3')),
NavItem('User Manage', NavURL('auth.user:add')),
)
class TestViewMetaInfo(object):
def test_current_route_matched(self):
node = NavItem('Foo', NavURL('public.home'))
with flask.current_app.test_request_context('/'):
assert node.has_current_route
def test_nav_group_not_assigned(self):
node = NavItem('Foo', NavURL('public.home'))
assert not node.nav_group
def test_node_invalid_endpoint(self):
with pytest.raises(
Exception,
match='Endpoint pink_unicorns in navigation is not registered'
):
NavItem('Foo', NavURL('pink_unicorns')).is_permitted
def test_lazy_string_label(self, _):
# Pass a non-string label so is_lazy_string gets called.
label = 12
node = NavItem(label, NavURL('public.home'))
assert node.label == label
def test_current_route_not_exists(self):
node = NavItem('Foo', NavURL('public.home'))
with flask.current_app.test_request_context('/some-random-route'):
assert not node.has_current_route
