initial commit

examples/archinstall/tests/test_mirrorlist.py

@@ -0,0 +1,50 @@
+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',
+	]