mirror of
https://github.com/freedombox/FreedomBox.git
synced 2026-02-04 08:13:38 +00:00
1ec16a7d58
Signed-off-by: Sunil Mohan Adapa <sunil@medhas.org> Reviewed-by: Veiko Aasa <veiko17@disroot.org>
68 lines
2.2 KiB
Python
68 lines
2.2 KiB
Python
# SPDX-License-Identifier: AGPL-3.0-or-later
|
|
"""
|
|
Test module for OpenVPN configuration.
|
|
"""
|
|
|
|
import os
|
|
from unittest.mock import patch
|
|
|
|
import pytest
|
|
|
|
from plinth.modules import openvpn
|
|
|
|
actions_name = 'openvpn'
|
|
|
|
|
|
@pytest.fixture(name='keys_directory')
|
|
def fixture_keys_directory(tmp_path):
|
|
return tmp_path
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def fixture_set_keys_directory(actions_module, keys_directory):
|
|
"""Set the keys directory in the actions module."""
|
|
actions_module.DH_PARAMS = f'{keys_directory}/pki/dh.pem'
|
|
actions_module.EC_PARAMS_DIR = f'{keys_directory}/pki/ecparams'
|
|
|
|
|
|
@pytest.fixture(name='conf_file')
|
|
def fixture_conf_file(tmp_path):
|
|
"""Fixture that returns an empty configuration file."""
|
|
return str(tmp_path / 'freedombox.conf')
|
|
|
|
|
|
def test_identify_rsa_configuration(conf_file):
|
|
"""Identify RSA configuration based on configuration file."""
|
|
with patch('plinth.modules.openvpn.SERVER_CONFIGURATION_FILE', conf_file):
|
|
with open(conf_file, 'w') as file_handle:
|
|
file_handle.write('dh /etc/openvpn/freedombox-keys/pki/dh.pem')
|
|
assert not openvpn.is_using_ecc()
|
|
|
|
|
|
def test_identify_ecc_configuration(conf_file):
|
|
"""Identify ECC configuration based on configuration file."""
|
|
with patch('plinth.modules.openvpn.SERVER_CONFIGURATION_FILE', conf_file):
|
|
with open(conf_file, 'w') as file_handle:
|
|
file_handle.write('dh none')
|
|
assert openvpn.is_using_ecc()
|
|
|
|
|
|
def test_is_setup_with_rsa(keys_directory, call_action):
|
|
"""is_setup should work with RSA configuration."""
|
|
with patch('plinth.actions.superuser_run', call_action):
|
|
(keys_directory / 'pki').mkdir()
|
|
dh_params_file = keys_directory / 'pki' / 'dh.pem'
|
|
dh_params_file.write_text('some content')
|
|
assert openvpn.is_setup()
|
|
os.remove(dh_params_file)
|
|
|
|
|
|
def test_is_setup_with_ecc(keys_directory, call_action):
|
|
"""is_setup should work with RSA configuration."""
|
|
with patch('plinth.actions.superuser_run', call_action):
|
|
(keys_directory / 'pki' / 'ecparams').mkdir(parents=True)
|
|
ec_params_file = keys_directory / 'pki' / 'ecparams' / 'somecurve.pem'
|
|
ec_params_file.write_text('some content')
|
|
assert openvpn.is_setup()
|
|
os.remove(ec_params_file)
|