mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2026-11-10 05:48:19 +01:00
Chris Lu
c1afb1b1cb
* test: pass installer scripts through stdin on Windows * ci: allow cold Intel macOS vector builds to finish * ci: retain completed Rust builds when later platform checks fail
60 lines
3.0 KiB
Python
60 lines
3.0 KiB
Python
import io
|
|
import os
|
|
from pathlib import Path
|
|
import subprocess
|
|
import tarfile
|
|
import tempfile
|
|
import unittest
|
|
import zipfile
|
|
|
|
SOURCE = (Path(__file__).resolve().parents[2] / 'install.sh').read_text().rsplit('\nmain\n', 1)[0]
|
|
BASH = os.environ.get('BASH_BINARY', 'bash')
|
|
|
|
|
|
class VectorInstallerTest(unittest.TestCase):
|
|
def shell(self, code, **env):
|
|
result = subprocess.run([BASH, '-s'], input=(SOURCE + '\n' + code + '\n').encode('utf-8'),
|
|
env=dict(os.environ, **env), capture_output=True)
|
|
result.stdout = result.stdout.decode('utf-8', errors='replace')
|
|
result.stderr = result.stderr.decode('utf-8', errors='replace')
|
|
return result
|
|
|
|
def test_platforms_and_asset_names(self):
|
|
for system, arch, supported, suffix in [
|
|
('linux', 'amd64', True, 'tar.gz'), ('linux', 'arm64', True, 'tar.gz'),
|
|
('darwin', 'amd64', True, 'tar.gz'), ('darwin', 'arm64', True, 'tar.gz'),
|
|
('windows', 'amd64', True, 'zip'), ('windows', 'arm64', False, 'zip'),
|
|
('freebsd', 'amd64', False, 'tar.gz'),
|
|
]:
|
|
with self.subTest(system=system, arch=arch):
|
|
result = self.shell('vector_platform_supported "$SYSTEM" "$ARCH"; result=$?; '
|
|
'vector_asset_name "$SYSTEM" "$ARCH"; exit "$result"', SYSTEM=system, ARCH=arch)
|
|
if supported:
|
|
self.assertEqual(result.returncode, 0, result.stderr)
|
|
self.assertEqual(result.stdout.strip(), f'weed-vector_{system}_{arch}.{suffix}')
|
|
else:
|
|
self.assertNotEqual(result.returncode, 0)
|
|
|
|
def test_installs_archive_binary(self):
|
|
for system in ['linux', 'darwin', 'windows']:
|
|
with self.subTest(system=system), tempfile.TemporaryDirectory() as directory:
|
|
directory = Path(directory)
|
|
name = 'weed-vector.exe' if system == 'windows' else 'weed-vector'
|
|
contents = b'vector engine fixture'
|
|
archive = directory / ('vector.zip' if system == 'windows' else 'vector.tar.gz')
|
|
if system == 'windows':
|
|
with zipfile.ZipFile(archive, 'w') as output:
|
|
output.writestr(name, contents)
|
|
else:
|
|
with tarfile.open(archive, 'w:gz') as output:
|
|
info = tarfile.TarInfo(name)
|
|
info.size = len(contents)
|
|
info.mode = 0o755
|
|
output.addfile(info, io.BytesIO(contents))
|
|
dest = directory / 'bin'
|
|
result = self.shell('download() { cp "$ARCHIVE" "$2"; }; VERSION=test; '
|
|
'INSTALL_DIR="$DEST"; install_component vector "$SYSTEM" amd64',
|
|
ARCHIVE=archive.as_posix(), DEST=dest.as_posix(), SYSTEM=system)
|
|
self.assertEqual(result.returncode, 0, result.stdout + result.stderr)
|
|
self.assertEqual((dest / name).read_bytes(), contents)
|