mirror of
https://github.com/freedombox/FreedomBox.git
synced 2026-02-25 08:43:36 +00:00
c5e0c68bec
Related to https://salsa.debian.org/freedombox-team/freedombox/-/merge_requests/1957 Signed-off-by: James Valleroy <jvalleroy@mailbox.org> [sunil: Fix name of the fixture method copied from other code] Signed-off-by: Sunil Mohan Adapa <sunil@medhas.org> Reviewed-by: Sunil Mohan Adapa <sunil@medhas.org>
71 lines
1.9 KiB
Python
71 lines
1.9 KiB
Python
# SPDX-License-Identifier: AGPL-3.0-or-later
|
|
"""
|
|
Test module for sso module operations.
|
|
"""
|
|
|
|
import imp
|
|
import os
|
|
import pathlib
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from plinth.modules.sso.views import PRIVATE_KEY_FILE_NAME
|
|
|
|
|
|
def _action_file():
|
|
"""Return the path to the 'sso' actions file."""
|
|
current_directory = pathlib.Path(__file__).parent
|
|
return str(current_directory / '..' / '..' / '..' / '..' / 'actions' /
|
|
'auth-pubtkt')
|
|
|
|
|
|
sso_actions = imp.load_source('sso', _action_file())
|
|
|
|
|
|
@pytest.fixture(name='call_action')
|
|
def fixture_call_action(tmpdir, capsys):
|
|
"""Run actions with custom keys path."""
|
|
|
|
def _call_action(args, **kwargs):
|
|
sso_actions.KEYS_DIRECTORY = str(tmpdir)
|
|
with patch('argparse._sys.argv', ['sso'] + args):
|
|
sso_actions.main()
|
|
captured = capsys.readouterr()
|
|
return captured.out
|
|
|
|
return _call_action
|
|
|
|
|
|
@pytest.fixture(name='existing_key_pair')
|
|
def fixture_existing_key_pair(call_action):
|
|
"""A fixture to create key pair if needed."""
|
|
call_action(['create-key-pair'])
|
|
|
|
|
|
def test_generate_ticket(call_action, existing_key_pair):
|
|
"""Test generating a ticket."""
|
|
username = 'tester'
|
|
groups = 'freedombox-share,syncthing,web-search'
|
|
|
|
private_key_file = os.path.join(sso_actions.KEYS_DIRECTORY,
|
|
PRIVATE_KEY_FILE_NAME)
|
|
ticket = call_action([
|
|
'generate-ticket', '--uid', username, '--private-key-file',
|
|
private_key_file, '--tokens', groups
|
|
])
|
|
|
|
fields = {}
|
|
for item in ticket.split(';'):
|
|
try:
|
|
key, value = item.split('=')
|
|
fields[key] = value
|
|
except ValueError:
|
|
# The 'sig' field can also contain '='.
|
|
continue
|
|
|
|
assert fields['uid'] == username
|
|
assert int(fields['validuntil']) > 0
|
|
assert fields['tokens'] == groups
|
|
assert int(fields['graceperiod']) > 0
|