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",
	]