68 lines
2.3 KiB
Python
68 lines
2.3 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# © 2026 TechnoLibre (http://www.technolibre.ca)
|
||
|
|
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl)
|
||
|
|
"""Le miroir apt d'une VM Ubuntu, tel que son cloud-config l'écrit.
|
||
|
|
|
||
|
|
Derrière le cache, le miroir est fixe (« uri: ») : la recherche de
|
||
|
|
cloud-init (« search: ») écarte tout miroir derrière un résolveur qui répond
|
||
|
|
à tout nom, et une VM qui tire d'un autre miroir que les précédentes ne
|
||
|
|
retrouve rien de ce que le cache a gardé.
|
||
|
|
"""
|
||
|
|
|
||
|
|
import sys
|
||
|
|
import tempfile
|
||
|
|
import unittest
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
import yaml
|
||
|
|
|
||
|
|
RACINE = Path(__file__).resolve().parent.parent
|
||
|
|
sys.path.insert(0, str(RACINE))
|
||
|
|
|
||
|
|
from script.qemu import deploy_qemu as DQ # noqa: E402
|
||
|
|
|
||
|
|
|
||
|
|
class TestLeMiroirApt(unittest.TestCase):
|
||
|
|
def _apt(self, *extra):
|
||
|
|
args = DQ.build_parser().parse_args(
|
||
|
|
["--distro", "ubuntu", "--hostname", "vm", *extra]
|
||
|
|
)
|
||
|
|
return yaml.safe_load(DQ.build_cloud_config(args, None, []))["apt"]
|
||
|
|
|
||
|
|
def _ca(self):
|
||
|
|
f = tempfile.NamedTemporaryFile(
|
||
|
|
"w", suffix=".pem", delete=False, encoding="utf-8"
|
||
|
|
)
|
||
|
|
self.addCleanup(Path(f.name).unlink)
|
||
|
|
f.write(
|
||
|
|
"-----BEGIN CERTIFICATE-----\nMIIB\n-----END CERTIFICATE-----\n"
|
||
|
|
)
|
||
|
|
f.close()
|
||
|
|
return f.name
|
||
|
|
|
||
|
|
def test_sans_cache_cloud_init_cherche(self):
|
||
|
|
apt = self._apt()
|
||
|
|
for bloc in ("primary", "security"):
|
||
|
|
self.assertEqual(apt[bloc][0]["search"], DQ.APT_MIRRORS_MAIN)
|
||
|
|
self.assertNotIn("uri", apt[bloc][0])
|
||
|
|
|
||
|
|
def test_derriere_le_cache_le_premier_miroir_est_fixe(self):
|
||
|
|
apt = self._apt("--cache-ca", self._ca())
|
||
|
|
for bloc in ("primary", "security"):
|
||
|
|
self.assertEqual(apt[bloc][0]["uri"], DQ.APT_MIRRORS_MAIN[0])
|
||
|
|
self.assertNotIn("search", apt[bloc][0])
|
||
|
|
|
||
|
|
def test_un_miroir_impose_est_fixe(self):
|
||
|
|
miroir = "http://miroir.invalid/ubuntu"
|
||
|
|
apt = self._apt("--apt-mirror", miroir)
|
||
|
|
self.assertEqual(apt["primary"][0]["uri"], miroir)
|
||
|
|
self.assertNotIn("search", apt["primary"][0])
|
||
|
|
|
||
|
|
def test_les_arches_ports_fixent_leur_propre_miroir(self):
|
||
|
|
lignes = DQ.apt_mirror_lines("arm64", fixe=True)
|
||
|
|
self.assertIn(f" uri: {DQ.APT_MIRRORS_PORTS[0]}", lignes)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|