40 lines
2.4 KiB
Python
40 lines
2.4 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""Execute the installer's actual overlay against a stale disposable rootfs."""
|
||
|
|
import pathlib, subprocess, tempfile, shutil, unittest
|
||
|
|
ROOT = pathlib.Path(__file__).resolve().parents[2]
|
||
|
|
class DoctorOverlayTests(unittest.TestCase):
|
||
|
|
def test_cached_rootfs_and_missing_payload(self):
|
||
|
|
source = (ROOT / 'image-recipe/_archived/build-auto-installer-iso.sh').read_text()
|
||
|
|
block = source.split('# BEGIN DOCTOR OVERLAY\n', 1)[1].split('# END DOCTOR OVERLAY', 1)[0]
|
||
|
|
with tempfile.TemporaryDirectory() as temp:
|
||
|
|
base = pathlib.Path(temp)
|
||
|
|
target = base / 'target'
|
||
|
|
media = base / 'media'
|
||
|
|
payload = media / 'archipelago/scripts'
|
||
|
|
payload.mkdir(parents=True)
|
||
|
|
units = target / 'etc/systemd/system'
|
||
|
|
units.mkdir(parents=True)
|
||
|
|
for directory in ['opt/archipelago/scripts', 'home/archipelago/archy/scripts']:
|
||
|
|
dest = target / directory
|
||
|
|
dest.mkdir(parents=True)
|
||
|
|
(dest / 'container-doctor.sh').write_text('UNSAFE OLD SCRIPT')
|
||
|
|
files = [ROOT / 'scripts/container-doctor.sh',
|
||
|
|
ROOT / 'image-recipe/configs/archipelago-doctor.service',
|
||
|
|
ROOT / 'image-recipe/configs/archipelago-doctor.timer']
|
||
|
|
for path in files:
|
||
|
|
shutil.copyfile(path, payload / path.name)
|
||
|
|
script = block.replace('/mnt/target', str(target))
|
||
|
|
for _ in range(2):
|
||
|
|
subprocess.run(['bash', '-c', script], env={'BOOT_MEDIA': str(media), 'PATH': '/usr/bin:/bin'}, check=True)
|
||
|
|
for directory in ['opt/archipelago/scripts', 'home/archipelago/archy/scripts']:
|
||
|
|
dest = target / directory / 'container-doctor.sh'
|
||
|
|
self.assertEqual(dest.read_bytes(), files[0].read_bytes())
|
||
|
|
self.assertEqual(dest.stat().st_mode & 0o777, 0o755)
|
||
|
|
for path in files[1:]:
|
||
|
|
self.assertEqual((units / path.name).read_bytes(), path.read_bytes())
|
||
|
|
(payload / 'container-doctor.sh').unlink()
|
||
|
|
failed = subprocess.run(['bash', '-c', script], env={'BOOT_MEDIA': str(media), 'PATH': '/usr/bin:/bin'}, capture_output=True)
|
||
|
|
self.assertNotEqual(failed.returncode, 0, 'Missing safety overlay must fail installation')
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|