446 lines
10 KiB
Python
446 lines
10 KiB
Python
import json
|
|
import os
|
|
from importlib.metadata import version
|
|
from pathlib import Path
|
|
|
|
from pytest import MonkeyPatch
|
|
|
|
from archinstall.default_profiles.profile import CustomSetting, GreeterType
|
|
from archinstall.lib.args import ArchConfig, ArchConfigHandler, Arguments
|
|
from archinstall.lib.hardware import GfxDriver
|
|
from archinstall.lib.models.application import (
|
|
ApplicationConfiguration,
|
|
Audio,
|
|
AudioConfiguration,
|
|
BluetoothConfiguration,
|
|
PrintServiceConfiguration,
|
|
ZramConfiguration,
|
|
)
|
|
from archinstall.lib.models.authentication import AuthenticationConfiguration, U2FLoginConfiguration, U2FLoginMethod
|
|
from archinstall.lib.models.bootloader import Bootloader, BootloaderConfiguration
|
|
from archinstall.lib.models.device import DiskLayoutConfiguration, DiskLayoutType, Size
|
|
from archinstall.lib.models.locale import LocaleConfiguration
|
|
from archinstall.lib.models.mirrors import CustomRepository, CustomServer, MirrorConfiguration, MirrorRegion, SignCheck, SignOption
|
|
from archinstall.lib.models.network import NetworkConfiguration, Nic, NicType
|
|
from archinstall.lib.models.packages import Repository
|
|
from archinstall.lib.models.pacman import PacmanConfiguration
|
|
from archinstall.lib.models.profile import ProfileConfiguration
|
|
from archinstall.lib.models.users import Password, User
|
|
from archinstall.lib.profile.profiles_handler import profile_handler
|
|
from archinstall.lib.translationhandler import translation_handler
|
|
|
|
|
|
def test_default_args(monkeypatch: MonkeyPatch) -> None:
|
|
monkeypatch.setattr('sys.argv', ['archinstall'])
|
|
handler = ArchConfigHandler()
|
|
args = handler.args
|
|
assert args == Arguments(
|
|
config=None,
|
|
config_url=None,
|
|
creds=None,
|
|
creds_url=None,
|
|
creds_decryption_key=None,
|
|
silent=False,
|
|
dry_run=False,
|
|
script=None,
|
|
mountpoint=Path('/mnt'),
|
|
skip_ntp=False,
|
|
skip_wkd=False,
|
|
skip_boot=False,
|
|
debug=False,
|
|
offline=False,
|
|
no_pkg_lookups=False,
|
|
plugin=None,
|
|
plugin_url=None,
|
|
skip_version_check=False,
|
|
advanced=False,
|
|
)
|
|
|
|
|
|
def test_correct_parsing_args(
|
|
monkeypatch: MonkeyPatch,
|
|
config_fixture: Path,
|
|
creds_fixture: Path,
|
|
) -> None:
|
|
monkeypatch.setattr(
|
|
'sys.argv',
|
|
[
|
|
'archinstall',
|
|
'--config',
|
|
str(config_fixture),
|
|
'--config-url',
|
|
'https://example.com',
|
|
'--creds',
|
|
str(creds_fixture),
|
|
'--script',
|
|
'execution_script',
|
|
'--mountpoint',
|
|
'/tmp',
|
|
'--skip-ntp',
|
|
'--skip-wkd',
|
|
'--skip-boot',
|
|
'--debug',
|
|
'--offline',
|
|
'--no-pkg-lookups',
|
|
'--plugin',
|
|
'pytest_plugin.py',
|
|
'--skip-version-check',
|
|
'--advanced',
|
|
'--dry-run',
|
|
'--silent',
|
|
],
|
|
)
|
|
|
|
handler = ArchConfigHandler()
|
|
args = handler.args
|
|
|
|
assert args == Arguments(
|
|
config=config_fixture,
|
|
config_url='https://example.com',
|
|
creds=creds_fixture,
|
|
silent=True,
|
|
dry_run=True,
|
|
script='execution_script',
|
|
mountpoint=Path('/tmp'),
|
|
skip_ntp=True,
|
|
skip_wkd=True,
|
|
skip_boot=True,
|
|
debug=True,
|
|
offline=True,
|
|
no_pkg_lookups=True,
|
|
plugin=Path('pytest_plugin.py'),
|
|
plugin_url=None,
|
|
skip_version_check=True,
|
|
advanced=True,
|
|
)
|
|
|
|
|
|
def test_config_file_parsing(
|
|
monkeypatch: MonkeyPatch,
|
|
config_fixture: Path,
|
|
creds_fixture: Path,
|
|
) -> None:
|
|
monkeypatch.setattr(
|
|
'sys.argv',
|
|
[
|
|
'archinstall',
|
|
'--config',
|
|
str(config_fixture),
|
|
'--creds',
|
|
str(creds_fixture),
|
|
],
|
|
)
|
|
|
|
handler = ArchConfigHandler()
|
|
arch_config = handler.config
|
|
|
|
# TODO: Use the real values from the test fixture instead of clearing out the entries
|
|
arch_config.disk_config.device_modifications = [] # type: ignore[union-attr]
|
|
|
|
assert arch_config == ArchConfig(
|
|
version=version('archinstall'),
|
|
script='test_script',
|
|
app_config=ApplicationConfiguration(
|
|
bluetooth_config=BluetoothConfiguration(enabled=True),
|
|
audio_config=AudioConfiguration(audio=Audio.PIPEWIRE),
|
|
print_service_config=PrintServiceConfiguration(enabled=True),
|
|
),
|
|
auth_config=AuthenticationConfiguration(
|
|
root_enc_password=Password(enc_password='password_hash'),
|
|
users=[
|
|
User(
|
|
username='user_name',
|
|
password=Password(enc_password='password_hash'),
|
|
sudo=True,
|
|
groups=['wheel'],
|
|
),
|
|
],
|
|
u2f_config=U2FLoginConfiguration(
|
|
u2f_login_method=U2FLoginMethod.Passwordless,
|
|
passwordless_sudo=True,
|
|
),
|
|
),
|
|
locale_config=LocaleConfiguration(
|
|
kb_layout='us',
|
|
sys_lang='en_US',
|
|
sys_enc='UTF-8',
|
|
),
|
|
archinstall_language=translation_handler.get_language_by_abbr('en'),
|
|
disk_config=DiskLayoutConfiguration(
|
|
config_type=DiskLayoutType.Default,
|
|
device_modifications=[],
|
|
lvm_config=None,
|
|
mountpoint=None,
|
|
),
|
|
profile_config=ProfileConfiguration(
|
|
profile=profile_handler.parse_profile_config(
|
|
{
|
|
'custom_settings': {
|
|
'Hyprland': {
|
|
CustomSetting.SeatAccess: 'polkit',
|
|
},
|
|
'Sway': {
|
|
CustomSetting.SeatAccess: 'seatd',
|
|
},
|
|
'KDE Plasma': {
|
|
CustomSetting.PlasmaFlavor: 'plasma-meta',
|
|
},
|
|
},
|
|
'details': [
|
|
'Sway',
|
|
'Hyprland',
|
|
],
|
|
'main': 'Desktop',
|
|
}
|
|
),
|
|
gfx_driver=GfxDriver.AllOpenSource,
|
|
greeter=GreeterType.Lightdm,
|
|
),
|
|
mirror_config=MirrorConfiguration(
|
|
mirror_regions=[
|
|
MirrorRegion(
|
|
name='Australia',
|
|
urls=['http://archlinux.mirror.digitalpacific.com.au/$repo/os/$arch'],
|
|
),
|
|
],
|
|
custom_servers=[CustomServer('https://mymirror.com/$repo/os/$arch')],
|
|
optional_repositories=[Repository.Testing],
|
|
custom_repositories=[
|
|
CustomRepository(
|
|
name='myrepo',
|
|
url='https://myrepo.com/$repo/os/$arch',
|
|
sign_check=SignCheck.Required,
|
|
sign_option=SignOption.TrustAll,
|
|
),
|
|
],
|
|
),
|
|
network_config=NetworkConfiguration(
|
|
type=NicType.MANUAL,
|
|
nics=[
|
|
Nic(
|
|
iface='eno1',
|
|
ip='192.168.1.15/24',
|
|
dhcp=True,
|
|
gateway='192.168.1.1',
|
|
dns=[
|
|
'192.168.1.1',
|
|
'9.9.9.9',
|
|
],
|
|
),
|
|
],
|
|
),
|
|
bootloader_config=BootloaderConfiguration(
|
|
bootloader=Bootloader.Systemd,
|
|
uki=False,
|
|
removable=False,
|
|
),
|
|
hostname='archy',
|
|
kernels=['linux-zen'],
|
|
ntp=True,
|
|
packages=['firefox'],
|
|
pacman_config=PacmanConfiguration(parallel_downloads=66),
|
|
swap=ZramConfiguration(enabled=False),
|
|
timezone='UTC',
|
|
services=['service_1', 'service_2'],
|
|
custom_commands=["echo 'Hello, World!'"],
|
|
)
|
|
|
|
|
|
def test_deprecated_mirror_config_parsing(
|
|
monkeypatch: MonkeyPatch,
|
|
deprecated_mirror_config: Path,
|
|
) -> None:
|
|
monkeypatch.setattr(
|
|
'sys.argv',
|
|
[
|
|
'archinstall',
|
|
'--config',
|
|
str(deprecated_mirror_config),
|
|
],
|
|
)
|
|
|
|
handler = ArchConfigHandler()
|
|
arch_config = handler.config
|
|
|
|
assert arch_config.mirror_config == MirrorConfiguration(
|
|
mirror_regions=[
|
|
MirrorRegion(
|
|
name='Australia',
|
|
urls=['http://archlinux.mirror.digitalpacific.com.au/$repo/os/$arch'],
|
|
),
|
|
],
|
|
custom_servers=[],
|
|
optional_repositories=[Repository.Testing],
|
|
custom_repositories=[
|
|
CustomRepository(
|
|
name='my_mirror',
|
|
url='example.com',
|
|
sign_check=SignCheck.Optional,
|
|
sign_option=SignOption.TrustedOnly,
|
|
),
|
|
],
|
|
)
|
|
|
|
|
|
def test_deprecated_creds_config_parsing(
|
|
monkeypatch: MonkeyPatch,
|
|
deprecated_creds_config: Path,
|
|
) -> None:
|
|
monkeypatch.setattr(
|
|
'sys.argv',
|
|
[
|
|
'archinstall',
|
|
'--creds',
|
|
str(deprecated_creds_config),
|
|
],
|
|
)
|
|
|
|
handler = ArchConfigHandler()
|
|
arch_config = handler.config
|
|
|
|
assert arch_config.auth_config is not None
|
|
assert arch_config.auth_config.root_enc_password == Password(plaintext='rootPwd')
|
|
|
|
assert arch_config.auth_config.users == [
|
|
User(
|
|
username='user_name',
|
|
password=Password(plaintext='userPwd'),
|
|
sudo=True,
|
|
groups=['wheel'],
|
|
),
|
|
]
|
|
|
|
|
|
def test_deprecated_audio_config_parsing(
|
|
monkeypatch: MonkeyPatch,
|
|
deprecated_audio_config: Path,
|
|
) -> None:
|
|
monkeypatch.setattr(
|
|
'sys.argv',
|
|
[
|
|
'archinstall',
|
|
'--config',
|
|
str(deprecated_audio_config),
|
|
],
|
|
)
|
|
|
|
handler = ArchConfigHandler()
|
|
arch_config = handler.config
|
|
|
|
assert arch_config.app_config == ApplicationConfiguration(
|
|
audio_config=AudioConfiguration(audio=Audio.PIPEWIRE),
|
|
)
|
|
|
|
|
|
def test_encrypted_creds_with_arg(
|
|
monkeypatch: MonkeyPatch,
|
|
encrypted_creds_fixture: Path,
|
|
) -> None:
|
|
monkeypatch.setattr(
|
|
'sys.argv',
|
|
[
|
|
'archinstall',
|
|
'--creds',
|
|
str(encrypted_creds_fixture),
|
|
'--creds-decryption-key',
|
|
'master',
|
|
],
|
|
)
|
|
|
|
handler = ArchConfigHandler()
|
|
arch_config = handler.config
|
|
|
|
assert arch_config.auth_config is not None
|
|
assert arch_config.auth_config.root_enc_password == Password(enc_password='$y$j9T$FWCInXmSsS.8KV4i7O50H.$Hb6/g.Sw1ry888iXgkVgc93YNuVk/Rw94knDKdPVQw7')
|
|
assert arch_config.auth_config.users == [
|
|
User(
|
|
username='t',
|
|
password=Password(enc_password='$y$j9T$3KxMigAEnjtzbjalhLewE.$gmuoQtc9RNY/PmO/GxHHYvkZNO86Eeftg1Oc7L.QSO/'),
|
|
sudo=True,
|
|
groups=[],
|
|
),
|
|
]
|
|
|
|
|
|
def test_encrypted_creds_with_env_var(
|
|
monkeypatch: MonkeyPatch,
|
|
encrypted_creds_fixture: Path,
|
|
) -> None:
|
|
os.environ['ARCHINSTALL_CREDS_DECRYPTION_KEY'] = 'master'
|
|
monkeypatch.setattr(
|
|
'sys.argv',
|
|
[
|
|
'archinstall',
|
|
'--creds',
|
|
str(encrypted_creds_fixture),
|
|
],
|
|
)
|
|
|
|
handler = ArchConfigHandler()
|
|
arch_config = handler.config
|
|
|
|
assert arch_config.auth_config is not None
|
|
assert arch_config.auth_config.root_enc_password == Password(enc_password='$y$j9T$FWCInXmSsS.8KV4i7O50H.$Hb6/g.Sw1ry888iXgkVgc93YNuVk/Rw94knDKdPVQw7')
|
|
assert arch_config.auth_config.users == [
|
|
User(
|
|
username='t',
|
|
password=Password(enc_password='$y$j9T$3KxMigAEnjtzbjalhLewE.$gmuoQtc9RNY/PmO/GxHHYvkZNO86Eeftg1Oc7L.QSO/'),
|
|
sudo=True,
|
|
groups=[],
|
|
),
|
|
]
|
|
|
|
|
|
def test_example_config_parsing(
|
|
monkeypatch: MonkeyPatch,
|
|
example_config_fixture: Path,
|
|
example_creds_fixture: Path,
|
|
) -> None:
|
|
monkeypatch.setattr(
|
|
'sys.argv',
|
|
[
|
|
'archinstall',
|
|
'--config',
|
|
str(example_config_fixture),
|
|
'--creds',
|
|
str(example_creds_fixture),
|
|
],
|
|
)
|
|
|
|
handler = ArchConfigHandler()
|
|
arch_config = handler.config
|
|
|
|
assert arch_config.disk_config is not None
|
|
assert arch_config.profile_config is not None
|
|
assert arch_config.auth_config is not None
|
|
assert arch_config.auth_config.users
|
|
|
|
|
|
def test_example_config_partitions(example_config_fixture: Path) -> None:
|
|
# partition entries are only parsed when the configured device is present on
|
|
# the machine, which is never the case in CI, so read them here directly
|
|
config = json.loads(example_config_fixture.read_text())
|
|
device_modifications = config['disk_config']['device_modifications']
|
|
|
|
assert device_modifications
|
|
|
|
for device in device_modifications:
|
|
partitions = device['partitions']
|
|
|
|
assert partitions
|
|
|
|
previous_end = None
|
|
|
|
for partition in partitions:
|
|
assert 'dev_path' in partition
|
|
|
|
start = Size.parse_args(partition['start'])
|
|
end = start + Size.parse_args(partition['size'])
|
|
|
|
assert start.is_valid_start()
|
|
|
|
if previous_end is not None:
|
|
assert start >= previous_end
|
|
|
|
previous_end = end
|