51 lines
1.5 KiB
Python
51 lines
1.5 KiB
Python
from pathlib import Path
|
|
|
|
from archinstall.lib.mirrors import MirrorListHandler
|
|
|
|
|
|
def test_mirrorlist_no_country(mirrorlist_no_country_fixture: Path) -> None:
|
|
handler = MirrorListHandler(local_mirrorlist=mirrorlist_no_country_fixture)
|
|
handler.load_local_mirrors()
|
|
|
|
regions = handler.get_mirror_regions()
|
|
|
|
assert len(regions) == 1
|
|
assert regions[0].name == 'Local'
|
|
assert regions[0].urls == [
|
|
'https://geo.mirror.pkgbuild.com/$repo/os/$arch',
|
|
'https://america.mirror.pkgbuild.com/$repo/os/$arch',
|
|
]
|
|
|
|
|
|
def test_mirrorlist_with_country(mirrorlist_with_country_fixture: Path) -> None:
|
|
handler = MirrorListHandler(local_mirrorlist=mirrorlist_with_country_fixture)
|
|
handler.load_local_mirrors()
|
|
|
|
regions = handler.get_mirror_regions()
|
|
|
|
assert len(regions) == 1
|
|
assert regions[0].name == 'United States'
|
|
assert regions[0].urls == [
|
|
'https://geo.mirror.pkgbuild.com/$repo/os/$arch',
|
|
'https://america.mirror.pkgbuild.com/$repo/os/$arch',
|
|
]
|
|
|
|
|
|
def test_mirrorlist_multiple_countries(mirrorlist_multiple_countries_fixture: Path) -> None:
|
|
handler = MirrorListHandler(local_mirrorlist=mirrorlist_multiple_countries_fixture)
|
|
handler.load_local_mirrors()
|
|
|
|
regions = handler.get_mirror_regions()
|
|
|
|
assert len(regions) == 2
|
|
assert regions[0].name == 'United States'
|
|
assert regions[0].urls == [
|
|
'https://geo.mirror.pkgbuild.com/$repo/os/$arch',
|
|
'https://america.mirror.pkgbuild.com/$repo/os/$arch',
|
|
]
|
|
|
|
assert regions[1].name == 'Australia'
|
|
assert regions[1].urls == [
|
|
'https://au.mirror.pkgbuild.com/$repo/os/$arch',
|
|
]
|