def __init__(self, config):
pluginmanager.PluginBase.__init__(self)
self._db = SessionDatabase()
self._expiration = int(config.get("expiration", 60)) * 60
hookmanager.register("HOOK_USER_DELETE", lambda user: self._db.delete_session(user=user))
def __init__(self, config):
pluginmanager.PluginBase.__init__(self)
self._db = SessionDatabase()
self._expiration = int(config.get("expiration", 60)) * 60
hookmanager.register("HOOK_USER_DELETE", lambda user: self._db.delete_session(user=user))
def test_baseview_render(baseview_fixtures):
"""
Test `prewikka.baseview.BaseView.render` method.
"""
from prewikka.baseview import BaseView
base_view = BaseView()
# register a fake HOOK to test all lines in baseview
hookmanager.register('HOOK_LOAD_HEAD_CONTENT',
'<script src="foo.js"></script>')
hookmanager.register('HOOK_LOAD_BODY_CONTENT', '<foo>bar</foo>')
# default render
base_view.render()
# no user
backup_user = env.request.user
env.request.user = None
base_view.render()
env.request.user = backup_user
# clean
hookmanager.unregister('HOOK_LOAD_HEAD_CONTENT',
'<script src="foo.js"></script>')
hookmanager.unregister('HOOK_LOAD_BODY_CONTENT', '<foo>bar</foo>')
def test_hookmanager_unregister():
"""
Test `prewikka.hookmanager.HookManager.unregister()` method.
"""
hook = 'hook_6'
method = lambda x: x
hookmanager.register(hook, method)
hookmanager.unregister(hook, method)
assert list(hookmanager.trigger(hook, 'bar')) == []
def test_hookmanager_register():
"""
Test `prewikka.hookmanager.HookManager.register()` method.
"""
hook = 'hook_1'
assert hook not in hookmanager.hookmgr
hookmanager.register(hook, lambda x: 42)
assert hook in hookmanager.hookmgr
def __init__(self):
env.dataprovider.check_datatype(self.type)
self._formatter = self.formatter(self.type)
self.all_fields = []
self._main_fields = list(self.default_columns.keys())
self.fields_info = collections.OrderedDict()
self.columns_properties = collections.OrderedDict()
self._prepare_fields()
view.View.__init__(self)
hookmanager.register(
"HOOK_LOAD_HEAD_CONTENT",
[resource.CSSLink("datasearch/css/datasearch.css")])
section = self.section or env.dataprovider.get_label(self.type)
tabs = self.tabs or (N_("Forensic"), N_("Dashboard"))
view.route("/%s/forensic/ajax_timeline" % self.name,
self.ajax_timeline)
view.route("/%s/forensic/ajax_table" % self.name, self.ajax_table)
view.route("/%s/forensic/ajax_details" % self.name, self.ajax_details)
view.route("/%s/forensic/ajax_infos" % self.name, self.ajax_infos)
view.route("/%s/forensic/ajax_groupby" % self.name, self.ajax_groupby)
view.route("/%s/forensic/csv_download" % self.name,
self.csv_download,
methods=["POST"])
view.route("/%s/forensic" % self.name,
self.forensic,
menu=(section, tabs[0]),
keywords=["listing", "inheritable"],
datatype=self.type,
priority=1,
help="#%sforensic" % self.type,
methods=["POST", "GET"])
view.route("/%s/dashboard" % self.name,
self.dashboard,
menu=(section, tabs[1]),
datatype=self.type,
help="#%sdashboard" % self.type,
methods=["POST", "GET"])
def test_hookmanager_trigger():
"""
Test `prewikka.hookmanager.HookManager.trigger()` method.
"""
# Test method return value
hook = 'hook_2'
hookmanager.register(hook, lambda x: x + '42')
assert list(hookmanager.trigger(hook, 'bar')) == ['bar42']
with pytest.raises(TypeError):
list(hookmanager.trigger(hook, 'foo', 'bar'))
with pytest.raises(TypeError):
list(hookmanager.trigger(hook, 'bar', type=int))
# Test exception handling
hook = 'hook_3'
hookmanager.register(hook, lambda x: 1 / x)
with pytest.raises(ZeroDivisionError):
list(hookmanager.trigger(hook, 0))
assert list(hookmanager.trigger(hook, 0, _except=lambda e: None)) == []
# Test constant value
hook = 'hook_4'
hookmanager.register(hook, 42)
assert list(hookmanager.trigger(hook, type=int)) == [42]
# Test return ordering
hook = 'hook_5'
hookmanager.register(hook, 'a', _order=2)
hookmanager.register(hook, 'b', _order=1)
hookmanager.register(hook, 'r', _order=3)
assert ''.join(hookmanager.trigger(hook)) == 'bar'
def __init__(self):
view.View.__init__(self)
hookmanager.register(
"HOOK_LOAD_HEAD_CONTENT",
[resource.CSSLink("statistics/css/gridstack.min.css")])
def __init__(self):
env.dataprovider.check_datatype("alert")
StaticStats.__init__(self)
hookmanager.register("HOOK_DASHBOARD_DEFAULT_GRAPHS", _DEFAULT_GRAPHS)
def __init__(self):
self._reinit()
self._joblist = set()
hookmanager.register("HOOK_PLUGINS_RELOAD", self._reinit)
# CSRF string will still match.
request.add_cookie("CSRF_COOKIE", get_token(request))
if request.method in ('GET', 'HEAD', 'OPTIONS', 'TRACE'):
return
if request.get_origin() != request.get_target_origin():
raise Exception("Origin check failed")
if request._csrf_cookie is None:
raise Exception("CSRF cookie not set")
# Check non-cookie token for match.
request_csrf_token = request.arguments.get(CSRF_POST_KEY)
if not request_csrf_token:
# Fall back to X-CSRFToken, to make things easier for AJAX,
# and possible for PUT/DELETE.
request_csrf_token = request.headers.get("x-csrftoken", '')
if not request_csrf_token:
raise Exception("CSRF token has not been provided")
if not _compare_salted_tokens(request_csrf_token, request._csrf_cookie):
raise Exception("CSRF token is invalid")
def _rotate_token(user):
env.request.web.add_cookie("CSRF_COOKIE", _get_new_csrf_token())
hookmanager.register("HOOK_SESSION_CREATE", _rotate_token)