44 lines
1.7 KiB
Python
44 lines
1.7 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
"""Exercise the release payload gate with real archives, including stale bytes."""
|
||
|
|
import importlib.util
|
||
|
|
import io
|
||
|
|
import pathlib
|
||
|
|
import tarfile
|
||
|
|
import tempfile
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
ROOT = pathlib.Path(__file__).resolve().parents[2]
|
||
|
|
spec = importlib.util.spec_from_file_location('check_ota_runtime', ROOT / 'scripts/check-ota-runtime.py')
|
||
|
|
validator = importlib.util.module_from_spec(spec)
|
||
|
|
spec.loader.exec_module(validator)
|
||
|
|
|
||
|
|
|
||
|
|
class RuntimePayloadTests(unittest.TestCase):
|
||
|
|
def test_actual_archives(self):
|
||
|
|
with tempfile.TemporaryDirectory() as directory:
|
||
|
|
base = pathlib.Path(directory)
|
||
|
|
source = base / 'first-boot-containers.sh'
|
||
|
|
source.write_bytes(b'#!/bin/sh\necho qualified\n')
|
||
|
|
for case, contents, valid in [
|
||
|
|
('qualified', [source.read_bytes()], True),
|
||
|
|
('stale', [b'#!/bin/sh\necho obsolete\n'], False),
|
||
|
|
('missing', [], False),
|
||
|
|
('duplicate', [source.read_bytes(), source.read_bytes()], False),
|
||
|
|
]:
|
||
|
|
with self.subTest(case=case):
|
||
|
|
archive = base / (case + '.tar.gz')
|
||
|
|
with tarfile.open(archive, 'w:gz') as output:
|
||
|
|
for content in contents:
|
||
|
|
member = tarfile.TarInfo('./archipelago-runtime/scripts/first-boot-containers.sh')
|
||
|
|
member.size = len(content)
|
||
|
|
output.addfile(member, io.BytesIO(content))
|
||
|
|
if valid:
|
||
|
|
validator.check(archive, source)
|
||
|
|
else:
|
||
|
|
with self.assertRaises(ValueError):
|
||
|
|
validator.check(archive, source)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|