updated the installer so that it should actually work
Some checks failed
Build / build (push) Failing after 5m23s
Some checks failed
Build / build (push) Failing after 5m23s
This commit is contained in:
tumillanino
373
examples/archinstall/tests/test_args.py
Normal file
373
examples/archinstall/tests/test_args.py
Normal file
@@ -0,0 +1,373 @@
|
||||
import os
|
||||
from importlib.metadata import version
|
||||
from pathlib import Path
|
||||
|
||||
from pytest import MonkeyPatch
|
||||
|
||||
from archinstall.default_profiles.profile import 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
|
||||
from archinstall.lib.models.authentication import AuthenticationConfiguration, U2FLoginConfiguration, U2FLoginMethod
|
||||
from archinstall.lib.models.bootloader import Bootloader
|
||||
from archinstall.lib.models.device import DiskLayoutConfiguration, DiskLayoutType
|
||||
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.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,
|
||||
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='pytest_plugin.py',
|
||||
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),
|
||||
),
|
||||
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': {
|
||||
'seat_access': 'polkit',
|
||||
},
|
||||
'Sway': {
|
||||
'seat_access': 'seatd',
|
||||
},
|
||||
},
|
||||
'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=Bootloader.Systemd,
|
||||
uki=False,
|
||||
hostname='archy',
|
||||
kernels=['linux-zen'],
|
||||
ntp=True,
|
||||
packages=['firefox'],
|
||||
parallel_downloads=66,
|
||||
swap=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=[],
|
||||
),
|
||||
]
|
||||
Reference in New Issue
Block a user