65 lines
2.8 KiB
Python
65 lines
2.8 KiB
Python
import hashlib
|
|
import json
|
|
from pathlib import Path
|
|
import tempfile
|
|
import unittest
|
|
import zipfile
|
|
|
|
from package_windows_cli import package
|
|
from test_package_windows import executable
|
|
import stage_extraction_worker as extraction
|
|
|
|
|
|
class CliPackageTests(unittest.TestCase):
|
|
def fixture(self, directory):
|
|
root = Path(directory)
|
|
(root / 'Cargo.toml').write_text('workspace')
|
|
(root / 'Cargo.lock').write_text('lock')
|
|
for crate in ('searchi-core', 'searchi-media', 'searchi-extract-worker'):
|
|
base = root / 'crates' / crate
|
|
(base / 'src').mkdir(parents=True)
|
|
(base / 'Cargo.toml').write_text(crate)
|
|
(base / 'src' / ('main.rs' if crate == 'searchi-extract-worker' else 'lib.rs')).write_text('// fixture')
|
|
(root / 'LICENSE').write_text('license')
|
|
cli = root / 'cli.exe'
|
|
helper = root / 'helper.exe'
|
|
executable(cli)
|
|
executable(helper)
|
|
resources = root / 'resources'
|
|
extraction.stage(helper, extraction.fingerprint(root), resources, root)
|
|
return root, cli, resources, root / 'client.zip'
|
|
|
|
def test_integrity_and_minimal_contents(self):
|
|
with tempfile.TemporaryDirectory() as directory:
|
|
root, cli, resources, output = self.fixture(directory)
|
|
self.assertEqual(package(root, cli, resources, output), 'x86_64')
|
|
with zipfile.ZipFile(output) as archive:
|
|
self.assertEqual(set(archive.namelist()), {'searchi-cli.exe', 'searchi-extract-worker.exe',
|
|
'extraction-worker.json', 'LICENSE.txt', 'PACKAGE.json'})
|
|
manifest = json.loads(archive.read('PACKAGE.json'))
|
|
for record in manifest['files']:
|
|
data = archive.read(record['path'])
|
|
self.assertEqual(len(data), record['bytes'])
|
|
self.assertEqual(hashlib.sha256(data).hexdigest(), record['sha256'])
|
|
|
|
def test_rejected_helper_preserves_output(self):
|
|
for mutation in ('missing', 'tampered', 'architecture'):
|
|
with self.subTest(mutation=mutation), tempfile.TemporaryDirectory() as directory:
|
|
root, cli, resources, output = self.fixture(directory)
|
|
output.write_bytes(b'previous')
|
|
helper = resources / 'searchi-extract-worker.exe'
|
|
if mutation == 'missing':
|
|
helper.unlink()
|
|
elif mutation == 'tampered':
|
|
helper.write_bytes(helper.read_bytes() + b'changed')
|
|
else:
|
|
executable(cli, 0xAA64)
|
|
with self.assertRaises(ValueError):
|
|
package(root, cli, resources, output)
|
|
self.assertEqual(output.read_bytes(), b'previous')
|
|
self.assertEqual(list(root.glob('*.partial')), [])
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|